Neil Band

I'm a fourth-year PhD student in the Computer Science Department at Stanford University. My advisors are Tatsunori Hashimoto and Tengyu Ma. My work is supported by the NSF GRFP, Quad, and Mercor Fellowships.

I was previously a Masters by Research student in the Oxford Applied and Theoretical Machine Learning Group (OATML) at the University of Oxford, where I was advised by Yarin Gal and supported by a Rhodes Scholarship. Before that, I did my A.B. in Computer Science and Economics at Harvard College. I worked with Stratos Idreos in the Harvard Data Systems Lab on systems machine learning, and Manolis Kellis in the MIT Computational Biology Group on machine learning in single-cell transcriptomics.

Google Scholar  /  Twitter  /  Github

profile photo
Research

I'm interested in data efficiency---learning more from limited data---in settings such as pretraining and continual learning. I work on data efficiency methods using synthetic data, alternative architectures and objectives such as latent variable modeling and meta-learning, and reinforcement learning.

Papers

For a full list, please see my Google Scholar.

Synthetic Data for any Differentiable Target
Tristan Thrush, Sung Min Park, Herman Brunborg, Luke Bailey, Marcel Roed, Neil Band, Christopher Potts, Tatsunori Hashimoto.
CoLM, 2026

paper / code

We optimize synthetic data for pretraining end-to-end against downstream task performance, by using metagradients to obtain fine-grained per-document rewards and running RL on those rewards.

Reasoning to Learn from Latent Thoughts
Yangjun Ruan, Neil Band, Chris Maddison, Tatsunori Hashimoto.
arXiv preprint, 2025

paper / code

We show that LMs can bootstrap their capabilities on task-agnostic data by inferring the latent thoughts that plausibly underlie pretraining documents.

Synthetic Continued Pretraining
Zitong Yang*, Neil Band*, Shuangping Li, Emmanuel Candès, Tatsunori Hashimoto.
ICLR, 2025 (Oral, 1.8% of submitted papers)

paper / code

A simple approach to learn knowledge from corpora as small as ~1M tokens, using graph-based synthetic data generation followed by continued pretraining.

Linguistic Calibration of Long-Form Generations
Neil Band, Xuechen Li, Tengyu Ma, Tatsunori Hashimoto.
ICML, 2024

paper / code

An alignment objective to train language models that express calibrated verbal statements of confidence in their long-form natural language generations.

Benchmarking Bayesian Deep Learning on Diabetic Retinopathy Detection Tasks
Neil Band*, Tim G. J. Rudner*, Qixuan Feng, Angelos Filos, Zachary Nado,
Michael W. Dusenberry, Ghassen Jerfel, Dustin Tran, Yarin Gal.
NeurIPS Datasets and Benchmarks Track, 2021
NeurIPS Workshop on Distribution Shifts, 2021 (Spotlight, 4.6% of accepted papers)

paper / poster / code

An easy-to-use, expert-guided, open-source suite of diabetic retinopathy detection benchmarking tasks for Bayesian deep learning.

Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning
Jannik Kossen*, Neil Band*, Clare Lyle, Aidan N. Gomez, Tom Rainforth, Yarin Gal.
NeurIPS, 2021

paper / poster / code / community video

A novel deep learning architecture that takes the entire dataset as input and learns to reason about relationships between datapoints using self-attention.


Based on Jon Barron's website (source code here).