← Search

Zhengxuan Wu

22 accepted papers

2026

Language Model Circuits Are Sparse in the Neuron Basis

ICML 2026spotlight

The high-level concepts that a neural network uses to perform computation need not be aligned to individual neurons (Smolensky, 1986). Language model interpretability research has thus turned to techniques such as *sparse autoencoders* (SAEs) to decompose the neuron basis into more interpretable uni…

Cited by 0SourceScholar
2025

AxBench: Steering LLMs? Even Simple Baselines Outperform Sparse Autoencoders

ICML 2025spotlight

Fine-grained steering of language model outputs is essential for safety and reliability. Prompting and finetuning are widely used to achieve these goals, but interpretability researchers have proposed a variety of representation-based techniques as well, including sparse autoencoders (SAEs), linear…

2025

Improved Representation Steering for Language Models

NeurIPS 2025spotlight

Steering methods for language models (LMs) seek to provide fine-grained and interpretable control over model generations by variously changing model inputs, weights, or representations to adjust behavior. Recent work has shown that adjusting weights or representations is often less effective than st…

Cited by 0SourcecodeScholar
2024

Dancing in Chains: Reconciling Instruction Following and Faithfulness in Language Models

EMNLP 2024main

Modern language models (LMs) need to follow human instructions while being faithful; yet, they often fail to achieve both. Here, we provide concrete evidence of a trade-off between instruction following (i.e., follow open-ended instructions) and faithfulness (i.e., ground responses in given context)…

2024

In-Context Sharpness as Alerts: An Inner Representation Perspective for Hallucination Mitigation

ICML 2024poster

Large language models (LLMs) frequently hallucinate, e.g., making factual errors, yet our understanding of why they make these errors remains limited. In this study, we aim to understand the underlying mechanisms of LLM hallucinations from the perspective of *inner representations*. We discover a pa…

2024

RAVEL: Evaluating Interpretability Methods on Disentangling Language Model Representations

ACL 2024long

Individual neurons participate in the representation of multiple high-level concepts. To what extent can different interpretability methods successfully disentangle these roles? To help address this question, we introduce RAVEL (Resolving Attribute-Value Entanglements in Language Models), a dataset…

2024

ReFT: Representation Finetuning for Language Models

NeurIPS 2024spotlight

Parameter-efficient finetuning (PEFT) methods seek to adapt large neural models via updates to a small number of *weights*. However, much prior interpretability work has shown that *representations* encode rich semantic information, suggesting that editing representations might be a more powerful al…

2024

pyvene: A Library for Understanding and Improving PyTorch Models via Interventions

NAACL 2024system demonstrations

Interventions on model-internal states are fundamental operations in many areas of AI, including model editing, steering, robustness, and interpretability. To facilitate such research, we introduce pyvene, an open-source Python library that supports customizable interventions on a range of different…

2023

Causal Proxy Models for Concept-based Model Explanations

ICML 2023poster

Explainability methods for NLP systems encounter a version of the fundamental problem of causal inference: for a given ground-truth input text, we never truly observe the counterfactual texts necessary for isolating the causal effects of model representations on outputs. In response, many explainabi…

2023

Inducing Character-level Structure in Subword-based Language Models with Type-level Interchange Intervention Training

ACL 2023findings

Language tasks involving character-level manipulations (e.g., spelling corrections, arithmetic operations, word games) are challenging for models operating on subword units. To address this, we develop a causal intervention framework to learn robust and interpretable character representations inside…

2023

Interpretability at Scale: Identifying Causal Mechanisms in Alpaca

NeurIPS 2023poster

Obtaining human-interpretable explanations of large, general-purpose language models is an urgent goal for AI safety. However, it is just as important that our interpretability methods are faithful to the causal dynamics underlying model behavior and able to robustly generalize to unseen inputs. Dis…

2023

MQuAKE: Assessing Knowledge Editing in Language Models via Multi-Hop Questions

EMNLP 2023long main

The information stored in large language models (LLMs) falls out of date quickly, and retraining from scratch is often not an option. This has recently given rise to a range of techniques for injecting new facts through updating model weights. Current evaluation paradigms are extremely limited, main…

Cited by 0SourcecodeScholar
2023

Oolong: Investigating What Makes Transfer Learning Hard with Controlled Studies

EMNLP 2023short main

When we transfer a pretrained language model to a new language, there are many axes of variation that change at once. To disentangle the impact of different factors like syntactic similarity and vocabulary similarity, we propose a set of \emph{controlled transfer studies}: we systematically transfor…

Cited by 0SourcecodeScholar
2022

CEBaB: Estimating the Causal Effects of Real-World Concepts on NLP Model Behavior

NeurIPS 2022accept

The increasing size and complexity of modern ML systems has improved their predictive capabilities but made their behavior harder to explain. Many techniques for model explanation have been developed in response, but we lack clear criteria for assessing these techniques. In this paper, we cast model…

Cited by 54SourcePDFScholar
2022

Causal Distillation for Language Models

NAACL 2022long

Distillation efforts have led to language models that are more compact and efficient without serious drops in performance. The standard approach to distillation trains a student model against two objectives: a task-specific objective (e.g., language modeling) and an imitation objective that encourag…

2022

Inducing Causal Structure for Interpretable Neural Networks

ICML 2022spotlight

In many areas, we have well-founded insights about causal structure that would be useful to bring into our trained models while still allowing them to learn in a data-driven fashion. To achieve this, we present the new method of interchange intervention training (IIT). In IIT, we (1) align variables…

2022

ZeroC: A Neuro-Symbolic Model for Zero-shot Concept Recognition and Acquisition at Inference Time

NeurIPS 2022accept

Humans have the remarkable ability to recognize and acquire novel visual concepts in a zero-shot manner. Given a high-level, symbolic description of a novel concept in terms of previously learned visual concepts and their relations, humans can recognize novel concepts without seeing any examples. Mo…

2021

DynaSent: A Dynamic Benchmark for Sentiment Analysis

ACL 2021long

We introduce DynaSent (‘Dynamic Sentiment’), a new English-language benchmark task for ternary (positive/negative/neutral) sentiment analysis. DynaSent combines naturally occurring sentences with sentences created using the open-source Dynabench Platform, which facilities human-and-model-in-the-loop…

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

ReaSCAN: Compositional Reasoning in Language Grounding

NeurIPS 2021poster

The ability to compositionally map language to referents, relations, and actions is an essential component of language understanding. The recent gSCAN dataset (Ruis et al. 2020, NeurIPS) is an inspiring attempt to assess the capacity of models to learn this kind of grounding in scenarios involving n…

Cited by 24SourcecodeScholar