← Search

Tristan Thrush

16 accepted papers

2025

Improving Pretraining Data Using Perplexity Correlations

ICLR 2025poster

Quality pretraining data is often seen as the key to high-performance language models. However, progress in understanding pretraining data has been slow due to the costly pretraining runs required for data selection experiments. We present a framework that avoids these costs and selects high-quality…

2025

MixMin: Finding Data Mixtures via Convex Minimization

ICML 2025poster

Modern machine learning pipelines are increasingly combining and mixing data from diverse and disparate sources, e.g., pre-training large language models. Yet, finding the optimal data mixture is a challenging and open problem. We formalize this data mixing problem as a bi-level objective: the best…

Cited by 0SourcePDFScholar
2024

ColorSwap: A Color and Word Order Dataset for Multimodal Evaluation

ACL 2024findings

This paper introduces the ColorSwap dataset, designed to assess and improve the proficiency of multimodal models in matching objects with their colors. The dataset is comprised of 2,000 unique image-caption pairs, grouped into 1,000 examples. Each example includes a caption-image pair, along with a…

2024

I am a Strange Dataset: Metalinguistic Tests for Language Models

ACL 2024long

Statements involving metalinguistic self-reference (“This paper has six sections.”) are prevalent in many domains. Can large language models (LLMs) handle such language? In this paper, we present “I am a Strange Dataset”, a new dataset for addressing this question. There are two subtasks: generation…

2024

Nearest Neighbor Normalization Improves Multimodal Retrieval

EMNLP 2024main

Multimodal models leverage large-scale pretraining to achieve strong but still imperfect performance on tasks such as image captioning, visual question answering, and cross-modal retrieval. In this paper, we present a simple and efficient method for correcting errors in trained contrastive image-tex…

2023

DataPerf: Benchmarks for Data-Centric AI Development

NeurIPS 2023poster

Machine learning research has long focused on models rather than datasets, and prominent datasets are used for common ML tasks without regard to the breadth, difficulty, and faithfulness of the underlying problems. Neglecting the fundamental importance of data has given rise to inaccuracy, bias, and…

2022

Hatemoji: A Test Suite and Adversarially-Generated Dataset for Benchmarking and Detecting Emoji-Based Hate

NAACL 2022long

Detecting online hate is a complex task, and low-performing models have harmful consequences when used for sensitive applications such as content moderation. Emoji-based hate is an emerging challenge for automated detection. We present HatemojiCheck, a test suite of 3,930 short-form statements that…

2022

Models in the Loop: Aiding Crowdworkers with Generative Annotation Assistants

NAACL 2022long

In Dynamic Adversarial Data Collection (DADC), human annotators are tasked with finding examples that models struggle to predict correctly. Models trained on DADC-collected training data have been shown to be more robust in adversarial and out-of-domain settings, and are considerably harder for huma…

2022

The BigScience ROOTS Corpus: A 1.6TB Composite Multilingual Dataset

NeurIPS 2022accept

As language models grow ever larger, the need for large-scale high-quality text datasets has never been more pressing, especially in multilingual settings. The BigScience workshop, a 1-year international and multidisciplinary initiative, was formed with the goal of researching and training large lan…

Cited by 214SourcePDFScholar
2022

Winoground: Probing Vision and Language Models for Visio-Linguistic Compositionality

CVPR 2022poster

We present a novel task and dataset for evaluating the ability of vision and language models to conduct visio-linguistic compositional reasoning, which we call Winoground. Given two images and two captions, the goal is to match them correctly--but crucially, both captions contain a completely identi…

Cited by 440PDFScholar
2021

Dynabench: Rethinking Benchmarking in NLP

NAACL 2021long

We introduce Dynabench, an open-source platform for dynamic dataset creation and model benchmarking. Dynabench runs in a web browser and supports human-and-model-in-the-loop dataset creation: annotators seek to create examples that a target model will misclassify, but that another person will not. I…

Cited by 471SourcePDFScholar
2021

Dynaboard: An Evaluation-As-A-Service Platform for Holistic Next-Generation Benchmarking

NeurIPS 2021poster

We introduce Dynaboard, an evaluation-as-a-service framework for hosting benchmarks and conducting holistic model comparison, integrated with the Dynabench platform. Our platform evaluates NLP models directly instead of relying on self-reported metrics or predictions on a single dataset. Under this…

Cited by 66SourcePDFScholar
2021

Human-Adversarial Visual Question Answering

NeurIPS 2021poster

Performance on the most commonly used Visual Question Answering dataset (VQA v2) is starting to approach human accuracy. However, in interacting with state-of-the-art VQA models, it is clear that the problem is far from being solved. In order to stress test VQA models, we benchmark them against huma…

Cited by 67SourcePDFScholar
2021

Improving Question Answering Model Robustness with Synthetic Adversarial Data Generation

EMNLP 2021main

Despite recent progress, state-of-the-art question answering models remain vulnerable to a variety of adversarial attacks. While dynamic adversarial data collection, in which a human annotator tries to write examples that fool a model-in-the-loop, can improve model robustness, this process is expens…

2021

Learning from the Worst: Dynamically Generated Datasets to Improve Online Hate Detection

ACL 2021long

We present a human-and-model-in-the-loop process for dynamically generating datasets and training better performing and more robust hate detection models. We provide a new dataset of 40,000 entries, generated and labelled by trained annotators over four rounds of dynamic data creation. It includes 1…

2021

Rover Relocalization for Mars Sample Return by Virtual Template Synthesis and Matching

RA-L 2021

We consider the problem of rover relocalization in the context of the notional Mars Sample Return campaign. In this campaign, a rover (R1) needs to be capable of autonomously navigating and localizing itself within an area of approximately 50 ×50 m using reference images collected years earlier by a

Cited by 11SourceScholar