← Search

Akshita Bhagia

10 accepted papers

2025

DataDecide: How to Predict Best Pretraining Data with Small Experiments

ICML 2025poster

Because large language models are expensive to pretrain on different datasets, using smaller-scale experiments to decide on data is crucial for reducing costs. Which benchmarks and methods of making decisions from observed performance at small scale most accurately predict the datasets that yield th…

Cited by 0SourcePDFScholar
2025

FlexOLMo: Open Language Models for Flexible Data Use

NeurIPS 2025spotlight

We introduce FlexOLMo, a new class of language models (LMs) that supports (1) distributed training without data sharing, where different model parameters are independently trained on private datasets, and (2) data-flexible inference, where these parameters along with their associated data can be eas…

Cited by 0SourceScholar
2025

OLMoE: Open Mixture-of-Experts Language Models

ICLR 2025oral

We introduce OLMoE, a fully open, state-of-the-art language model leveraging sparse Mixture-of-Experts (MoE). OLMoE-1B-7B has 7 billion (B) parameters but uses only 1B per input token. We pretrain it on 5 trillion tokens and further adapt it to create OLMoE-1B-7B-Instruct. Our models outperform all…

2024

Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Research

ACL 2024long

Information about pretraining corpora used to train the current best-performing language models is seldom discussed: commercial models rarely detail their data, and even open models are often released without accompanying training data or recipes to reproduce them. As a result, it is challenging to…

2024

OLMo: Accelerating the Science of Language Models

ACL 2024long

Language models (LMs) have become ubiquitous in both NLP research and in commercial product offerings. As their commercial importance has surged, the most powerful models have become closed off, gated behind proprietary interfaces, with important details of their training data, architectures, and de…

2024

Paloma: A Benchmark for Evaluating Language Model Fit

NeurIPS 2024poster

Evaluations of language models (LMs) commonly report perplexity on monolithic data held out from training. Implicitly or explicitly, this data is composed of domains—varying distributions of language. We introduce Perplexity Analysis for Language Model Assessment (Paloma), a benchmark to measure LM…

Cited by 7SourcePDFScholar
2024

What's In My Big Data?

ICLR 2024spotlight

Large text corpora are the backbone of language models. However, we have a limited understanding of the content of these corpora, including general statistics, quality, social factors, and inclusion of evaluation data (contamination). In this work, we propose What's In My Big Data? (WIMBD), a platfo…

2023

HINT: Hypernetwork Instruction Tuning for Efficient Zero- and Few-Shot Generalisation

ACL 2023long

Recent NLP models have shown the remarkable ability to effectively generalise ‘zero-shot’ to new tasks using only natural language instructions as guidance. However, many of these approaches suffer from high computational costs due to their reliance on concatenating lengthy instructions with every i…

2022

Continued Pretraining for Better Zero- and Few-Shot Promptability

EMNLP 2022main

Recently introduced language model prompting methods can achieve high accuracy in zero- and few-shot settings while requiring few to no learned task-specific parameters. Nevertheless, these methods still often trail behind full model finetuning. In this work, we investigate if a dedicated continued…

2022

On Advances in Text Generation from Images Beyond Captioning: A Case Study in Self-Rationalization

EMNLP 2022finding

Combining the visual modality with pretrained language models has been surprisingly effective for simple descriptive tasks such as image captioning. More general text generation however remains elusive. We take a step back and ask: How do these models work for more complex generative tasks, i.e. con…

Cited by 3SourcePDFScholar