arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.13751stat.MLcs.AIcs.LG
收藏

基于元梯度下降的机器学习训练优化

Optimizing ML Training with Metagradient Descent

MIT(麻省理工学院);Stanford(斯坦福大学);UIUC(伊利诺伊大学厄巴纳-香槟分校)

Logan Engstrom, Andrew Ilyas, Benjamin Chen, Axel Feldmann, William Moses, Aleksander Madry

AI总结:

针对大规模ML训练配置难的问题,提出元梯度计算算法与平滑训练框架,用元梯度下降优化训练设置,在数据集选择、抗投毒、学习率调度上取得显著效果。

摘要:

大规模机器学习模型训练的一大挑战是配置训练流程以最大化模型性能,即从庞大的设计空间中找到最佳训练设置。本研究提出了一种基于梯度的方法来解决该问题。我们首先提出一种可高效大规模计算元梯度(即贯穿模型训练过程的梯度)的算法,随后引入“平滑模型训练”框架,实现利用元梯度的有效优化。借助元梯度下降(MGD),我们大幅改进了现有数据集选择方法,性能比会降低准确率的数据投毒攻击高出一个数量级,还能自动找到具有竞争力的学习率调度方案。

英文摘要:

A major challenge in training large-scale machine learning models is configuring the training process to maximize model performance, i.e., finding the best training setup from a vast design space. In this work, we unlock a gradient-based approach to this problem. We first introduce an algorithm for efficiently calculating metagradients -- gradients through model training -- at scale. We then introduce a "smooth model training" framework that enables effective optimization using metagradients. With metagradient descent (MGD), we greatly improve on existing dataset selection methods, outperform accuracy-degrading data poisoning attacks by an order of magnitude, and automatically find competitive learning rate schedules.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑