Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Yash Raj Sarrof

dblp:378/0980 · DBLP profile ↗
← Back
3ranked-venue papers
1as first author
3since 2021 · last 2025
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 3 · 1 first-author · 3 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
3 papers
Deep learning architectures and training · 48% Language models and text generation · 38% Graph learning · 14%
Theoretical computer science
2 papers
Automata and formal languages · 100%

Topics — the 6 heaviest of 7, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Natural language and speech › Language models and text generation › compositional generalization
length generalization
1.722025
Born a Transformer - Always a Transformer? On the Effect of Pretraining on Architectural Abilities · NeurIPS 2025
A Formal Framework for Understanding Length Generalization in Transformers · ICLR 2025
Machine learning › Deep learning architectures and training
transformer
0.912025
A Formal Framework for Understanding Length Generalization in Transformers · ICLR 2025
Machine learning › Deep learning architectures and training › neural network expressivity
transformer expressivity
0.912025
Born a Transformer - Always a Transformer? On the Effect of Pretraining on Architectural Abilities · NeurIPS 2025
Machine learning › Graph learning › graph neural network
expressive power
0.812024
The Expressive Capacity of State Space Models: A Formal Language Perspective · NeurIPS 2024
Machine learning › Deep learning architectures and training
state space model
0.812024
The Expressive Capacity of State Space Models: A Formal Language Perspective · NeurIPS 2024
Natural language and speech › Language models and text generation › language modeling
language model architecture
0.212024
The Expressive Capacity of State Space Models: A Formal Language Perspective · NeurIPS 2024

Methods — techniques the papers use, named apart from their topics

norm-based regularizer · 1.7formal language analysis · 1.5empirical verification · 1.5retrieval and copying tasks · 0.9mechanistic interpretability · 0.9fine-tuning · 0.9causal transformers · 0.9causal transformer · 0.9absolute positional encodings · 0.9absolute positional encoding · 0.9
YearPublicationVenuePosition
2025 A Formal Framework for Understanding Length Generalization in Transformers
abstract
A major challenge for transformers is generalizing to sequences longer than those observed during training. While previous works have empirically shown that transformers can either succeed or fail at length generalization depending on the task, theoretical understanding of this phenomenon remains limited. In this work, we introduce a rigorous theoretical framework to analyze length generalization in causal transformers with learnable absolute positional encodings. In particular, we characterize those functions that are identifiable in the limit from sufficiently long inputs with absolute positional encodings under an idealized inference scheme using a norm-based regularizer. This enables us to prove the possibility of length generalization for a rich family of problems. We experimentally validate the theory as a predictor of success and failure of length generalization across a range of algorithmic and formal language tasks. Our theory not only explains a broad set of empirical observations but also opens the way to provably predicting length generalization capabilities in transformers.
Xinting Huang, Andy Yang, Satwik Bhattamishra, Yash Raj Sarrof, Andreas Krebs, Hattie Zhou, Preetum Nakkiran, Michael Hahn 0001
ICLR4
2025 Born a Transformer - Always a Transformer? On the Effect of Pretraining on Architectural Abilities
abstract
Transformers have theoretical limitations in modeling certain sequence-to-sequence tasks, yet it remains largely unclear if these limitations play a role in large-scale pretrained LLMs, or whether LLMs might effectively overcome these constraints in practice due to the scale of both the models themselves and their pretraining data. We explore how these architectural constraints manifest after pretraining by studying a family of *retrieval* and *copying* tasks inspired by Liu et al. [2024a]. We use a recently proposed framework for studying length generalization [Huang et al., 2025] to provide guarantees for each of our settings. Empirically, we observe an *induction-versus-anti-induction asymmetry*, where pretrained models are better at retrieving tokens to the right (induction) rather than the left (anti-induction) of a query token. This asymmetry disappears upon targeted fine-tuning if length-generalization is guaranteed by theory. Mechanistic analysis reveals that this asymmetry is connected to the differences in the strength of induction versus anti-induction circuits within pretrained transformers. We validate our findings through practical experiments on real-world tasks demonstrating reliability risks. Our results highlight that pretraining selectively enhances certain transformer capabilities, but does not overcome fundamental length-generalization limits.
Mayank Jobanputra, Yana Veitsman, Yash Raj Sarrof, Aleksandra Bakalova, Vera Demberg, Ellie Pavlick, Michael Hahn 0001
NeurIPS3
2024 The Expressive Capacity of State Space Models: A Formal Language Perspective
abstract
Recently, recurrent models based on linear state space models (SSMs) have shown promising performance in language modeling (LM), competititve with transformers. However, there is little understanding of the in-principle abilities of such models, which could provide useful guidance to the search for better LM architectures. We present a comprehensive theoretical study of the capacity of such SSMs as it compares to that of transformers and traditional RNNs. We find that SSMs and transformers have overlapping but distinct strengths. In star-free state tracking, SSMs implement straightforward and exact solutions to problems that transformers struggle to represent exactly. They can also model bounded hierarchical structure with optimal memory even without simulating a stack. On the other hand, we identify a design choice in current SSMs that limits their expressive power. We discuss implications for SSM and LM research, and verify results empirically on a recent SSM, Mamba.
Yash Raj Sarrof, Yana Veitsman, Michael Hahn 0001
NeurIPS1