|
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.
|
|
|
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.
|
|