← Search

Jiaxi Hu

4 accepted papers

2026

MoM: Linear Sequence Modeling with Mixture-of-Memories

ICLR 2026poster

Linear sequence modeling methods, such as linear attention, state space modeling, and linear RNNs, offer significant efficiency improvements by reducing the complexity of training and inference. However, these methods typically compress the entire input sequence into a single fixed-size memory state…

Cited by 0SourcecodeScholar
2025

Improving Bilinear RNN with Closed-loop Control

NeurIPS 2025spotlight

Recent efficient sequence modeling methods, such as Gated DeltaNet, TTT, and RWKV-7, have achieved performance improvements by supervising the recurrent memory management through the Delta learning rule. Unlike previous state-space models (e.g., Mamba) and gated linear attentions (e.g., GLA), these…

Cited by 0SourceScholar
2025

Liger: Linearizing Large Language Models to Gated Recurrent Structures

ICML 2025poster

Transformers with linear recurrent modeling offer linear-time training and constant-memory inference. Despite their demonstrated efficiency and performance, pretraining such non-standard architectures from scratch remains costly and risky. The linearization of large language models (LLMs) transforms…

2024

Attractor Memory for Long-Term Time Series Forecasting: A Chaos Perspective

NeurIPS 2024poster

In long-term time series forecasting (LTSF) tasks, an increasing number of works have acknowledged that discrete time series originate from continuous dynamic systems and have attempted to model their underlying dynamics. Recognizing the chaotic nature of real-world data, our model, Attraos, incorpo…