Saved in:
Bibliographic Details
Main Authors: Zhang, Mingtao, Yang, Guoli, Zhu, Zhanxing, Wang, Mengzhu, Bai, Xiaoying
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.03190
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908629859303424
author Zhang, Mingtao
Yang, Guoli
Zhu, Zhanxing
Wang, Mengzhu
Bai, Xiaoying
author_facet Zhang, Mingtao
Yang, Guoli
Zhu, Zhanxing
Wang, Mengzhu
Bai, Xiaoying
contents Attention mechanisms have been extensively employed in various applications, including time series modeling, owing to their capacity to capture intricate dependencies; however, their utility is often constrained by quadratic computational complexity, which impedes scalability for long sequences. In this work, we propose a novel linear attention mechanism designed to overcome these limitations. Our approach is grounded in a theoretical demonstration that entropy, as a strictly concave function on the probability simplex, implies that distributions with aligned probability rankings and similar entropy values exhibit structural resemblance. Building on this insight, we develop an efficient approximation algorithm that computes the entropy of dot-product-derived distributions with only linear complexity, enabling the implementation of a linear attention mechanism based on entropy equality. Through rigorous analysis, we reveal that the effectiveness of attention in spatio-temporal time series modeling may not primarily stem from the non-linearity of softmax but rather from the attainment of a moderate and well-balanced weight distribution. Extensive experiments on four spatio-temporal datasets validate our method, demonstrating competitive or superior forecasting performance while achieving substantial reductions in both memory usage and computational time.
format Preprint
id arxiv_https___arxiv_org_abs_2511_03190
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Efficient Linear Attention for Multivariate Time Series Modeling via Entropy Equality
Zhang, Mingtao
Yang, Guoli
Zhu, Zhanxing
Wang, Mengzhu
Bai, Xiaoying
Machine Learning
Artificial Intelligence
Attention mechanisms have been extensively employed in various applications, including time series modeling, owing to their capacity to capture intricate dependencies; however, their utility is often constrained by quadratic computational complexity, which impedes scalability for long sequences. In this work, we propose a novel linear attention mechanism designed to overcome these limitations. Our approach is grounded in a theoretical demonstration that entropy, as a strictly concave function on the probability simplex, implies that distributions with aligned probability rankings and similar entropy values exhibit structural resemblance. Building on this insight, we develop an efficient approximation algorithm that computes the entropy of dot-product-derived distributions with only linear complexity, enabling the implementation of a linear attention mechanism based on entropy equality. Through rigorous analysis, we reveal that the effectiveness of attention in spatio-temporal time series modeling may not primarily stem from the non-linearity of softmax but rather from the attainment of a moderate and well-balanced weight distribution. Extensive experiments on four spatio-temporal datasets validate our method, demonstrating competitive or superior forecasting performance while achieving substantial reductions in both memory usage and computational time.
title Efficient Linear Attention for Multivariate Time Series Modeling via Entropy Equality
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2511.03190