← Search

Jason Wei

22 accepted papers

2024

A Pretrainer’s Guide to Training Data: Measuring the Effects of Data Age, Domain Coverage, Quality, & Toxicity

NAACL 2024long

Pretraining data design is critically under-documented and often guided by empirically unsupported intuitions. We pretrain models on data curated (1) at different collection times, (2) with varying toxicity and quality filters, and (3) with different domain compositions. First, we find that temporal…

2024

FreshLLMs: Refreshing Large Language Models with Search Engine Augmentation

ACL 2024findings

Since most large language models (LLMs) are trained once and never updated, they struggle to dynamically adapt to our ever-changing world. In this work, we present FreshQA, a dynamic QA benchmark that tests a model’s ability to answer questions that may require reasoning over up-to-date world knowle…

2024

Mixture-of-Experts Meets Instruction Tuning: A Winning Combination for Large Language Models

ICLR 2024poster

Sparse Mixture-of-Experts (MoE) is a neural architecture design that adds learnable parameters to Large Language Models (LLMs) without increasing computational complexity (FLOPs). Instruction tuning is a technique for training LLMs to follow instructions. We advocate combining these two approaches,…

Cited by 78SourcePDFScholar
2023

Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Them

ACL 2023findings

BIG-Bench (Srivastava et al., 2022) is a diverse evaluation suite that focuses on tasks believed to be beyond the capabilities of current language models. Language models have already made good progress on this benchmark, with the best model in the BIG-Bench paper outperforming average reported huma…

2023

Language models are multilingual chain-of-thought reasoners

ICLR 2023poster

We evaluate the reasoning abilities of large language models in multilingual settings. We introduce the Multilingual Grade School Math (MGSM) benchmark, by manually translating 250 grade-school math problems from the GSM8K dataset (Cobbe et al., 2021) into ten typologically diverse languages. We fin…

2023

Least-to-Most Prompting Enables Complex Reasoning in Large Language Models

ICLR 2023poster

Chain-of-thought prompting has demonstrated remarkable performance on various natural language reasoning tasks. However, it tends to perform poorly on tasks which requires solving problems harder than the exemplars shown in the prompts. To overcome this challenge of easy-to-hard generalization, we p…

Cited by 1438SourcePDFScholar
2023

Mind's Eye: Grounded Language Model Reasoning through Simulation

ICLR 2023poster

Successful and effective communication between humans and AI relies on a shared experience of the world. By training solely on written text, current language models (LMs) miss the grounded experience of humans in the real-world---their failure to relate language to the physical world causes knowledg…

Cited by 84SourcePDFScholar
2023

Self-Consistency Improves Chain of Thought Reasoning in Language Models

ICLR 2023poster

Chain-of-thought prompting combined with pretrained large language models has achieved encouraging results on complex reasoning tasks. In this paper, we propose a new decoding strategy, self-consistency, to replace the naive greedy decoding used in chain-of-thought prompting. It first samples a dive…

Cited by 1586SourcePDFScholar
2023

The Flan Collection: Designing Data and Methods for Effective Instruction Tuning

ICML 2023poster

We study the design decision of publicly available instruction tuning methods, by reproducing and breaking down the development of Flan 2022 (Chung et al., 2022). Through careful ablation studies on the Flan Collection of tasks and methods, we tease apart the effect of design decisions which enable…

2023

Transcending Scaling Laws with 0.1% Extra Compute

EMNLP 2023long main

Scaling language models improves performance but comes with significant computational costs. This paper proposes UL2R, a method that substantially improves existing language models and their scaling curves with a relatively tiny amount of extra compute. The key idea is to continue training a state-o…

Cited by 0SourceScholar
2023

UL2: Unifying Language Learning Paradigms

ICLR 2023poster

Existing pre-trained models are generally geared towards a particular class of problems. To date, there seems to be still no consensus on what the right architecture and pre-training setup should be. This paper presents a unified framework for pre-training models that are universally effective acros…

2022

A Recipe for Arbitrary Text Style Transfer with Large Language Models

ACL 2022short

In this paper, we leverage large language models (LLMs) to perform zero-shot text style transfer. We present a prompting method that we call augmented zero-shot learning, which frames style transfer as a sentence rewriting task and requires only a natural language instruction, without model fine-tun…

Cited by 186SourcePDFScholar
2022

Chain-of-Thought Prompting Elicits Reasoning in Large Language Models

NeurIPS 2022accept

We explore how generating a chain of thought---a series of intermediate reasoning steps---significantly improves the ability of large language models to perform complex reasoning. In particular, we show how such reasoning abilities emerge naturally in sufficiently large language models via a simple…

Cited by 13586SourcePDFScholar
2022

Finetuned Language Models are Zero-Shot Learners

ICLR 2022oral

This paper explores a simple method for improving the zero-shot learning abilities of language models. We show that instruction tuning—finetuning language models on a collection of datasets described via instructions—substantially improves zero-shot performance on unseen tasks. We take a 137B parame…

2022

The MultiBERTs: BERT Reproductions for Robustness Analysis

ICLR 2022spotlight

Experiments with pre-trained models such as BERT are often based on a single checkpoint. While the conclusions drawn apply to the artifact tested in the experiment (i.e., the particular instance of the model), it is not always clear whether they hold for the more general procedure which includes the…

2021

Few-Shot Text Classification with Triplet Networks, Data Augmentation, and Curriculum Learning

NAACL 2021long

Few-shot text classification is a fundamental NLP task in which a model aims to classify text into a large number of categories, given only a few training examples per category. This paper explores data augmentation—a technique particularly suitable for training with limited data—for this few-shot,…

2021

Frequency Effects on Syntactic Rule Learning in Transformers

EMNLP 2021main

Pre-trained language models perform well on a variety of linguistic tasks that require symbolic reasoning, raising the question of whether such models implicitly represent abstract symbols and rules. We investigate this question using the case study of BERT’s performance on English subject–verb agre…

2021

Good-Enough Example Extrapolation

EMNLP 2021main

This paper asks whether extrapolating the hidden space distribution of text examples from one class onto another is a valid inductive bias for data augmentation. To operationalize this question, I propose a simple data augmentation protocol called “good-enough example extrapolation” (GE3). GE3 is li…

Cited by 6SourcePDFScholar
2021

Linguistic Complexity Loss in Text-Based Therapy

NAACL 2021long

The complexity loss paradox, which posits that individuals suffering from disease exhibit surprisingly predictable behavioral dynamics, has been observed in a variety of both human and animal physiological systems. The recent advent of online text-based therapy presents a new opportunity to analyze…

Cited by 9SourcePDFScholar