Rethinking Attention Output Projection: Structured Hadamard Transforms for Efficient Transformers
Multi-Head Attention 的稠密输出投影层参数量随模型维度二次增长,显著占据参数总量、内存和推理成本。本文提出用固定的无参数 Walsh-Hadamard 变换 (WHT) 加对角仿射变换替代该投影,消除了 O(d²) 的参数矩阵,将投影操作降为 O(d log d) 复杂度。实验表明该方法在保持模型性能的同时大幅减少参数量和推理延迟。
INSIGHT用结构化 Hadamard 变换替代 Attention 输出投影是一种范式级效率优化,核心洞察在于输出投影的信息混合功能可由确定性正交变换近似完成,无需可学习参数。这对搜索/推荐系统中的 Transformer 排序模型有直接启示:在不显著损失效果的前提下,可将模型参数量和推理 FLOPs 降低 10-20%,特别适合对延迟敏感的在线 serving 场景。