跳转到内容

Symbolic Discovery of Optimization Algorithms

作者: Xiangning Chen, Chen Liang, Da Huang, Esteban Real, Kaiyuan Wang, Yao Liu, Hieu Pham, Xuanyi Dong, Thang Luong, Cho-Jui Hsieh, Yifeng Lu, Quoc V. Le (2023)

arXiv: 2302.06675

领域

预训练

TLDR(中文)

用 AutoML 程序搜索自动发现优化器,产出 Lion:只取动量的符号与梯度符号比较生成更新,更新幅值恒定、不需要二阶动量、显存更省,在视觉和语言任务上匹配或超过 AdamW。证明了「优化器可以被搜出来」,启发了后续一系列符号化更新规则。

TLDR (English)

Discovers optimizers via AutoML program search, yielding Lion: it compares the sign of momentum with the sign of the gradient and emits constant-magnitude updates without any second moment, saving memory and matching or beating AdamW on vision and language tasks. Showed that optimizers can be searched for, inspiring a wave of sign-based update rules.

出现在这些文章里

同被引用

这些论文与本文出现在同一篇文章中

相关论文

同一领域的其他论文