稀疏注意力实现:用 Indexer 蒸馏完整注意力分数

nrehiew_ · x · 2026-08-27

介绍了带 Indexer 的稀疏注意力机制:在块级别使用压缩潜在空间中的 Indexer。训练时,要求 Indexer 预测完整的注意力分数:1. 将密集注意力分数蒸馏到 Indexer 中;2. 计算 Indexer 与完整注意力教师模型之间的 KL 损失。

原文链接 →

「研究」频道最新

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