← Search

Tianyu Gao

19 accepted papers

2025

DTR: Dynamic Tree-Ring Watermarking Framework for Diffusion-Based Video Generation

ICASSP 2025accepted

The growing capabilities of diffusion-based text-to-video models have raised significant concerns about copyright protection and the traceability of synthetic video content. To address these concerns, existing watermarking techniques have been developed to invisibly embed information within video co…

Cited by 0SourceScholar
2025

HELMET: How to Evaluate Long-context Models Effectively and Thoroughly

ICLR 2025poster

Many benchmarks exist for evaluating long-context language models (LCLMs), yet developers often rely on synthetic tasks such as needle-in-a-haystack (NIAH) or an arbitrary subset of tasks. However, it remains unclear whether these benchmarks reflect the diverse downstream applications of LCLMs, and…

Cited by 0SourcePDFScholar
2025

How to Train Long-Context Language Models (Effectively)

ACL 2025long

We study continued training and supervised fine-tuning (SFT) of a language model (LM) to make effective use of long-context information. We first establish a reliable evaluation protocol to guide model development—instead of perplexity or simple needle-in-a-haystack (NIAH) tests, we use a broad set…

2025

Metadata Conditioning Accelerates Language Model Pre-training

ICML 2025poster

The vast diversity of styles, domains, and quality levels present in language model pre-training corpora is essential in developing general model capabilities, but efficiently learning and deploying the correct behaviors exemplified in each of these heterogeneous data sources is challenging. To addr…

2024

Evaluating Large Language Models at Evaluating Instruction Following

ICLR 2024poster

As research in large language models (LLMs) continues to accelerate, LLM-based evaluation has emerged as a scalable and cost-effective alternative to human evaluations for comparing the ever increasing list of models. This paper investigates the efficacy of these “LLM evaluators”, particularly in us…

2024

LitSearch: A Retrieval Benchmark for Scientific Literature Search

EMNLP 2024main

Literature search questions, such as “where can I find research on the evaluation of consistency in generated summaries?” pose significant challenges for modern search engines and retrieval systems. These questions often require a deep understanding of research concepts and the ability to reason ove…

2024

Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning

ICLR 2024poster

The popularity of LLaMA (Touvron et al., 2023a;b) and other recently emerged moderate-sized large language models (LLMs) highlights the potential of building smaller yet powerful LLMs. Regardless, the cost of training such models from scratch on trillions of tokens remains high. In this work, we stu…

2023

Enabling Large Language Models to Generate Text with Citations

EMNLP 2023long main

Large language models (LLMs) have emerged as a widely-used tool for information seeking, but their generated outputs are prone to hallucination. In this work, our aim is to allow LLMs to generate text with citations, improving their factual correctness and verifiability. Existing work mainly relies…

Cited by 0SourcecodeScholar
2023

Fine-Tuning Language Models with Just Forward Passes

NeurIPS 2023oral

Fine-tuning language models (LMs) has yielded success on diverse downstream tasks, but as LMs grow in size, backpropagation requires a prohibitively large amount of memory. Zeroth-order (ZO) methods can in principle estimate gradients using only two forward passes but are theorized to be catastrophi…

2023

The CRINGE Loss: Learning what language not to model

ACL 2023long

Standard language model training employs gold human documents or human-human interaction data, and treats all training data as positive examples. Growing evidence shows that even with very large amounts of positive training data, issues remain that can be alleviated with relatively small amounts of…

Cited by 35SourcePDFScholar
2023

What In-Context Learning “Learns” In-Context: Disentangling Task Recognition and Task Learning

ACL 2023findings

Large language models (LLMs) exploit in-context learning (ICL) to solve tasks with only a few demonstrations, but its mechanisms are not yet well-understood. Some works suggest that LLMs only recall already learned concepts from pre-training, while others hint that ICL performs implicit learning ove…

2022

Automatic Label Sequence Generation for Prompting Sequence-to-sequence Models

COLING 2022main

Prompting, which casts downstream applications as language modeling tasks, has shown to be sample efficient compared to standard fine-tuning with pre-trained models. However, one pitfall of prompting is the need of manually-designed patterns, whose outcome can be unintuitive and requires large valid…

2022

Ditch the Gold Standard: Re-evaluating Conversational Question Answering

ACL 2022long

Conversational question answering aims to provide natural-language answers to users in information-seeking conversations. Existing conversational QA benchmarks compare models with pre-collected human-human conversations, using ground-truth answers provided in conversational history. It remains uncle…

2022

Recovering Private Text in Federated Learning of Language Models

NeurIPS 2022accept

Federated learning allows distributed users to collaboratively train a model while keeping each user’s data private. Recently, a growing body of work has demonstrated that an eavesdropping attacker can effectively recover image data from gradients transmitted during federated learning. However, litt…

2020

Few-shot Relation Extraction via Bayesian Meta-learning on Relation Graphs

ICML 2020poster

This paper studies few-shot relation extraction, which aims at predicting the relation for a pair of entities in a sentence by training with a few labeled examples in each relation. To more effectively generalize to new relations, in this paper we study the relationships between different relations…

2020

Meta-Information Guided Meta-Learning for Few-Shot Relation Classification

COLING 2020main

Few-shot classification requires classifiers to adapt to new classes with only a few training instances. State-of-the-art meta-learning approaches such as MAML learn how to initialize and fast adapt parameters from limited instances, which have shown promising results in few-shot classification. How…