← Search

Michael Munn

5 accepted papers

2026

Equivalence of Context and Parameter Updates in Modern Transformer Blocks

ICML 2026oral

Recent research has established that the impact of context in a vanilla transformer can be represented implicitly by forming a token-dependent, rank-1 patch to its MLP weights. This work extends that foundational theory to the diverse architectures of modern Large Language Models. We first demonstra…

Cited by 0SourceScholar
2024

The Impact of Geometric Complexity on Neural Collapse in Transfer Learning

NeurIPS 2024poster

Many of the recent advances in computer vision and language models can be attributed to the success of transfer learning via the pre-training of large foundation models. However, a theoretical framework which explains this empirical success is incomplete and remains an active area of research. Flatn…

Cited by 0SourcePDFScholar
2022

Why neural networks find simple solutions: The many regularizers of geometric complexity

NeurIPS 2022accept

In many contexts, simpler models are preferable to more complex models and the control of this model complexity is the goal for many methods in machine learning such as regularization, hyperparameter tuning and architecture design. In deep learning, it has been difficult to understand the underlying…

Cited by 39SourcePDFScholar
2020

COT-GAN: Generating Sequential Data via Causal Optimal Transport

NeurIPS 2020poster

We introduce COT-GAN, an adversarial algorithm to train implicit generative models optimized for producing sequential data. The loss function of this algorithm is formulated using ideas from Causal Optimal Transport (COT), which combines classic optimal transport methods with an additional temporal…