← Search

Armin W Thomas

8 accepted papers

2025

Quantifying Memory Utilization with Effective State-Size

ICML 2025poster

As the space of causal sequence modeling architectures continues to grow, the need to develop a general framework for their analysis becomes increasingly important. With this aim, we draw insights from classical signal processing and control theory, to develop a quantitative measure of *memory utili…

Cited by 0SourcePDFScholar
2025

STAR: Synthesis of Tailored Architectures

ICLR 2025oral

Iterative improvement of model architectures is fundamental to deep learning: Transformers first enabled scaling, and recent advances in model hybridization have pushed the quality-efficiency frontier. However, optimizing architectures remains challenging and expensive, with a variety of automated o…

Cited by 2SourcePDFScholar
2024

Mechanistic Design and Scaling of Hybrid Architectures

ICML 2024poster

The development of deep learning architectures is a resource-demanding process, due to a vast design space, long prototyping times, and high compute costs associated with at-scale model training and evaluation. We set out to simplify this process by grounding it in an end-to-end mechanistic architec…

2023

Hungry Hungry Hippos: Towards Language Modeling with State Space Models

ICLR 2023top-25%

State space models (SSMs) have demonstrated state-of-the-art sequence modeling performance in some modalities, but underperform attention in language modeling. Moreover, despite scaling nearly linearly in sequence length instead of quadratically, SSMs are still slower than Transformers due to poor h…

2023

HyenaDNA: Long-Range Genomic Sequence Modeling at Single Nucleotide Resolution

NeurIPS 2023spotlight

Genomic (DNA) sequences encode an enormous amount of information for gene regulation and protein synthesis. Similar to natural language models, researchers have proposed foundation models in genomics to learn generalizable features from unlabeled genome data that can then be fine-tuned for downstrea…

2023

Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture

NeurIPS 2023oral

Machine learning models are increasingly being scaled in both sequence length and model dimension to reach longer contexts and better performance. However, existing architectures such as Transformers scale quadratically along both these axes. We ask: are there performant architectures that can scale…

Cited by 59SourcePDFScholar
2023

Simple Hardware-Efficient Long Convolutions for Sequence Modeling

ICML 2023poster

State space models (SSMs) have high performance on long sequence modeling but require sophisticated initialization techniques and specialized implementations for high quality and runtime performance. We study whether a simple alternative can match SSMs in performance and efficiency: directly learnin…

2022

Self-Supervised Learning of Brain Dynamics from Broad Neuroimaging Data

NeurIPS 2022accept

Self-supervised learning techniques are celebrating immense success in natural language processing (NLP) by enabling models to learn from broad language data at unprecedented scales. Here, we aim to leverage the success of these techniques for mental state decoding, where researchers aim to identify…