← Search

Ali Behrouz

10 accepted papers

2026

ATLAS: Learning to Optimally Memorize the Context at Test Time

ICML 2026poster

Transformers have been established as the most popular backbones in sequence modeling, mainly due to their effectiveness in in-context retrieval tasks and the ability to learn at scale. Their quadratic memory and time complexity, however, bound their applicability in longer sequences and so has moti…

Cited by 0SourceScholar
2026

It's All Connected: A Journey Through Test-Time Memorization, Attentional Bias, Retention, and Online Optimization

ICLR 2026poster

Designing efficient and effective architectural backbones has been in the core of research efforts to enhance the capability of foundation models. Inspired by the human cognitive phenomenon of attentional bias—the natural tendency to prioritize certain events or stimuli—we reconceptualize neural arc…

Cited by 0SourceScholar
2026

Memory Caching: RNNs with Growing Memory

ICML 2026poster

Transformers have been established as the de-facto backbones for most recent advances in sequence modeling, mainly due to their growing memory capacity that scales with the context length. While plausible for retrieval tasks, it causes quadratic complexity and so has motivated recent studies to expl…

Cited by 0SourceScholar
2026

TNT: Improving Chunkwise Training for Test-Time Memorization

ICLR 2026poster

Recurrent neural networks (RNNs) with deep test-time memorization modules, such as Titans and TTT, represent a promising, linearly-scaling paradigm distinct from Transformers. While these expressive models do not yet match the peak performance of state-of-the-art Transformers, their potential has be…

Cited by 0SourceScholar
2025

Best of Both Worlds: Advantages of Hybrid Graph Sequence Models

ICML 2025poster

Modern sequence models (e.g., Transformers and linear RNNs) emerged as dominant backbones of recent deep learning frameworks, mainly due to their efficiency, representational power, and/or ability to capture long-range dependencies. Recently, adopting these sequence models for graph-structured data…

Cited by 2SourcePDFScholar
2025

Nested Learning: The Illusion of Deep Learning Architectures

NeurIPS 2025poster

Over the last decades, developing more powerful neural architectures and simultaneously designing optimization algorithms to effectively train them have been the core of research efforts to enhance the capability of machine learning models. Despite the recent progresses, particularly in developing L…

Cited by 0SourceScholar
2024

Chimera: Effectively Modeling Multivariate Time Series with 2-Dimensional State Space Models

NeurIPS 2024poster

Modeling multivariate time series is a well-established problem with a wide range of applications from healthcare to financial markets. It, however, is challenging as it requires methods to (1) have high expressive power of representing complicated dependencies along the time axis to capture both lo…

Cited by 6SourcePDFScholar
2024

Unsupervised Representation Learning of Brain Activity via Bridging Voxel Activity and Functional Connectivity

ICML 2024poster

Effective brain representation learning is a key step toward the understanding of cognitive processes and diagnosis of neurological diseases/disorders. Existing studies have focused on either (1) voxel-level activity, where only a single weight relating the voxel activity to the task (i.e., aggregat…

Cited by 7SourcePDFScholar
2023

CAT-Walk: Inductive Hypergraph Learning via Set Walks

NeurIPS 2023poster

Temporal hypergraphs provide a powerful paradigm for modeling time-dependent, higher-order interactions in complex systems. Representation learning for hypergraphs is essential for extracting patterns of the higher-order interactions that are critically important in real-world problems in social net…