Properties of the stochastic approximation EM algorithm with mini-batch sampling

被引:8
|
作者
Kuhn, Estelle [1 ]
Matias, Catherine [2 ]
Rebafka, Tabea [2 ]
机构
[1] Univ Paris Saclay, MaIAGE, INRAE, Jouy En Josas, France
[2] Univ Paris, Sorbonne Univ, CNRS, Lab Probabil Stat & Modelisat LPSM, Paris, France
关键词
EM algorithm; Mini-batch sampling; Stochastic approximation; Monte Carlo Markov chain; MAXIMUM-LIKELIHOOD; CONVERGENCE;
D O I
10.1007/s11222-020-09968-0
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
To deal with very large datasets a mini-batch version of the Monte Carlo Markov Chain Stochastic Approximation Expectation-Maximization algorithm for general latent variable models is proposed. For exponential models the algorithm is shown to be convergent under classical conditions as the number of iterations increases. Numerical experiments illustrate the performance of the mini-batch algorithm in various models. In particular, we highlight that mini-batch sampling results in an important speed-up of the convergence of the sequence of estimators generated by the algorithm. Moreover, insights on the effect of the mini-batch size on the limit distribution are presented. Finally, we illustrate how to use mini-batch sampling in practice to improve results when a constraint on the computing time is given.
引用
收藏
页码:1725 / 1739
页数:15
相关论文
共 50 条
  • [1] Properties of the stochastic approximation EM algorithm with mini-batch sampling
    Estelle Kuhn
    Catherine Matias
    Tabea Rebafka
    [J]. Statistics and Computing, 2020, 30 : 1725 - 1739
  • [2] Mini-batch stochastic approximation methods for nonconvex stochastic composite optimization
    Ghadimi, Saeed
    Lan, Guanghui
    Zhang, Hongchao
    [J]. MATHEMATICAL PROGRAMMING, 2016, 155 (1-2) : 267 - 305
  • [3] Mini-batch stochastic approximation methods for nonconvex stochastic composite optimization
    Saeed Ghadimi
    Guanghui Lan
    Hongchao Zhang
    [J]. Mathematical Programming, 2016, 155 : 267 - 305
  • [4] An Asynchronous Mini-Batch Algorithm for Regularized Stochastic Optimization
    Feyzmahdavian, Hamid Reza
    Aytekin, Arda
    Johansson, Mikael
    [J]. IEEE TRANSACTIONS ON AUTOMATIC CONTROL, 2016, 61 (12) : 3740 - 3754
  • [5] An Asynchronous Mini-batch Algorithm for Regularized Stochastic Optimization
    Feyzmandavian, Hamid Reza
    Aytekin, Arda
    Johansson, Mikael
    [J]. 2015 54TH IEEE CONFERENCE ON DECISION AND CONTROL (CDC), 2015, : 1384 - 1389
  • [6] A mini-batch stochastic conjugate gradient algorithm with variance reduction
    Kou, Caixia
    Yang, Han
    [J]. JOURNAL OF GLOBAL OPTIMIZATION, 2023, 87 (2-4) : 1009 - 1025
  • [7] A mini-batch stochastic conjugate gradient algorithm with variance reduction
    Caixia Kou
    Han Yang
    [J]. Journal of Global Optimization, 2023, 87 : 1009 - 1025
  • [8] Convergence of the Mini-Batch SIHT Algorithm
    Damadi, Saeed
    Shen, Jinglai
    [J]. INTELLIGENT SYSTEMS AND APPLICATIONS, VOL 1, INTELLISYS 2023, 2024, 822 : 223 - 233
  • [9] Efficient Mini-batch Training for Stochastic Optimization
    Li, Muu
    Zhang, Tong
    Chen, Yuqiang
    Smola, Alexander J.
    [J]. PROCEEDINGS OF THE 20TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING (KDD'14), 2014, : 661 - 670
  • [10] Mini-batch stochastic subgradient for functional constrained optimization
    Singh, Nitesh Kumar
    Necoara, Ion
    Kungurtsev, Vyacheslav
    [J]. OPTIMIZATION, 2023,