← Search

He He

44 accepted papers

2026

Estimating Tail Risks in Language Model Output Distributions

ICML 2026spotlight

Language models are increasingly capable and are being rapidly deployed on a population-level scale. As a result, the safety of these models is increasingly high-stakes. Fortunately, advances in alignment have significantly reduced the likelihood of harmful model outputs. However, when models are qu…

Cited by 0SourceScholar
2026

Is it Thinking or Cheating? Detecting Implicit Reward Hacking by Measuring Reasoning Effort

ICLR 2026oral

Reward hacking, where a reasoning model exploits loopholes in a reward function to achieve high rewards without solving the intended task, poses a significant threat. This behavior may be explicit, i.e. verbalized in the model's chain-of-thought (CoT), or implicit, where the CoT appears benign thus…

Cited by 0SourceScholar
2026

Measuring LLM Novelty As The Frontier Of Original And High-Quality Output

ICLR 2026poster

As large language models (LLMs) are increasingly used for ideation and scientific discovery, it is important to evaluate their ability to generate novel output. Prior work evaluates novelty as originality with respect to model training data, but original outputs can be of low quality. In contrast, n…

Cited by 0SourceScholar
2026

Monitoring Decomposition Attacks with Lightweight Sequential Monitors

ICLR 2026poster

As LLMs become more agentic, a critical risk emerges: attackers can \emph{decompose} harmful goals into stateful, benign subtasks that trick LLM agents into executing them without realizing the harmful intent in the same context. The challenge lies in the existing shallow safety alignment techniques…

Cited by 0SourcecodeScholar
2026

Reasoning Models Struggle to Control their Chains of Thought

ICML 2026poster

Instruction following in LLMs captures models' ability to change their visible behaviors as requested by users. Instead, we study models' ability to control their chain-of-thought (CoT). This capability -- CoT controllability -- is undesirable because it could allow models to suppress signs of misbe…

Cited by 0SourceScholar
2025

Adaptive Deployment of Untrusted LLMs Reduces Distributed Threats

ICLR 2025poster

As large language models (LLMs) grow more powerful, they also become more difficult to trust. They could be either aligned with human intentions, or exhibit "subversive misalignment" -- introducing subtle errors that bypass safety checks. Although individual errors may not immediately cause harm, ea…

Cited by 3SourcePDFScholar
2025

Language Models Learn to Mislead Humans via RLHF

ICLR 2025poster

Language models (LMs) can produce errors that are hard to detect for humans, especially when the task is complex. RLHF, the most popular post-training method, may exacerbate this problem: to achieve higher rewards, LMs might get better at convincing humans that they are right even when they are wron…

2025

Predicting Empirical AI Research Outcomes with Language Models

NeurIPS 2025poster

Many promising-looking ideas in AI research fail to deliver, but their validation takes substantial human labor and compute. Predicting an idea's chance of success is thus crucial for accelerating empirical AI research, a skill that even expert researchers can only acquire through substantial experi…

Cited by 0SourceScholar
2025

Towards Consistent Natural-Language Explanations via Explanation-Consistency Finetuning

COLING 2025main

Large language models (LLMs) often generate convincing, fluent explanations. However, different from humans, they often generate inconsistent explanations on different inputs. For example, an LLM may explain “all birds can fly” when answering the question “Can sparrows fly?” but meanwhile answer “no…

2025

Transformers Struggle to Learn to Search

ICLR 2025poster

Search is an ability foundational in many important tasks, and recent studies have shown that large language models (LLMs) struggle to perform search robustly. It is unknown whether this inability is due to a lack of data, insufficient model parameters, or fundamental limitations of the transformer…

2025

When Benchmarks Talk: Re-Evaluating Code LLMs with Interactive Feedback

ACL 2025finding

Programming with a coding assistant is a fundamentally interactive process, yet existing static benchmarks fail to capture key features of model-user collaboration. We introduce an interactive evaluation pipeline to examine how LLMs incorporate different types of feedback in a collaborative setting,…

2024

Do Models Explain Themselves? Counterfactual Simulatability of Natural Language Explanations

ICML 2024spotlight

Large language models (LLMs) are trained to imitate humans to explain human decisions. However, do LLMs explain themselves? Can they help humans build mental models of how LLMs process different inputs? To answer these questions, we propose to evaluate $\textbf{counterfactual simulatability}$ of nat…

Cited by 57SourcePDFScholar
2024

Iterative Reasoning Preference Optimization

NeurIPS 2024poster

Iterative preference optimization methods have recently been shown to perform well for general instruction tuning tasks, but typically make little improvement on reasoning tasks. In this work we develop an iterative approach that optimizes the preference between competing generated Chain-of-Thought…

Cited by 136SourcePDFScholar
2024

Parallel Structures in Pre-training Data Yield In-Context Learning

ACL 2024long

Pre-trained language models (LMs) are capable of in-context learning (ICL): they can adapt to a task with only a few examples given in the prompt without any parameter update. However, it is unclear where this capability comes from as there is a stark distribution shift between pre-training text and…

Cited by 13SourcePDFScholar
2024

Personas as a Way to Model Truthfulness in Language Models

EMNLP 2024main

Large language models (LLMs) are trained on vast amounts of text from the internet, which contains both factual and misleading information about the world. While unintuitive from a classic view of LMs, recent work has shown that the truth value of a statement can be elicited from the model’s represe…

2024

The PRISM Alignment Dataset: What Participatory, Representative and Individualised Human Feedback Reveals About the Subjective and Multicultural Alignment of Large Language Models

NeurIPS 2024oral

Human feedback is central to the alignment of Large Language Models (LLMs). However, open questions remain about the methods (how), domains (where), people (who) and objectives (to what end) of feedback processes. To navigate these questions, we introduce PRISM, a new dataset which maps the sociodem…

2024

Your Co-Workers Matter: Evaluating Collaborative Capabilities of Language Models in Blocks World

ACL 2024findings

Language agents that interact with the world on their own have great potential for automating digital tasks. While large language model (LLM) agents have made progress in understanding and executing tasks such as textual games and webpage control, many real-world tasks also require collaboration wit…

2023

Efficient Shapley Values Estimation by Amortization for Text Classification

ACL 2023long

Despite the popularity of Shapley Values in explaining neural text classification models, computing them is prohibitive for large pretrained models due to a large number of model evaluations. In practice, Shapley Values are often estimated with a small number of stochastic model evaluations. However…

2023

Extrapolative Controlled Sequence Generation via Iterative Refinement

ICML 2023poster

We study the problem of extrapolative controlled generation, i.e., generating sequences with attribute values beyond the range seen in training. This task is of significant importance in automated design, especially drug discovery, where the goal is to design novel proteins that are better (e.g., mo…

2023

Language Models Are Greedy Reasoners: A Systematic Formal Analysis of Chain-of-Thought

ICLR 2023poster

Large language models (LLMs) have shown remarkable reasoning capabilities given chain-of-thought prompts (examples with intermediate reasoning steps). Existing benchmarks measure reasoning ability indirectly, by evaluating accuracy on downstream tasks such as mathematical reasoning. However, it is u…

2023

Measuring Inductive Biases of In-Context Learning with Underspecified Demonstrations

ACL 2023long

In-context learning (ICL) is an important paradigm for adapting large language models (LLMs) to new tasks, but the generalization behavior of ICL remains poorly understood. We investigate the inductive biases of ICL from the perspective of feature bias: which feature ICL is more likely to use given…

2023

Reward Gaming in Conditional Text Generation

ACL 2023long

To align conditional text generation model outputs with desired behaviors, there has been an increasing focus on training the model using reinforcement learning (RL) with reward functions learned from human annotations. Under this framework, we identify three common cases where high rewards are inco…

Cited by 21SourcePDFScholar
2023

Testing the General Deductive Reasoning Capacity of Large Language Models Using OOD Examples

NeurIPS 2023poster

Given the intractably large size of the space of proofs, any model that is capable of general deductive reasoning must generalize to proofs of greater complexity. Recent studies have shown that large language models (LLMs) possess some abstract deductive reasoning ability given chain-of-thought prom…

2022

Are All Spurious Features in Natural Language Alike? An Analysis through a Causal Lens

EMNLP 2022main

The term ‘spurious correlations’ has been used in NLP to informally denote any undesirable feature-label correlations. However, a correlation can be undesirable because (i) the feature is irrelevant to the label (e.g. punctuation in a review), or (ii) the feature’s effect on the label depends on the…

2022

Exploring the Role of Task Transferability in Large-Scale Multi-Task Learning

NAACL 2022long

Recent work has found that multi-task training with a large number of diverse tasks can uniformly improve downstream performance on unseen target tasks. In contrast, literature on task transferability has established that the choice of intermediate tasks can heavily affect downstream task performanc…

Cited by 20SourcePDFScholar
2022

Faithful or Extractive? On Mitigating the Faithfulness-Abstractiveness Trade-off in Abstractive Summarization

ACL 2022long

Despite recent progress in abstractive summarization, systems still suffer from faithfulness errors. While prior work has proposed models that improve faithfulness, it is unclear whether the improvement comes from an increased level of extractiveness of the model outputs as one naive way to improve…

2022

Help me write a poem: Instruction Tuning as a Vehicle for Collaborative Poetry Writing

EMNLP 2022main

Recent work in training large language models (LLMs) to follow natural language instructions has opened up exciting opportunities for natural language interface design. Building on the prior success of large language models in the realm of computer assisted creativity, in this work, we present CoPoe…

2022

Improving Faithfulness by Augmenting Negative Summaries from Fake Documents

EMNLP 2022main

Current abstractive summarization systems tend to hallucinate content that is unfaithful to the source document, posing a risk of misinformation. To mitigate hallucination, we must teach the model to distinguish hallucinated summaries from faithful ones. However, the commonly used maximum likelihood…

Cited by 6SourcePDFScholar
2022

Meta-learning via Language Model In-context Tuning

ACL 2022long

The goal of meta-learning is to learn to adapt to a new task with only a few labeled examples. Inspired by the recent progress in large language models, we propose in-context tuning (ICT), which recasts task adaptation and prediction as a simple sequence prediction problem: to form the input sequenc…

2022

QuALITY: Question Answering with Long Input Texts, Yes!

NAACL 2022long

To enable building and testing models on long-document comprehension, we introduce QuALITY, a multiple-choice QA dataset with context passages in English that have an average length of about 5,000 tokens, much longer than typical current models can process. Unlike in prior work with passages, our qu…

2022

SeqPATE: Differentially Private Text Generation via Knowledge Distillation

NeurIPS 2022accept

Protecting the privacy of user data is crucial for text generation models, which can leak sensitive information during generation. Differentially private (DP) learning methods provide guarantees against identifying the existence of a training sample from model outputs. PATE is a recent DP learning a…

Cited by 24SourcePDFScholar
2021

IRM—when it works and when it doesn't: A test case of natural language inference

NeurIPS 2021poster

Invariant Risk Minimization (IRM) is a recently proposed framework for out-of-distribution (o.o.d) generalization. Most of the studies on IRM so far have focused on theoretical results, toy problems, and simple models. In this work, we investigate the applicability of IRM to bias mitigation-a speci…

Cited by 16SourcePDFScholar
2016

A Credit Assignment Compiler for Joint Prediction

NeurIPS 2016poster

Many machine learning applications involve jointly predicting multiple mutually dependent output variables. Learning to search is a family of methods where the complex decision problem is cast into a sequence of decisions via a search space. Although these methods have shown promise both in theory a…

Cited by 11SourcePDFScholar
2016

Opponent Modeling in Deep Reinforcement Learning

ICML 2016poster

Opponent modeling is necessary in multi-agent settings where secondary agents with competing goals also adapt their strategies, yet it remains challenging because of strategies’ complex interaction and the non-stationary nature. Most previous work focuses on developing probabilistic models or parame…