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

局部代理网络降低 Transformer 训练内存

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

先了解这件事

AI 综述

研究者 Wu S. Zihan、Brian A. Yuan、Guillaume Bellec 提出一种面向 Transformer 的内存高效训练方法:为每个 Transformer block 训练一个局部代理网络,通过知识蒸馏预测全局网络输出,使梯度可独立回传,从而不必存储全部激活值。 作者称,该方法在深层线性网络中理论上精确,在非线性残差模型中梯度误差可被界定;在 CIFAR-100 的 vision transformer 实验中,其精度与内存的权衡优于其他局部算法。

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

报道时间线

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

10月19日
  1. arXiv cs.AI
    面向 Transformer 的内存高效训练:用局部代理网络估计梯度

    研究者提出为每个 Transformer block 训练一个局部代理网络,通过知识蒸馏预测全局网络输出,使梯度可独立回传,从而避免存储全部激活值。理论上该方法在深层线性网络中精确,在非线性残差模型中梯度误差可被界定;在 CIFAR-100 上的 vision transformer 实验中取得优于其他局部算法的精度/内存权衡。

本事件热度走势

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