← Search

Mohammad Aflah Khan

7 accepted papers

2026

Hubble: a Model Suite to Advance the Study of LLM Memorization

ICLR 2026oral

We present Hubble, a suite of open-source large language models (LLMs) for the scientific study of LLM memorization. Hubble models come as minimal pairs: standard models are pretrained on a large English corpus, and perturbed models are trained in the same way but with controlled insertion of text (…

Cited by 0SourcecodeScholar
2026

In Agents We Trust, but Who Do Agents Trust? Latent Preferences Steer LLM Generations

ICLR 2026poster

Large Language Model (LLM) based agents are increasingly being deployed as user-friendly front-ends on online platforms, where they filter, prioritize, and recommend information retrieved from the platforms' back-end databases or via web search. In these scenarios, LLM agents act as decision assista…

Cited by 0SourcecodeScholar
2026

Position: Don't Just "Fix it in Post'': A Science of AI Must Study Learning Dynamics

ICML 2026oral

What would it mean to have a *scientific* understanding of AI? Language models are not static objects—they are snapshots of time-evolving processes shaped by data, objectives, and optimization dynamics. Yet the field predominantly treats models as fixed artifacts, analyzing behaviors after training …

Cited by 0SourceScholar
2026

Rote Learning Considered Useful: Generalizing over Memorized Data in LLMs

ICLR 2026poster

Rote learning is a memorization technique based on repetition. Many researchers argue that rote learning hinders generalization because it encourages verbatim memorization rather than deeper understanding. This concern extends even to factual knowledge, which inevitably requires a certain degree of…

Cited by 0SourcecodeScholar
2025

QUENCH: Measuring the gap between Indic and Non-Indic Contextual General Reasoning in LLMs

COLING 2025main

The rise of large language models (LLMs) has created a need for advanced benchmarking systems beyond traditional setups. To this end, we introduce QUENCH, a novel text-based English Quizzing Benchmark manually curated and transcribed from YouTube quiz videos. QUENCH possesses masked entities and rat…

2025

Recite, Reconstruct, Recollect: Memorization in LMs as a Multifaceted Phenomenon

ICLR 2025poster

Memorization in language models is typically treated as a homogenous phenomenon, neglecting the specifics of the memorized data. We instead model memorization as the effect of a set of complex factors that describe each sample and relate it to the model and corpus. To build intuition around these fa…

2023

Pythia: A Suite for Analyzing Large Language Models Across Training and Scaling

ICML 2023oral

How do large language models (LLMs) develop and evolve over the course of training? How do these patterns change as models scale? To answer these questions, we introduce *Pythia*, a suite of 16 LLMs all trained on public data seen in the exact same order and ranging in size from 70M to 12B parameter…