SkewAdam proposes a hierarchical optimizer state allocation for MoE models: retain float32 momentum and decomposed second-order moments for the dense backb...
论文研究HuggingFace Daily Papers(社区热门论文)
Today AI Intelligence Brief
SkewAdam proposes a hierarchical optimizer state allocation for MoE models: retain float32 momentum
and decomposed second-order moments for the dense backbone network, retain only decomposed second-order moments for the expert layers, and retain exact second-order moments for the router. On a 6.78B parameter MoE, the optimizer state only takes up 1.29 GB (2.6% of AdamW), peak training memory drops from 81.4 GB to 31.3 GB, making it possible to fit on a 40 GB accelerator.
Original Article Excerpt
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs :https://info.arxiv.org/labs/index.html.
The readable text on this page was extracted from the public source and organized with attribution, publication time and the original link. Copyright remains with the original author and publisher.
SkewAdam proposes a hierarchical optimizer state allocation for MoE models: retain float32 momentum and decomposed second-order moments for the dense backb...