← Search

Alliot Nagle

5 accepted papers

2025

Attention with Markov: A Curious Case of Single-layer Transformers

ICLR 2025spotlight

Attention-based transformers have achieved tremendous success across a variety of disciplines including natural languages. To deepen our understanding of their sequential modeling capabilities, there is a growing interest in using Markov input processes to study them. A key finding is that when trai…

Cited by 3SourcePDFScholar
2024

Fundamental Limits of Prompt Compression: A Rate-Distortion Framework for Black-Box Language Models

NeurIPS 2024poster

We formalize the problem of prompt compression for large language models (LLMs) and present a framework to unify token-level prompt compression methods which create hard prompts for black-box models. We derive the distortion-rate function for this setup as a linear program, and provide an efficient…

Cited by 4SourcePDFScholar
2024

Local to Global: Learning Dynamics and Effect of Initialization for Transformers

NeurIPS 2024poster

In recent years, transformer-based models have revolutionized deep learning, particularly in sequence modeling. To better understand this phenomenon, there is a growing interest in using Markov input processes to study transformers. However, our current understanding in this regard remains limited w…

2022

Rare Gems: Finding Lottery Tickets at Initialization

NeurIPS 2022accept

Large neural networks can be pruned to a small fraction of their original size, with little loss in accuracy, by following a time-consuming "train, prune, re-train" approach. Frankle & Carbin conjecture that we can avoid this by training lottery tickets, i.e., special sparse subnetworks found at ini…

2020

Optimal Lottery Tickets via Subset Sum: Logarithmic Over-Parameterization is Sufficient

NeurIPS 2020spotlight

The strong lottery ticket hypothesis (LTH) postulates that one can approximate any target neural network by only pruning the weights of a sufficiently over-parameterized random network. A recent work by Malach et al. [MYSS20] establishes the first theoretical analysis for the strong LTH: one can pr…