Accelerated distributed stochastic optimization methods and applications in machine learning

加速分布式随机优化方法及其在机器学习中的应用

• 批准号: 2208394
• 负责人: Yangyang Xu
• 金额: $ 25万
• 依托单位: Rensselaer Polytechnic Institute
• 依托单位国家: 美国
• 项目类别: Standard Grant
• 财政年份: 2022
• 资助国家: 美国
• 起止时间: 2022-07-01 至 2025-06-30
• 项目状态: 未结题
• 来源: https://www.nsf.gov/awardsearch/showAward?AWD_ID=2208394&HistoricalAwards=false
• 关键词: Accelerated; distributed; stochastic; optimization; methods

Machine learning, and in particular, deep learning, has become increasingly impactful in a wide range of applications, including face recognition, digital image classification, natural language processing, self-driving vehicles, and scientific computing. The success of deep learning largely depends on the availability of a huge amount of data. This "big" data, on one hand, enables successful learning of the underlying distributions of the data, and thus the learned model can yield high prediction accuracy on new data points that follow similar distributions. On the other hand, the huge amount of data raises great challenges when designing efficient numerical approaches. This project focuses on addressing the challenges that are caused by distributed "big" data that can contain private information, from the computational and mathematical perspectives. Research findings from this project will be included in graduate-level topics courses, undergraduate and graduate students will be trained in this field and will participate in this research, and a weekly seminar will be organized to exchange ideas relevant to this project. New computational approaches will be developed for training machine learning models on a cluster of computing nodes as well as solving decentralized multi-agent optimization problems that have the capacity to handle coupling constraints. The main goal is to design fast-convergent and communication-efficient optimization methods with theoretical guarantees for solving large-scale distributed machine learning problems. Accelerated compressed proximal stochastic gradient methods will be designed for distributed composite smooth stochastic problems, accelerated compressed stochastic subgradient methods will be designed for distributed nonconvex nonsmooth problems, optimal decentralized stochastic gradient methods will be designed for solving multi-agent optimization with nonlinear coupling constraints, and asynchronous implementations will be performed in the proposed methods in order to have high parallelization speed up.This award reflects NSF's statutory mission and has been deemed worthy of support through evaluation using the Foundation's intellectual merit and broader impacts review criteria.

机器学习，特别是深度学习，在人脸识别、数字图像分类、自然语言处理、自动驾驶车辆和科学计算等广泛应用中的影响力越来越大。深度学习的成功很大程度上取决于大量数据的可用性。一方面，这种“大”数据能够成功学习数据的底层分布，因此学习的模型可以对遵循相似分布的新数据点产生较高的预测精度。另一方面，大量的数据给设计有效的数值方法带来了巨大的挑战。该项目的重点是从计算和数学的角度解决可能包含私人信息的分布式“大”数据所带来的挑战。该项目的研究成果将纳入研究生水平的主题课程，本科生和研究生将接受该领域的培训并参与该研究，并将每周组织一次研讨会来交流与该项目相关的想法。 将开发新的计算方法，用于在计算节点集群上训练机器学习模型，并解决能够处理耦合约束的分散多智能体优化问题。主要目标是设计快速收敛、通信高效的优化方法，为解决大规模分布式机器学习问题提供理论保证。加速压缩近端随机梯度方法将被设计用于分布式复合光滑随机问题，加速压缩随机次梯度方法将被设计用于分布式非凸非光滑问题，最优分散随机梯度方法将被设计用于求解具有非线性耦合约束的多智能体优化，以及异步实现将在提议的方法中执行，以便获得高并行化速度。该奖项反映了 NSF 的法定使命，并通过使用基金会的智力价值和更广泛的影响审查进行评估，被认为值得支持 标准。

项目成果

Proximal Stochastic Recursive Momentum Methods for Nonconvex Composite Decentralized Optimization

• DOI: 10.1609/aaai.v37i7.26087
• 发表时间: 2022-11
• 作者: Gabriel Mancino-Ball;Shengnan Miao;Yangyang Xu;Jiewei Chen

Parallel and distributed asynchronous adaptive stochastic gradient methods

• DOI: 10.1007/s12532-023-00237-5
• 发表时间: 2020-02
• 期刊: Mathematical Programming Computation
• 影响因子: 6.3
• 作者: Yangyang Xu;Yibo Xu;Yonggui Yan;Colin Sutcher-Shepard;Leopold Grinberg;Jiewei Chen

A Decentralized Primal-Dual Framework for Non-Convex Smooth Consensus Optimization

• DOI: 10.1109/tsp.2023.3239799
• 发表时间: 2021-07
• 期刊: IEEE Transactions on Signal Processing
• 影响因子: 5.4
• 作者: Gabriel Mancino-Ball;Yangyang Xu;Jiewei Chen