Passer à la navigation principale Passer à la recherche Passer au contenu principal

Geom-Spider-EM: Faster variance reduced stochastic expectation maximization for nonconvex finite-sum optimization

  • Université de Toulouse
  • The Chinese University of Hong Kong

Résultats de recherche: Contribution à un journalArticle de conférenceRevue par des pairs

Résumé

The Expectation Maximization (EM) algorithm is a key reference for inference in latent variable models; unfortunately, its computational cost is prohibitive in the large scale learning setting. In this paper, we propose an extension of the Stochastic Path-Integrated Differential EstimatoR EM (SPIDER-EM) and derive complexity bounds for this novel algorithm, designed to solve smooth nonconvex finite-sum optimization problems. We show that it reaches the same state of the art complexity bounds as SPIDER-EM; and provide conditions for a linear rate of convergence. Numerical results support our findings.

langue originaleAnglais
Pages (de - à)3135-3139
Nombre de pages5
journalICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings
Volume2021-June
Les DOIs
étatPublié - 1 janv. 2021
Evénement2021 IEEE International Conference on Acoustics, Speech, and Signal Processing, ICASSP 2021 - Virtual, Toronto, Canada
Durée: 6 juin 202111 juin 2021

Empreinte digitale

Examiner les sujets de recherche de « Geom-Spider-EM: Faster variance reduced stochastic expectation maximization for nonconvex finite-sum optimization ». Ensemble, ils forment une empreinte digitale unique.

Contient cette citation