利用软注意力模拟硬注意力
机器学习
2025-06-27 v2 计算与语言
形式语言与自动机理论
摘要
我们研究了使用软注意力的 transformer 在何种条件下能够模拟硬注意力,即有效地将注意力集中在某个位置子集上。首先,我们考察了硬注意力 transformer 所识别的若干语言子类,这些子类可以用线性时序逻辑的变体来定义。我们展示了软注意力 transformer 如何利用无界位置嵌入或温度缩放来计算这些逻辑的公式。其次,我们展示了温度缩放如何使 softmax transformer 模拟通用的硬注意力 transformer,所使用的温度取决于最大注意力得分与其他注意力得分之间的最小差距。
引用
@article{arxiv.2412.09925,
title = {Simulating Hard Attention Using Soft Attention},
author = {Andy Yang and Lena Strobl and David Chiang and Dana Angluin},
journal= {arXiv preprint arXiv:2412.09925},
year = {2025}
}
备注
19 pages