跳到正文
热点事件持续更新

Sonar-MoE:反向传播中更新专家以省显存

1 篇报道1 个报道来源9 小时前更新

先了解这件事

AI 综述

Avni Mittal 等作者提出 Sonar-MoE,把 optimizer-in-backward 方法用于批量 MoE 专家张量:在反向传播中直接更新每个专家张量,并立即释放其梯度,使 GPU 最多只保留一层专家梯度,从而降低 MoE 训练的显存占用。 该做法针对的是 MoE 训练中专家梯度占用显存的问题,作者称由此实现了更省内存的训练。目前披露的信息限于方法本身,未给出具体显存节省幅度或训练效果数据。

AI 根据报道生成 · 42 分钟前更新

报道时间线

沿着报道,了解事件的不同侧面。

11月4日
  1. arXiv cs.CL
    Sonar-MoE:在反向传播中更新专家,实现更省内存的 MoE 训练

    Sonar-MoE 将 optimizer-in-backward 引入批量 MoE 专家张量,在反向传播中直接更新每个专家张量并立即释放其梯度,使 GPU 最多只保留一层专家梯度。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。