← Search

Mayee F. Chen

10 accepted papers

2025

Aioli: A Unified Optimization Framework for Language Model Data Mixing

ICLR 2025poster

Language model performance depends on identifying the optimal mixture of data groups to train on (e.g., law, code, math). Prior work has proposed a diverse set of methods to efficiently learn mixture proportions, ranging from fitting regression models over training runs to dynamically updating propo…

2025

An Architecture Search Framework for Inference-Time Techniques

ICML 2025poster

Inference-time techniques, such as repeated sampling or iterative revisions, are emerging as powerful ways to enhance large-language models (LLMs) at test time. However, best practices for developing systems that combine these techniques remain underdeveloped due to our limited understanding of the…

Cited by 0SourcePDFScholar
2025

Weaver: Shrinking the Generation-Verification Gap by Scaling Compute for Verification

NeurIPS 2025poster

Verifiers can improve language model (LM) capabilities by providing feedback or selecting the best response from a pool of generated candidates. Currently, high-quality verifiers are either unscalable (e.g., humans) or limited in utility (e.g., tools like Lean for formal proofs). While LM judges and…

Cited by 0SourceScholar
2024

DataComp-LM: In search of the next generation of training sets for language models

NeurIPS 2024poster

We introduce DataComp for Language Models, a testbed for controlled dataset experiments with the goal of improving language models. As part of DCLM, we provide a standardized corpus of 240T tokens extracted from Common Crawl, effective pretraining recipes based on the OpenLM framework, and a broad s…

Cited by 64SourcePDFScholar
2024

Smoothie: Label Free Language Model Routing

NeurIPS 2024poster

Large language models (LLMs) are increasingly used in applications where LLM inputs may span many different tasks. Recent work has found that the choice of LLM is consequential, and different LLMs may be good for different input samples. Prior approaches have thus explored how engineers might select…

2023

A case for reframing automated medical image classification as segmentation

NeurIPS 2023poster

Image classification and segmentation are common applications of deep learning to radiology. While many tasks can be framed using either classification or segmentation, classification has historically been cheaper to label and more widely used. However, recent work has drastically reduced the cost o…

Cited by 5SourcePDFScholar
2023

Ask Me Anything: A simple strategy for prompting language models

ICLR 2023top-25%

Large language models (LLMs) transfer well to new tasks out-of-the-box simply given a natural language prompt that demonstrates how to perform the task and no additional training. Prompting is a brittle process wherein small modifications to the prompt can cause large variations in the model predict…

2023

Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification

NeurIPS 2023poster

Recent work has shown that language models' (LMs) prompt-based learning capabilities make them well suited for automating data labeling in domains where manual annotation is expensive. The challenge is that while writing an initial prompt is cheap, improving a prompt is costly---practitioners often…

Cited by 6SourcePDFScholar
2023

Skill-it! A data-driven skills framework for understanding and training language models

NeurIPS 2023spotlight

The quality of training data impacts the performance of pre-trained large language models (LMs). Given a fixed budget of tokens, we study how to best select data that leads to good downstream model performance across tasks. We develop a new framework based on a simple hypothesis: just as humans acqu…

Cited by 65SourcePDFScholar
2022

Shoring up the foundations: fusing model embeddings and weak supervision

UAI 2022poster

Foundation models offer an exciting new paradigm for constructing models with out-of-the-box embeddings and a few labeled examples. However, it is not clear how to best apply foundation models without labeled data. A potential approach is to fuse foundation models with weak supervision frameworks, w…