## Introduction

The concept of clustering became trendy in the time of __Machine Learning__ development. This concept, clustering, deal with many real-world problems like â€śFinding Similar kinds of people on Twitterâ€ť, â€śTag Suggestions,â€ť â€śSearch Engines,â€ť â€śCustomer Segmentation,â€ť etc. The expectation-Maximization Algorithm, popularly known as the EM algorithm, is a Model-based clustering algorithm that tries to optimize the fit between the given data and some mathematical model. These methods, Model-based clustering methods, basically made an assumption that the data are generated by a mixture of an underlying probability distribution. The EM algorithm is just an extension of the popular k-means partitioning algorithm.

## Why EM algorithm(Model-Based Algorithm)?

According to the math, we know that the data is a mixture of probabilistic distributions, where a parametric probability distribution represents each cluster. Here each individual distribution is typically referred to as â€ścomponent distribution.â€ť The main problem of introducing model-based algorithms or methods is to solve the problem of estimating the probability distributions' parameters to fit the data best.

Example:

A simple finite mixture density model. The two clusters follow a Gaussian distribution with their own mean and standard deviation.

**Reference**: Data Mining Concepts and Techniques Second Edition Jiawei Han

### The EM Algorithm

The EM Algorithm, Expectation-Maximization Algorithm, is a popular iterative refinement algorithm used for finding the parameter estimates. It is simple and easy to implement. In general, it converges fast and may not lead to generating the global optimal. We said that this algorithm is an extension of k-means. This is because in this EM algorithm, instead of assigning each object to a dedicated cluster, EM gives each object to a cluster according to a weight representing the probability of membership. In other words, there are no strict boundaries between clusters. Therefore, new means are computed based on weighted measures.

The main idea of the EM algorithm is to start with an initial guess estimate of parameters. Then it iteratively rescores the objects against the mixture density produced by the parameter vector. Then the parameter estimates are updated by the use of rescored objects. This can be done in two important steps.

**Step-1:** Randomly make an initial guess of the parameter vector. In this step, we will randomly select k objects as cluster means or centers and make guesses for additional parameters needed.

**Step-2: **Update the parameters estimates by using two steps:

(i) **Expectation Step: **

In this step, we will assign each object **x _{i}**

_{ }to cluster

**C**_{k}_{ }with the above probability, where each probability follows the gaussian distribution with mean

*and with expectation*

**m**_{k}*Thus we can say that this step will give how probable that each object will belong to a particular cluster. We can say that these probabilities are the expected cluster membership probabilities for object*

**E**_{k}.**x**.

_{i}(II) **Maximization Step: **Here, we try to maximize the probabilities of each object belonging to that cluster by re-estimating the model parameters by using

### Applications

- It has the ability to fill the missing data in a sample.
- It can be used as a simple unsupervised data clustering algorithm.
- It is frequently used in estimating parameters for mixed models or any other mathematical models. For example, it is estimating the parameters for HMM model.
- The fields of image reconstruction and other fields like medicine and structural engineering also use the EM algorithm.