SkewAdam เสนอการจัดสรรสถานะตัวปรับแต่งแบบชั้นสำหรับโมเดล MoE: เก็บความเฉื่อย float32
และเมทริกซ์อันดับสองแยกของเครือข่ายหลักหนาแน่น สำหรับชั้นผู้เชี่ยวชาญเก็บเพียงเมทริกซ์อันดับสองแยก สำหรับตัวกำหนดเส้นทางเก็บเมทริกซ์อันดับสองที่แม่นยำ ใน MoE ที่มีพารามิเตอร์ 6.78B สถานะตัวปรับแต่งใช้เพียง 1.29 GB (2.6% ของ AdamW) หน่วยความจำการฝึกสูงสุดลดจาก 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.


