局部代理网络降低 Transformer 训练内存
热点事件持续更新
局部代理网络降低 Transformer 训练内存
1 篇报道1 个报道来源10 小时前更新
先了解这件事
AI 综述
研究者 Wu S. Zihan、Brian A. Yuan、Guillaume Bellec 提出一种面向 Transformer 的内存高效训练方法:为每个 Transformer block 训练一个局部代理网络,通过知识蒸馏预测全局网络输出,使梯度可独立回传,从而不必存储全部激活值。 作者称,该方法在深层线性网络中理论上精确,在非线性残差模型中梯度误差可被界定;在 CIFAR-100 的 vision transformer 实验中,其精度与内存的权衡优于其他局部算法。
AI 根据报道生成 · 53 分钟前更新
最新进展10月19日 12:00
面向 Transformer 的内存高效训练:用局部代理网络估计梯度报道时间线
沿着报道,了解事件的不同侧面。
10月19日
- arXiv cs.AI面向 Transformer 的内存高效训练:用局部代理网络估计梯度
研究者提出为每个 Transformer block 训练一个局部代理网络,通过知识蒸馏预测全局网络输出,使梯度可独立回传,从而避免存储全部激活值。理论上该方法在深层线性网络中精确,在非线性残差模型中梯度误差可被界定;在 CIFAR-100 上的 vision transformer 实验中取得优于其他局部算法的精度/内存权衡。
本事件热度走势
还没有足够的连续观测数据,暂不绘制趋势。