FlashAttention 动效 05

FP8 的坑与 incoherent processing

注意力矩阵里常有几个离群值,直接把整块压到 FP8/低精度时,量化的刻度被离群值撑大,普通值全被抹平。FA3 的做法是先把 Q、K 乘一个随机 ±1 对角矩阵和 Hadamard 矩阵——它是正交的,不改变 QKᵀ,却把离群值摊开。

模拟:64 个近似 N(0,1) 的值 + 3 个离群值,按 8 档刻度做对称量化

直接量化:刻度被离群值撑大

先做 incoherent 变换,再量化

相对量化误差 · 直接
—
相对量化误差 · incoherent
—