MicroAdam: Accurate Adaptive Optimization with Low Space Overhead and Provable Convergence
CoRR(2024)
摘要
We propose a new variant of the Adam optimizer [Kingma and Ba, 2014] called
MICROADAM that specifically minimizes memory overheads, while maintaining
theoretical convergence guarantees. We achieve this by compressing the gradient
information before it is fed into the optimizer state, thereby reducing its
memory footprint significantly. We control the resulting compression error via
a novel instance of the classical error feedback mechanism from distributed
optimization [Seide et al., 2014, Alistarh et al., 2018, Karimireddy et al.,
2019] in which the error correction information is itself compressed to allow
for practical memory gains. We prove that the resulting approach maintains
theoretical convergence guarantees competitive to those of AMSGrad, while
providing good practical performance. Specifically, we show that MICROADAM can
be implemented efficiently on GPUs: on both million-scale (BERT) and
billion-scale (LLaMA) models, MicroAdam provides practical convergence
competitive to that of the uncompressed Adam baseline, with lower memory usage
and similar running time. Our code is available at
https://github.com/IST-DASLab/MicroAdam.
更多查看译文
AI 理解论文
溯源树
样例
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要