قدم SkewAdam توزيع حالات المحسن على شكل طبقات لنماذج MoE: احتفظ بزخم float32 والمصفوفة المربعة الثانية المفككة للشبكة الكثيفة الأساسية، واحتفظ فقط بالمصفوف...
论文研究HuggingFace Daily Papers(社区热门论文)
ملخص معلومات AI اليوم
قدم SkewAdam توزيع حالات المحسن على شكل طبقات لنماذج MoE: احتفظ بزخم float32 والمصفوفة المربعة
الثانية المفككة للشبكة الكثيفة الأساسية، واحتفظ فقط بالمصفوفة المربعة الثانية المفككة لطبقة الخبراء، واحتفظ بالمصفوفة المربعة الثانية الدقيقة للموجه. على نموذج MoE بمقدار 6.78 مليار معلمة، يحتل حالة المحسن مساحة 1.29 غيغابايت فقط (2.6% من AdamW)، وانخفضت ذروة ذاكرة التدريب من 81.4 غيغابايت إلى 31.3 غيغابايت، ويمكن تحميلها على مسرّع بسعة 40 غيغابايت.
مقتطف من المقال الأصلي
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 والمصفوفة المربعة الثانية المفككة للشبكة الكثيفة الأساسية، واحتفظ فقط بالمصفوف...