Skip to content
Financial Machine Learning / Unsupervised Learning and Latent Structure

Mixture of Gaussians and the EM Algorithm

k-means makes a hard decision at every step: each point belongs entirely to one cluster or another. The Gaussian mixture model (GMM) softens this into a genuinely probabilistic story — every point has some probability of having come from each of kk underlying Gaussian components — and the EM algorithm is exactly how such a model's parameters are fitted when the component membership is never directly observed. This note derives EM specifically for the Gaussian mixture case; the derivation sub-note generalises the same idea to any latent-variable model via Jensen's inequality.

In this lesson

  1. Learning objectives
  2. The Gaussian mixture generative model
  3. The EM algorithm for Gaussian mixtures
  4. EM as soft k-means
  5. Worked example

Before you start

The full lesson, with worked examples and exercise solutions, is for members.

Read the full lesson

New to SelfMadeQuant.com? See membership options or browse the Financial Machine Learning syllabus.