arXiv · 2505.06921
Stochastic ADMM with batch size adaptation for nonconvex nonsmooth optimization
Abstract
Stochastic alternating direction method of multipliers (SADMM) is a popular method for solving nonconvex nonsmooth optimization in various applications. However, it typically requires an empirical selection of the static batch size for gradient estimation, resulting in a challenging trade-off between variance reduction and computational cost. This paper proposes adaptive batch size SADMM, a practical method that dynamically adjusts the batch size based on accumulated differences along the optimization path. We develop a simple convergence analysis to handle the dependence of batch size adaptation that matches the best-known complexity with flexible parameter choices. We further extend this adaptive scheme to reduce the overall complexity of the popular variance-reduced methods, SVRG-ADMM and SPIDER-ADMM. Numerical results validate the effectiveness of our proposed methods.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Jiachen Jin, Kangkang Deng, Boyu Wang, Hongxia Wang. 2025-05-11. Stochastic ADMM with batch size adaptation for nonconvex nonsmooth optimization. https://doi.org/10.1016/j.apnum.2025.12.007
Cite the original work for its findings. Save a collection to share your selection of sources.