Publications

WAVE: Weighted Autoregressive Varying Gate for Time Series Forecasting

ICML 2025 PMLR 267: 40464–40490

Jiecheng Lu, Xu Han, Yan Sun, Shihao Yang

Adds ARMA structure to autoregressive attention via a weighted varying gate, decoupling long‑range and local effects and improving TSF quality without increasing asymptotic complexity.

Demonstrates that properly tokenized autoregressive decoder‑only Transformers are strong TSF baselines and that introducing AR and MA components through WAVE yields consistent accuracy gains across efficient attentions.