大数相减难题不止 FA:softmax 数值稳定或只能靠 fwd-bwd 一致性

YouJiacheng · x · 2026-10-04

接上一条讨论,作者指出这个大数相减问题即使不用 FlashAttention 也存在,因为 stable softmax 总是需要减去最大值。他因此提出一个思路:也许只需要保证 forward-backward 一致性即可,并对此表示无奈(":(")。

所属事件:FlashAttention 大数相减隐患引热议(2 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →