← Search

Robin Jia

45 accepted papers

2026

Cancer-Myth: Evaluating Large Language Models on Patient Questions with False Presuppositions

ICLR 2026poster

Cancer patients are increasingly turning to large language models (LLMs) for medical information, making it critical to assess how well these models handle complex, personalized questions. However, current medical benchmarks focus on medical exams or consumer-searched questions and do not evaluate…

Cited by 0SourcecodeScholar
2026

EPSVec: Efficient and Private Synthetic Text Generation via Dataset Vectors

ICML 2026poster

High-quality data is essential for modern machine learning, yet many valuable corpora are sensitive and cannot be freely shared. Synthetic data offers a practical substitute for downstream development, and large language models (LLMs) have emerged as powerful engines for generating it. However, exis…

Cited by 0SourceScholar
2026

FoNE: Precise Single-Token Number Embeddings via Fourier Features

ICLR 2026poster

Language models treat numbers in the same way as ordinary word tokens, which introduces two major issues: (1) embeddings of numerical tokens primarily reflect their frequency in text corpora rather than their inherent numerical properties, leading to frequency bias, and (2) numbers are often split i…

Cited by 0SourcecodeScholar
2026

Function Induction and Task Generalization: An Interpretability Study with Off-by-One Addition

ICLR 2026poster

Large language models demonstrate the intriguing ability to perform unseen tasks via in-context learning. However, it remains unclear what mechanisms inside the model drive such task-level generalization. In this work, we approach this question through the lens of off-by-one addition (i.e., 1+1=3, 2…

Cited by 0SourcecodeScholar
2026

Hubble: a Model Suite to Advance the Study of LLM Memorization

ICLR 2026oral

We present Hubble, a suite of open-source large language models (LLMs) for the scientific study of LLM memorization. Hubble models come as minimal pairs: standard models are pretrained on a large English corpus, and perturbed models are trained in the same way but with controlled insertion of text (…

Cited by 0SourcecodeScholar
2026

PSALM-V: Automating Symbolic Planning in Interactive Visual Environments with Large Language Models

ICRA 2026poster

We propose PSALM-V, the first autonomous neuro-symbolic learning system able to induce symbolic action semantics (i.e., pre- and post-conditions) in visual environments through interaction. PSALM-V bootstraps reliable symbolic planning without expert action definitions, using LLMs to generate heuris…

2026

Transformers Provably Learn Algorithmic Solutions for Graph Connectivity, But Only with the Right Data

ICML 2026poster

Transformers often fail to learn generalizable algorithms, instead relying on brittle heuristics. Using graph connectivity as a testbed, we explain this phenomenon both theoretically and empirically. We consider a simplified Transformer architecture, the Disentangled Transformer, and prove that an $…

Cited by 0SourceScholar
2025

Language Models Can Infer Action Semantics for Symbolic Planners from Environment Feedback

NAACL 2025long

Symbolic planners can discover a sequence of actions from initial to goal states given expert-defined, domain-specific logical action semantics. Large Language Models (LLMs) can directly generate such sequences, but limitations in reasoning and state-tracking often result in plans that are insuffici…

2025

Mechanistic Interpretability of Emotion Inference in Large Language Models

ACL 2025finding

Large language models (LLMs) show promising capabilities in predicting human emotions from text. However, the mechanisms through which these models process emotional stimuli remain largely unexplored. Our study addresses this gap by investigating how autoregressive LLMs infer emotions, showing that…

Cited by 0SourcePDFScholar
2025

Robust Data Watermarking in Language Models by Injecting Fictitious Knowledge

ACL 2025finding

Data watermarking in language models injects traceable signals, such as specific token sequences or stylistic patterns, into copyrighted text, allowing copyright holders to track and verify training data ownership. Previous data watermarking techniques primarily focus on effective memorization after…

2025

TLDR: Token-Level Detective Reward Model for Large Vision Language Models

ICLR 2025poster

Although reward models have been successful in improving multimodal large language models, the reward models themselves remain brutal and contain minimal information. Notably, existing reward models only mimic human annotations by assigning only one feedback to any text, no matter how long the text…

Cited by 5SourcePDFScholar
2024

Do Localization Methods Actually Localize Memorized Data in LLMs? A Tale of Two Benchmarks

NAACL 2024long

The concept of localization in LLMs is often mentioned in prior work; however, methods for localization have never been systematically and directly evaluated. We propose two complementary benchmarks that evaluate the ability of localization methods to pinpoint LLM components responsible for memorize…

2024

Efficient End-to-End Visual Document Understanding with Rationale Distillation

NAACL 2024long

Understanding visually situated language requires interpreting complex layouts of textual and visual elements. Pre-processing tools, such as optical character recognition (OCR), can map document image inputs to textual tokens, then large language models (LLMs) can reason over text.However, such meth…

Cited by 3SourcePDFScholar
2024

Pre-trained Large Language Models Use Fourier Features to Compute Addition

NeurIPS 2024poster

Pre-trained large language models (LLMs) exhibit impressive mathematical reasoning capabilities, yet how they compute basic arithmetic, such as addition, remains unclear. This paper shows that pre-trained LLMs add numbers using Fourier features---dimensions in the hidden state that represent number…

Cited by 5SourcePDFScholar
2024

Transformers Learn to Achieve Second-Order Convergence Rates for In-Context Linear Regression

NeurIPS 2024poster

Transformers excel at *in-context learning* (ICL)---learning from demonstrations without parameter updates---but how they do so remains a mystery. Recent work suggests that Transformers may internally run Gradient Descent (GD), a first-order optimization method, to perform ICL. In this paper, we ins…

2024

When Parts Are Greater Than Sums: Individual LLM Components Can Outperform Full Models

EMNLP 2024main

This paper studies in-context learning by decomposing the output of large language models into the individual contributions of attention heads and MLPs (components). We observe curious components: good-performing ones that individually do well on a classification task, even when the model performs p…

2023

Chain-of-Questions Training with Latent Answers for Robust Multistep Question Answering

EMNLP 2023long main

We propose Chain-of-Questions, a framework that trains a model to robustly answer multistep questions by generating and answering sub-questions. We obtain supervision for sub-questions from human-annotated question decomposition meaning representation (QDMR), but QDMR does not include annotated answ…

Cited by 0SourceScholar
2023

Contrastive Novelty-Augmented Learning: Anticipating Outliers with Large Language Models

ACL 2023long

In many task settings, text classification models are likely to encounter examples from novel classes on which they cannot predict correctly. Selective prediction, in which models abstain on low-confidence examples, provides a possible solution, but existing models are often overly confident on unse…

2023

Do Question Answering Modeling Improvements Hold Across Benchmarks?

ACL 2023long

Do question answering (QA) modeling improvements (e.g., choice of architecture and training procedure) hold consistently across the diverse landscape of QA benchmarks? To study this question, we introduce the notion of concurrence—two benchmarks have high concurrence on a set of modeling approaches…

2023

Estimating Large Language Model Capabilities without Labeled Test Data

EMNLP 2023long findings

Large Language Models (LLMs) have exhibited an impressive ability to perform in-context learning (ICL) from only a few examples, but the success of ICL varies widely from task to task. Thus, it is important to quickly determine whether ICL is applicable to a new task, but directly evaluating ICL acc…

Cited by 0SourcecodeScholar
2023

How Predictable Are Large Language Model Capabilities? A Case Study on BIG-bench

EMNLP 2023long findings

We investigate the predictability of large language model (LLM) capabilities: given records of past experiments using different model families, numbers of parameters, tasks, and numbers of in-context examples, can we accurately predict LLM performance on new experiment configurations? Answering this…

Cited by 0SourcecodeScholar
2023

SCENE: Self-Labeled Counterfactuals for Extrapolating to Negative Examples

EMNLP 2023long main

Detecting negatives (such as non-entailment relationships, unanswerable questions, and false claims) is an important and challenging aspect of many natural language understanding tasks. Though manually collecting challenging negative examples can help models detect them, it is both costly and domain…

Cited by 0SourcecodeScholar
2022

Analyzing Dynamic Adversarial Training Data in the Limit

ACL 2022findings

To create models that are robust across a wide range of test inputs, training datasets should include diverse examples that span numerous phenomena. Dynamic adversarial data collection (DADC), where annotators craft examples that challenge continually improving models, holds promise as an approach f…

2022

Generalization Differences between End-to-End and Neuro-Symbolic Vision-Language Reasoning Systems

EMNLP 2022finding

For vision-and-language reasoning tasks, both fully connectionist, end-to-end methods and hybrid, neuro-symbolic methods have achieved high in-distribution performance. In which out-of-distribution settings does each paradigm excel? We investigate this question on both single-image and multi-image v…

Cited by 2SourcePDFScholar
2022

Knowledge Base Question Answering by Case-based Reasoning over Subgraphs

ICML 2022spotlight

Question answering (QA) over knowledge bases (KBs) is challenging because of the diverse, essentially unbounded, types of reasoning patterns needed. However, we hypothesize in a large KB, reasoning patterns required to answer a query type reoccur for various entities in their respective subgraph nei…

2022

Models in the Loop: Aiding Crowdworkers with Generative Annotation Assistants

NAACL 2022long

In Dynamic Adversarial Data Collection (DADC), human annotators are tasked with finding examples that models struggle to predict correctly. Models trained on DADC-collected training data have been shown to be more robust in adversarial and out-of-domain settings, and are considerably harder for huma…

2022

On Continual Model Refinement in Out-of-Distribution Data Streams

ACL 2022long

Real-world natural language processing (NLP) models need to be continually updated to fix the prediction errors in out-of-distribution (OOD) data streams while overcoming catastrophic forgetting. However, existing continual learning (CL) problem setups cannot cover such a realistic and complex scena…

Cited by 26SourcePDFScholar
2022

On the Robustness of Reading Comprehension Models to Entity Renaming

NAACL 2022long

We study the robustness of machine reading comprehension (MRC) models to entity renaming—do models make more wrong predictions when the same questions are asked about an entity whose name has been changed? Such failures imply that models overly rely on entity information to answer questions, and thu…

2022

Question Answering Infused Pre-training of General-Purpose Contextualized Representations

ACL 2022findings

We propose a pre-training objective based on question answering (QA) for learning general-purpose contextual representations, motivated by the intuition that the representation of a phrase in a passage should encode all questions that the phrase can answer in context. To this end, we train a bi-enco…

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

Dynaboard: An Evaluation-As-A-Service Platform for Holistic Next-Generation Benchmarking

NeurIPS 2021poster

We introduce Dynaboard, an evaluation-as-a-service framework for hosting benchmarks and conducting holistic model comparison, integrated with the Dynabench platform. Our platform evaluates NLP models directly instead of relying on self-reported metrics or predictions on a single dataset. Under this…

Cited by 66SourcePDFScholar
2021

Evaluation Examples are not Equally Informative: How should that change NLP Leaderboards?

ACL 2021long

Leaderboards are widely used in NLP and push the field forward. While leaderboards are a straightforward ranking of NLP models, this simplicity can mask nuances in evaluation items (examples) and subjects (NLP models). Rather than replace leaderboards, we advocate a re-imagining so that they better…

2021

Improving Question Answering Model Robustness with Synthetic Adversarial Data Generation

EMNLP 2021main

Despite recent progress, state-of-the-art question answering models remain vulnerable to a variety of adversarial attacks. While dynamic adversarial data collection, in which a human annotator tries to write examples that fool a model-in-the-loop, can improve model robustness, this process is expens…

2021

Masked Language Modeling and the Distributional Hypothesis: Order Word Matters Pre-training for Little

EMNLP 2021main

A possible explanation for the impressive performance of masked language model (MLM) pre-training is that such models have learned to represent the syntactic structures prevalent in classical NLP pipelines. In this paper, we propose a different explanation: MLMs succeed on downstream tasks almost en…

2021

Swords: A Benchmark for Lexical Substitution with Improved Data Coverage and Quality

NAACL 2021long

We release a new benchmark for lexical substitution, the task of finding appropriate substitutes for a target word in a context. For writing, lexical substitution systems can assist humans by suggesting words that humans cannot easily think of. However, existing benchmarks depend on human recall as…

2017

Learning concepts through conversations in spoken dialogue systems

ICASSP 2017accepted

Spoken dialogue systems must be able to recover gracefully from unexpected user inputs. In many cases, these unexpected utterances may be within the scope of the system, but include previously unseen phrases that the system cannot interpret. In this work, we augment a spoken dialogue system with the…

Cited by 0SourceScholar