Spatio-temporal forecasting plays a vital role in various domains such as transportation, meteorology, and urban management. However, existing approaches often struggle to model complex spatio-temporal dependencies effectively while maintaining computational efficiency, particularly when handling full spatio-temporal sequences. The quadratic time-space complexity of the traditional self-attention mechanism significantly limits its scalability to long-sequence tasks. To address these challenges, we propose LOSTFormer (Linear Orthogonal Spatio-Temporal Transformer) for efficient and accurate spatio-temporal forecasting. LOSTFormer constructs hierarchical embeddings across temporal, spatial, and joint spatio-temporal dimensions to enhance representation learning. Linear attention framework, a novel Learnable Orthogonal Rotation Attention (Lor-Attention) is introduced, which employs the Cayley transform to learn orthogonal rotation matrices and adaptively refine the mapping of Positive Orthogonal Random Features. Our approach can achieve linear computational complexity and effectively capture long-range and cross-dimensional dependencies in spatio-temporal data. Extensive experiments on spatio-temporal datasets demonstrate that LOSTFormer consistently outperforms state-of-the-art methods in prediction accuracy while maintaining efficiency. The code can be found on our GitHub repository( https://github.com/QAQYYC/LOSTFormer .).

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

LOSTFormer: Linear Orthogonal Spatio-Temporal Transformer with Learnable Rotation

  • Yechen Yu,
  • Liang Xie,
  • Jiankai Zheng,
  • Peilin Tan

摘要

Spatio-temporal forecasting plays a vital role in various domains such as transportation, meteorology, and urban management. However, existing approaches often struggle to model complex spatio-temporal dependencies effectively while maintaining computational efficiency, particularly when handling full spatio-temporal sequences. The quadratic time-space complexity of the traditional self-attention mechanism significantly limits its scalability to long-sequence tasks. To address these challenges, we propose LOSTFormer (Linear Orthogonal Spatio-Temporal Transformer) for efficient and accurate spatio-temporal forecasting. LOSTFormer constructs hierarchical embeddings across temporal, spatial, and joint spatio-temporal dimensions to enhance representation learning. Linear attention framework, a novel Learnable Orthogonal Rotation Attention (Lor-Attention) is introduced, which employs the Cayley transform to learn orthogonal rotation matrices and adaptively refine the mapping of Positive Orthogonal Random Features. Our approach can achieve linear computational complexity and effectively capture long-range and cross-dimensional dependencies in spatio-temporal data. Extensive experiments on spatio-temporal datasets demonstrate that LOSTFormer consistently outperforms state-of-the-art methods in prediction accuracy while maintaining efficiency. The code can be found on our GitHub repository( https://github.com/QAQYYC/LOSTFormer .).