Local linear convergence of gradient methods for overparameterized Gaussian mixtures

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Jingxing, Charisopoulos, Vasileios, Fazel, Maryam
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913172271661056
author Wang, Jingxing
Charisopoulos, Vasileios
Fazel, Maryam
author_facet Wang, Jingxing
Charisopoulos, Vasileios
Fazel, Maryam
contents We study the problem of learning Gaussian mixture models under overparameterization. Prior work has shown that while overparameterization is essential for avoiding spurious local optima and enables global recovery of the ground-truth model using the gradient-EM (expectation-maximization) algorithm, it can dramatically slow down the local rate of convergence. Under certain assumptions on the mixture weights, we show that a standard divergence measure minimized by statistical learning procedures possesses a manifold of slow growth on which the well-known Polyak stepsize reduces the loss geometrically, and design a gradient-based method that converges to minimizers at a locally linear rate. Additionally, we show that our method converges to nearly optimal solutions -- up to a natural misspecification threshold -- for mixtures with arbitrary weights. At a high level, the method alternates between several "short" gradient descent steps that approach the manifold and "long" Polyak steps that contract the distance to minimizers. Our results suggest that slow convergence is not an intrinsic challenge of overparameterization, but can be overcome by exploiting the favorable structure of the loss landscape.
format Preprint
id arxiv_https___arxiv_org_abs_2605_30936
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Local linear convergence of gradient methods for overparameterized Gaussian mixtures
Wang, Jingxing
Charisopoulos, Vasileios
Fazel, Maryam
Machine Learning
Optimization and Control
We study the problem of learning Gaussian mixture models under overparameterization. Prior work has shown that while overparameterization is essential for avoiding spurious local optima and enables global recovery of the ground-truth model using the gradient-EM (expectation-maximization) algorithm, it can dramatically slow down the local rate of convergence. Under certain assumptions on the mixture weights, we show that a standard divergence measure minimized by statistical learning procedures possesses a manifold of slow growth on which the well-known Polyak stepsize reduces the loss geometrically, and design a gradient-based method that converges to minimizers at a locally linear rate. Additionally, we show that our method converges to nearly optimal solutions -- up to a natural misspecification threshold -- for mixtures with arbitrary weights. At a high level, the method alternates between several "short" gradient descent steps that approach the manifold and "long" Polyak steps that contract the distance to minimizers. Our results suggest that slow convergence is not an intrinsic challenge of overparameterization, but can be overcome by exploiting the favorable structure of the loss landscape.
title Local linear convergence of gradient methods for overparameterized Gaussian mixtures
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2605.30936