2 个月前

持续变换器:无冗余注意力机制用于在线推理

Lukas Hedegaard; Arian Bakhtiarnia; Alexandros Iosifidis
持续变换器:无冗余注意力机制用于在线推理
摘要

常见的Transformer架构本质上受限于只能对整个标记序列进行操作,而不能逐个处理标记。因此,在时间序列数据的在线推理过程中,由于连续标记序列之间的重叠,其使用会导致相当大的冗余。在本研究中,我们提出了缩放点积注意力机制(Scaled Dot-Product Attention)的新公式,使得Transformer能够在持续输入流上高效地进行逐个标记的在线推理。重要的是,我们的修改仅涉及计算顺序,而输出结果和学习到的权重与原始Transformer编码器完全相同。我们通过在THUMOS14、TVSeries和GTZAN数据集上的实验验证了所提出的持续Transformer编码器,并取得了显著的结果:我们的单块和双块持续架构分别将每次预测所需的浮点运算量减少了最多63倍和2.6倍,同时保持了预测性能。