The maximization step, or M step, updates each Gaussian’s mean, covariance, and density using the soft assignments calculated during the expectation step. section 5.6.1
The update uses probability-weighted calculations:
- Compute each covariance by weighting points according to their probability of belonging to Gaussian j. section 5.6.1
- Compute each mean as the probability-weighted mean of the points assigned to Gaussian j. section 5.6.1
- Compute each density by summing the probabilities for Gaussian j and dividing by the total number of points. section 5.6.1
- Repeat the expectation and maximization steps until the model converges and the log-likelihood is maximized. section 5.6.1
(5.11)
Inspect the covariance update formula alongside the corresponding mean, density, and probability equations. section 5.6.1
