Revisiting Incremental Stochastic Majorization-Minimization Algorithms with Applications to Mixture of Experts
On the map: Mixture-of-experts → MoE theory and gating functions
Abstract (arXiv)
Processing high-volume, streaming data is increasingly common in modern statistics and machine learning, where batch-mode algorithms are often impractical because they require repeated passes over the full dataset. This has motivated incremental stochastic estimation methods, including the incremental stochastic Expectation-Maximization (EM) algorithm formulated via stochastic approximation. In this work, we revisit and analyze an incremental stochastic variant of the Majorization-Minimization (MM) algorithm, which generalizes incremental stochastic EM as a special case. Our approach relaxes key EM requirements, such as explicit latent-variable representations, enabling broader applicability and greater algorithmic flexibility. We establish theoretical guarantees for the incremental stochastic MM algorithm, proving consistency in the sense that the iterates converge to a stationary point characterized by a vanishing gradient of the objective. We demonstrate these advantages on a softmax-gated mixture of experts (MoE) regression problem, for which no stochastic EM algorithm is available. Empirically, our method consistently outperforms widely used stochastic optimizers, including stochastic gradient descent, root mean square propagation, adaptive moment estimation, and second-order clipped stochastic optimization. These results support the development of new incremental stochastic algorithms, given the central role of softmax-gated MoE architectures in contemporary deep neural networks for heterogeneous data modeling. Beyond synthetic experiments, we also validate practical effectiveness on two real-world datasets, including a bioinformatics study of dent maize genotypes under drought stress that integrates high-dimensional proteomics with ecophysiological traits, where incremental stochastic MM yields stable gains in predictive performance.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in MoE theory and gating functions, one of 12 directions in Mixture-of-experts. Builds on 48 of its 126 references that are in the map. First posted on arXiv 27 Jan 2026.
Builds on
48 of 126 references.
Closest ancestors · 8
Same direction, newest first
- 2025Dendrograms of Mixing Measures for Softmax-Gated Gaussian Mixture of Experts: Consistency Without Model Sweeps14 Oct
- 2025Model Selection for Gaussian-gated Gaussian Mixture of Experts Using Dendrograms of Mixing Measures19 May
- 2023A General Theory for Softmax Gating Multinomial Logistic Mixture of Experts22 Oct
- 2023Towards Convergence Rates for Parameter Estimation in Gaussian-gated Mixture of Experts12 May
- 2022Prediction Sets for High-Dimensional Mixture of Experts Models30 Oct
- 2021A non-asymptotic approach for model selection via penalization in high-dimensional mixture of experts models6 Apr
- 2019Convergence Rates for Gaussian Mixtures of Experts9 Jul
- 2018Regularized Maximum Likelihood Estimation and Feature Selection in Mixtures-of-Experts Models29 Oct
Other references · 40
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2024Mixtral of Expertscited by 1,823
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 265
- 2023From Sparse to Soft Mixtures of Expertscited by 248
- 2024A Survey on Mixture of Experts in Large Language Modelscited by 237
- 2021MoEfication: Transformer Feed-forward Layers are Mixtures of Expertscited by 152
- 2013Deep Learning of Representations: Looking Forwardcited by 84
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2019Non-asymptotic Analysis of Biased Stochastic Approximation Schemecited by 45
- 2024Mixture of Experts Meets Prompt-Based Continual Learningcited by 43
28 more
- 2014Incremental Majorization-Minimization Optimization with Application to Large-Scale Machine Learningcited by 35
- 2013Stochastic Majorization-Minimization Algorithms for Large-Scale Optimizationcited by 28
- 2021SpeechMoE2: Mixture-of-Experts Model with Improved Routingcited by 26
- 2024CompeteSMoE -- Effective Training of Sparse Mixture of Experts via Competitioncited by 23
- 2023HyperRouter: Towards Efficient Training and Inference of Sparse Mixture of Expertscited by 21
- 2016A Universal Approximation Theorem for Mixture of Experts Modelscited by 18
- 2019On the Global Convergence of (Fast) Incremental Expectation Maximization Methodscited by 12
- 2023Stochastic Approximation Beyond Gradient for Signal Processing and Machine Learningcited by 10
- 2025ODE approximation for the Adam algorithm: General and overparametrized settingcited by 10
- 2020Geom-SPIDER-EM: Faster Variance Reduced Stochastic Expectation Maximization for Nonconvex Finite-Sum Optimizationcited by 7
- 2020Fast Incremental Expectation Maximization for finite-sum optimization: nonasymptotic convergencecited by 7
- 2022Strong identifiability and parameter learning in regression with heterogeneous responsecited by 6
- 2017Stochastic Subsampling for Factorizing Huge Matricescited by 5
- 2019Properties of the Stochastic Approximation EM Algorithm with Mini-batch Samplingcited by 5
- 2021Nonconvex Optimization via MM Algorithms: Convergence Theorycited by 5
- 2021The perturbed prox-preconditioned spider algorithm: non-asymptotic convergence boundscited by 4
- 2015A Stochastic Majorize-Minimize Subspace Algorithm for Online Penalized Least Squares Estimationcited by 3
- 2020Online nonnegative CP-dictionary learning for Markovian datacited by 3
- 2021Online Stochastic DCA with applications to Principal Component Analysiscited by 3
- 2023Stochastic Variable Metric Proximal Gradient with variance reduction for non-convex composite optimizationcited by 3
- 2024Risk Bounds for Mixture Density Estimation on Compact Domains via the h-Lifted Kullback--Leibler Divergencecited by 3
- 2013A Stochastic Successive Minimization Method for Nonsmooth Nonconvex Optimization with Applications to Transceiver Design in Wireless Communication Networkscited by 2
- 2014Hyper-Spectral Image Analysis with Partially-Latent Regression and Spatial Markov Dependenciescited by 2
- 2022Stochastic regularized majorization-minimization with weakly convex and multi-convex surrogatescited by 2
- 2019Tensor-variate Mixture of Experts for Proportional Myographic Control of a Robotic Handcited by 1
- 2021Convergence analysis of stochastic higher-order majorization-minimization algorithmscited by 1
- 2022The Stochastic Proximal Distance Algorithmcited by 1
- 2025Federated Majorize-Minimization: Beyond Parameter Aggregationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Learning Mixtures of Experts with EM: A Mirror Descent Perspective
- 2019Regularized Estimation and Feature Selection in Mixtures of Generalized Linear Experts
- 2026Fast Model Selection and Stable Optimization for Softmax-Gated Multinomial-Logistic Mixture of Experts Models
- 2025Mixture of Many Zero-Compute Experts: A High-Rate Quantization Theory Perspective
- 2019Regularized Estimation and Feature Selection in Mixtures of Gaussian-Gated Experts Models
- 2025A General-Purpose Theorem for High-Probability Bounds of Stochastic Approximation with Polyak Averaging
- 2024Sigmoid Gating is More Sample Efficient than Softmax Gating in Mixture of Experts
- 2026On Bayesian Softmax-Gated Mixture-of-Experts Models