Interpretable LSTMs For Whole-Brain Neuroimaging Analyses. - Armin W. Thomas; Hauke R. Heekeren; Klaus-Robert Müller; Wojciech Samek (CoRR, 2018)

Deep Transfer Learning for Whole-Brain FMRI Analyses. - Armin W. Thomas; Klaus-Robert Müller; Wojciech Samek (OR/MLCN@MICCAI, 2019)

Challenges for cognitive decoding using deep learning methods. - Armin W. Thomas; Christopher Ré; Russell A. Poldrack (CoRR, 2021)

Deep Learning for Whole-Brain Cognitive Decoding. - Klaus-Robert Müller; Armin W. Thomas; Wojciech Samek (BCI, 2022)

Differentiable programming for functional connectomics. - Rastko Ciric; Armin W. Thomas; Oscar Esteban; Russell A. Poldrack (ML4H@NeurIPS, 2022)

Self-Supervised Learning of Brain Dynamics from Broad Neuroimaging Data. - Armin W. Thomas; Christopher Ré; Russell A. Poldrack (NeurIPS, 2022)

Comparing interpretation methods in mental state decoding analyses with deep learning models. - Armin W. Thomas; Christopher Ré; Russell A. Poldrack (CoRR, 2022)

Gaze-dependent evidence accumulation predicts multi-alternative risky choice behaviour. - Felix Molter; Armin W. Thomas; Scott A. Huettel; Hauke R. Heekeren; Peter N. C. Mohr (PLoS Comput. Biol., 2022)

Hungry Hungry Hippos: Towards Language Modeling with State Space Models. - Daniel Y. Fu; Tri Dao; Khaled Kamal Saab; Armin W. Thomas; Atri Rudra; Christopher Ré (ICLR, 2023)

Simple Hardware-Efficient Long Convolutions for Sequence Modeling. - Daniel Y. Fu; Elliot L. Epstein; Eric Nguyen; Armin W. Thomas; Michael Zhang; Tri Dao; Atri Rudra; Christopher Ré (ICML, 2023)

Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture. - Daniel Y. Fu; Simran Arora; Jessica Grogan; Isys Johnson; Evan Sabri Eyuboglu; Armin W. Thomas; Benjamin Spector; Michael Poli; Atri Rudra; Christopher Ré (NeurIPS, 2023)

HyenaDNA: Long-Range Genomic Sequence Modeling at Single Nucleotide Resolution. - Eric Nguyen; Michael Poli; Marjan Faizi; Armin W. Thomas; Michael Wornow; Callum Birch-Sykes; Stefano Massaroli; Aman Patel; Clayton M. Rabideau; Yoshua Bengio; Stefano Ermon; Christopher Ré; et al. (NeurIPS, 2023)

Benchmarking explanation methods for mental state decoding with deep learning models. - Armin W. Thomas; Christopher Ré; Russell A. Poldrack (NeuroImage, 2023)

Evaluating deep transfer learning for whole-brain cognitive decoding. - Armin W. Thomas; Ulman Lindenberger; Wojciech Samek; Klaus-Robert Müller (J. Frankl. Inst., 2023)

Mechanistic Design and Scaling of Hybrid Architectures. - Michael Poli; Armin W. Thomas; Eric Nguyen; Pragaash Ponnusamy; Björn Deiseroth; Kristian Kersting; Taiji Suzuki; Brian L. Hie; Stefano Ermon; Christopher Ré; Ce Zhang 001; Stefano Massaroli (ICML, 2024)

STAR: Synthesis of Tailored Architectures. - Armin W. Thomas; Rom N. Parnichkun; Alexander Amini; Stefano Massaroli; Michael Poli (ICLR, 2025)

Quantifying Memory Utilization with Effective State-Size. - Rom N. Parnichkun; Neehal Tumma; Armin W. Thomas; Alessandro Moro; Qi An; Taiji Suzuki; Atsushi Yamashita; Michael Poli; Stefano Massaroli (CoRR, 2025)