SkewAdam ने MoE मॉडल के लिए स्तरित ऑप्टिमाइज़र स्टेट असाइनमेंट प्रस्तावित किया: डेंस बैकबोन नेटवर्क के लिए float32 मोमेंटम और फैक्टराइज्ड सेकंड-ऑर्डर मैट्र...
论文研究HuggingFace Daily Papers(社区热门论文)
आज का AI इंटेलिजेंस ब्रीफ
SkewAdam ने MoE मॉडल के लिए स्तरित ऑप्टिमाइज़र स्टेट असाइनमेंट प्रस्तावित किया: डेंस बैकबोन नेटवर्क
के लिए float32 मोमेंटम और फैक्टराइज्ड सेकंड-ऑर्डर मैट्रिक्स सुरक्षित रखें, विशेषज्ञ स्तर के लिए केवल फैक्टराइज्ड सेकंड-ऑर्डर मैट्रिक्स और राउटर के लिए सटीक सेकंड-ऑर्डर मैट्रिक्स सुरक्षित रखें। 6.78B पैरामीटर MoE पर, ऑप्टिमाइज़र स्टेट केवल 1.29 GB रखता है (AdamW का 2.6%), पीक ट्रेनिंग मेमोरी 81.4 GB से घटकर 31.3 GB हो गई, और इसे 40 GB एक्सेलेरेटर में डाल सकते हैं।
मूल लेख का अंश
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.
SkewAdam ने MoE मॉडल के लिए स्तरित ऑप्टिमाइज़र स्टेट असाइनमेंट प्रस्तावित किया: डेंस बैकबोन नेटवर्क के लिए float32 मोमेंटम और फैक्टराइज्ड सेकंड-ऑर्डर मैट्र...