Adam-mini: Use Fewer Learning Rates To Gain More
Yushun Zhang, Congliang Chen, Ziniu Li +5 authors
Adam-mini reduces memory usage by optimizing learning rate allocation in parameter blocks, achieving comparable or better performance than AdamW with increased throughput.