arXiv:2505.08516cs.LGcs.AI2025-05IJCAI被引 6

通过奇异值域学习高级图滤波器,提升线性Transformer性能

Learning Advanced Self-Attention for Linear Transformers in the Singular Value Domain

  • 从图信号处理视角,将自注意力建模为奇异值域的图滤波器
  • 在长序列任务和时间序列分类上达到当前最优表现
  • 保持线性复杂度,适合长序列建模场景

Transformers 在多个领域表现出色,其核心是自注意力机制,用于学习输入序列中任意两个标记之间的关系。近期研究指出,自注意力可被理解为图的归一化邻接矩阵。从图信号处理(GSP)角度,自注意力等价于一个图滤波器,以值向量作为信号。然而,现有自注意力仅使用一阶多项式矩阵定义,表现为低通滤波器,难以有效利用多频段信息,导致设计过于简化。为此,本文提出一种新方法AGF,从有向图的图信号处理视角,将自注意力解释为在奇异值域学习图滤波器,实现与输入长度 n 线性相关的复杂度 $/mathcal{O}(nd^2)$。实验表明,AGF 在 Long Range Arena 基准和时间序列分类任务中均取得当前最优性能。

原文摘要 · Abstract (English)

Transformers have demonstrated remarkable performance across diverse domains. The key component of Transformers is self-attention, which learns the relationship between any two tokens in the input sequence. Recent studies have revealed that the self-attention can be understood as a normalized adjacency matrix of a graph. Notably, from the perspective of graph signal processing (GSP), the self-attention can be equivalently defined as a simple graph filter, applying GSP using the value vector as the signal. However, the self-attention is a graph filter defined with only the first order of the polynomial matrix, and acts as a low-pass filter preventing the effective leverage of various frequency information. Consequently, existing self-attention mechanisms are designed in a rather simplified manner. Therefore, we propose a novel method, called \underline{\textbf{A}}ttentive \underline{\textbf{G}}raph \underline{\textbf{F}}ilter (AGF), interpreting the self-attention as learning the graph filter in the singular value domain from the perspective of graph signal processing for directed graphs with the linear complexity w.r.t. the input length $n$, i.e., $\mathcal{O}(nd^2)$. In our experiments, we demonstrate that AGF achieves state-of-the-art performance on various tasks, including Long Range Arena benchmark and time series classification.

自注意力图神经网络线性Transformer奇异值

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。