← Search

Jiecheng Lu

9 accepted papers

2026

Free Energy Mixer

ICLR 2026poster

Standard attention stores keys/values losslessly but reads them via a per-head convex average, blocking channel-wise selection. We propose the Free Energy Mixer (FEM): a free-energy (log-sum-exp) read that applies a value-driven, per-channel log-linear tilt to a fast prior (e.g., from queries/keys i…

Cited by 0SourcecodeScholar
2026

StretchTime: Adaptive Time Series Forecasting via Symplectic Attention

ICML 2026poster

Transformer architectures have established strong baselines in time series forecasting, yet they typically rely on positional encodings that assume uniform, index-based temporal progression. However, real-world systems, from shifting financial cycles to elastic biological rhythms, frequently exhibit…

Cited by 0SourceScholar
2025

Linear Transformers as VAR Models: Aligning Autoregressive Attention Mechanisms with Autoregressive Forecasting

ICML 2025poster

Autoregressive attention-based time series forecasting (TSF) has drawn increasing interest, with mechanisms like linear attention often outperforming vanilla attention. However, deeper Transformer architectures frequently misalign with autoregressive objectives, obscuring the underlying VAR structur…

2025

ZeroS: Zero‑Sum Linear Attention for Efficient Transformers

NeurIPS 2025spotlight

Linear attention methods offer Transformers $O(N)$ complexity but typically underperform standard softmax attention. We identify two fundamental limitations affecting these approaches: the restriction to convex combinations that only permits additive information blending, and uniform accumulated wei…

Cited by 0SourceScholar
2024

ARM: Refining Multivariate Forecasting with Adaptive Temporal-Contextual Learning

ICLR 2024poster

Long-term time series forecasting (LTSF) is important for various domains but is confronted by challenges in handling the complex temporal-contextual relationships. As multivariate input models underperforming some recent univariate counterparts, we posit that the issue lies in the inefficiency of e…

Cited by 5SourcePDFScholar
2024

CATS: Enhancing Multivariate Time Series Forecasting by Constructing Auxiliary Time Series as Exogenous Variables

ICML 2024poster

For Multivariate Time Series Forecasting (MTSF), recent deep learning applications show that univariate models frequently outperform multivariate ones. To address the deficiency in multivariate models, we introduce a method to Construct Auxiliary Time Series (CATS) that functions like a 2D temporal-…

Cited by 22SourcePDFScholar