[控场AI]
论文ReWY

ReWY

一种针对Delta-rule线性注意力系列算子反向传播的代数重构优化方法,通过消去逆伴随、压缩WY矩阵相关矩阵乘法次数来加速梯度计算,适用于DeltaNet、GDN、KDA和GDN2等模型。

时间轴 (近 90 天)

11月3日

ReWY通过代数重构将WY矩阵相关的七个局部矩阵乘法压缩到三个,并消去逆伴随。

待验证50%
11月3日

ReWY的方法在精确算术下保持递推关系和梯度不变,不改变模型架构和训练目标。

待验证50%
11月3日

ReWY对DeltaNet、GDN、KDA和GDN2四种算子均适用。

待验证50%
11月3日

在NVIDIA GPU上,完整低层反向传播的几何平均加速为1.15–1.81倍。

待验证50%
11月3日

在H100和H200上,四种算子的加速在1.18–1.70倍之间,且保留原始FP32工作矩阵。

待验证50%
11月3日

在昇腾950DT上,独立公共autograd评测(包含梯度转换)取得1.09–2.89倍加速。

待验证50%
11月3日

ReWY利用中心化矩阵收缩替代逐对向量门控处理。

待验证50%
11月3日

ReWY的标量门控归约不再构造键的Gram矩阵,而是复用键梯度乘积。

待验证50%
11月3日

1.15–1.81倍的加速是低层反向传播的几何平均,全模型端到端收益会因模型规模和训练配置而异。

待验证50%

全部知识事实 (9)

来源文章