SkewAdam, MoE modelleri için katmanlı bir optimize edici durumu tahsisi öneriyor: Yoğun omurga ağı için float32 momentum ve ayrıştırılmış ikinci momentleri...
论文研究HuggingFace Daily Papers(社区热门论文)
Bugünün AI bilgi özeti
SkewAdam, MoE modelleri için katmanlı bir optimize edici durumu tahsisi öneriyor: Yoğun omurga ağı
için float32 momentum ve ayrıştırılmış ikinci momentleri koruyor, uzman katmanı için sadece ayrıştırılmış ikinci momentleri tutuyor, yönlendirici için ise hassas ikinci momentleri saklıyor. 6.78 milyar parametreli MoE üzerinde, optimize edici durumu yalnızca 1.29 GB (AdamW'nin %2.6'sı) yer kaplıyor, maksimum eğitim belleği 81.4 GB'den 31.3 GB'ye düşüyor ve 40 GB hızlandırıcıya sığabiliyor.
Orijinal makale alıntısı
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 modelleri için katmanlı bir optimize edici durumu tahsisi öneriyor: Yoğun omurga ağı için float32 momentum ve ayrıştırılmış ikinci momentleri...