随机方差缩减 ADMM
机器学习
2016-10-18 v3 最优化与控制
机器学习
摘要
交替方向乘子法 (ADMM) 是机器学习中一种强大的优化求解器。最近,随机 ADMM 已与随机梯度的方差缩减方法相结合,产生了具有快速收敛速度和低迭代复杂度的 SAG-ADMM 和 SDCA-ADMM。然而,它们的空间需求仍然可能很高。在本文中,我们提出将 ADMM 与随机方差缩减梯度 (SVRG) 方法相结合。与另一种最近称为 SCAS-ADMM 的结合尝试不同,所提出的算法保留了 SAG-ADMM 和 SDCA-ADMM 的快速收敛优势,但更具优势的是其存储需求非常低,甚至与样本量 无关。我们还将所提出的方法扩展到非凸问题,并获得了 的收敛速度。实验结果表明,它与 SAG-ADMM 和 SDCA-ADMM 一样快,比 SCAS-ADMM 快得多,并且可用于更大的数据集。
引用
@article{arxiv.1604.07070,
title = {Stochastic Variance-Reduced ADMM},
author = {Shuai Zheng and James T. Kwok},
journal= {arXiv preprint arXiv:1604.07070},
year = {2016}
}