技术讨论:在潜空间使用索引器实现稀疏注意力

SkyLi0n · x · 2026-08-28

SkyLi0n 讨论了利用块稀疏注意力掩码的优势,并分享了在压缩潜空间中使用块级索引器来实现稀疏注意力的方法。训练过程中,索引器被用来预测完整的注意力分数,具体做法包括将稠密注意力分数蒸馏到索引器中,并使用 KL 散度损失来优化索引器与完整注意力教师模型的一致性。

所属事件:用潜空间 Indexer 蒸馏实现稀疏注意力(2 条相关)→

原文链接 →

「研究」频道最新

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