Federated-EM with heterogeneity mitigation and variance reduction