Attn-QAT: 4-Bit Attention With Quantization-Aware Training
首次系统研究 4-bit (FP4) 注意力机制的量化感知训练,解决新一代 FP4 GPU 的端到端低精度推理瓶颈。
INSIGHTFP4 注意力量化的两个关键原则:反向传播中使用低精度重计算匹配前向,以及修正 Flash Attention 梯度中的精度假设。无需显式 outlier 缓解即可恢复质量。
业务关联随着 RTX 5090 等新 GPU 支持 FP4 计算,搜索排序模型(如 cross-encoder reranker)可利用此技术实现端到端 FP4 推理,大幅降低 serving 成本。
证据在 RTX 5090 上实现 1.5x 推理加速;在 diffusion 和语言模型上均验证有效;提供了 Triton 融合算子实现训练和 FP4 推理。