← Search

Daniel Khashabi

56 accepted papers

2026

Compute When Worth It: Risk Control for Reasoning on a Compute Budget

ICML 2026poster

Reasoning Large Language Models (LLMs) enable test-time scaling, with dataset-level accuracy improving as the token budget increases, motivating adaptive reasoning---spending tokens when they improve reliability and stopping early when additional computation is unlikely to help. However, setting the…

Cited by 0SourceScholar
2026

Controlling the Risk of Corrupted Contexts for Language Models via Early-Exiting

ICML 2026poster

Large language models (LLMs) can be influenced by harmful or irrelevant context, which can significantly harm model performance on downstream tasks. This motivates principled designs in which LLM systems include built-in mechanisms to guard against such "garbage in, garbage out" scenarios. We propos…

Cited by 0SourceScholar
2026

Linguistic Nepotism: Trading-off Quality for Language Preference in Multilingual RAG

ICML 2026spotlight

Multilingual Retrieval-Augmented Generation (mRAG) systems enable language models to answer knowledge-intensive queries with citation-supported responses across languages. Despite their growing use, an open questions is whether the mixture of different document languages impacts generation and citat…

Cited by 0SourceScholar
2026

The Alignment Waltz: Jointly Training Agents to Collaborate for Safety

ICLR 2026poster

Harnessing the power of LLMs requires a delicate dance between being helpful and harmless, leading to two critical challenges: vulnerability to adversarial attacks that elicit unsafe content, and a tendency for overrefusal on benign but sensitive prompts. Current approaches often navigate this dance…

Cited by 0SourceScholar
2026

Trust Functions: Near Lossless Weak-to-Strong Generalization by Learning to Trust the Weak Teacher

ICML 2026poster

Weak-to-strong generalization studies how to improve a strong student using supervision from a weaker teacher when reliable labels are scarce. We view this primarily as a data selection problem, where the key challenge is to identify which weak labels are reliable enough to serve as a training signa…

Cited by 0SourceScholar
2026

WoW!: World Models in a Closed-Loop World

ICLR 2026oral

Generative world models (WMs) can now simulate worlds with striking visual realism, which naturally raises the question of whether they can endow embodied agents with predictive perception for decision making. Progress on this question has been limited by fragmented evaluation: most existing benchma…

Cited by 0SourcecodeScholar
2025

Benchmarking Language Model Creativity: A Case Study on Code Generation

NAACL 2025long

As LLMs become increasingly prevalent, it is interesting to consider how “creative” these models can be. From cognitive science, creativity consists of at least two key characteristics: convergent thinking (purposefulness to achieve a given goal) and divergent thinking (adaptability to explore new e…

2025

CLAIMCHECK: How Grounded are LLM Critiques of Scientific Papers?

EMNLP 2025

A core part of scientific peer review involves providing expert critiques that directly assess the scientific claims a paper makes. While it is now possible to automatically generate plausible (if generic) reviews, ensuring that these reviews are sound and grounded in the papers’ claims remains chal

2025

Certified Mitigation of Worst-Case LLM Copyright Infringement

EMNLP 2025

The exposure of large language models (LLMs) to copyrighted material during pre-training raises concerns about unintentional copyright infringement post deployment. This has driven the development of “copyright takedown” methods—post-training approaches aimed at preventing models from generating con

2025

Controllable Safety Alignment: Inference-Time Adaptation to Diverse Safety Requirements

ICLR 2025poster

The current paradigm for safety alignment of large language models (LLMs) follows a _one-size-fits-all_ approach: the model refuses to interact with any content deemed unsafe by the model provider. This approach lacks flexibility in the face of varying social norms across cultures and regions. In ad…

Cited by 0SourcePDFScholar
2025

Core: Robust Factual Precision with Informative Sub-Claim Identification

ACL 2025finding

Hallucinations pose a challenge to the application of large language models (LLMs) thereby motivating the development of metrics to evaluate factual precision. We observe that popular metrics using the Decompose-Then-Verify framework, such as FActScore, can be manipulated by adding obvious or repeti…

2025

Evaluating the Evaluators: Are readability metrics good measures of readability?

EMNLP 2025

Plain Language Summarization (PLS) aims to distill complex documents into accessible summaries for non-expert audiences. In this paper, we conduct a thorough survey of PLS literature, and identify that the current standard practice for readability evaluation is to use traditional readability metrics

Cited by 0SourcePDFScholar
2025

FEEDBACK FRICTION: LLMs Struggle to Fully Incorporate External Feedback

NeurIPS 2025poster

Recent studies have shown LLMs possess some ability to improve their responses when given external feedback. However, it remains unclear how effectively and thoroughly these models can incorporate extrinsic feedback. In an ideal scenario, if LLMs receive near-perfect and complete feedback, we would…

Cited by 0SourceScholar
2025

ICL CIPHERS: Quantifying ”Learning” in In-Context Learning via Substitution Ciphers

EMNLP 2025

Recent works have suggested that In-Context Learning (ICL) operates in dual modes, i.e. task retrieval (remember learned patterns from pre-training) and task learning (inference-time ”learning” from demonstrations). However, disentangling these the two modes remains a challenging goal. We introduce

Cited by 0SourcePDFScholar
2025

Jailbreak Distillation: Renewable Safety Benchmarking

EMNLP 2025

Large language models (LLMs) are rapidly deployed in critical applications, raising urgent needs for robust safety benchmarking. We propose Jailbreak Distillation (JBDistill), a novel benchmark construction framework that “distills” jailbreak attacks into high-quality and easily-updatable safety ben

Cited by 0SourcePDFScholar
2025

RATIONALYST: Pre-training Process-Supervision for Improving Reasoning

ACL 2025long

The reasoning steps generated by LLMs might be incomplete, as they mimic logical leaps common in everyday communication found in their pre-training data: underlying rationales are frequently left implicit (unstated). To address this challenge, we introduce RATIONALYST, a model for process-supervisio…

2025

SELF-[IN]CORRECT: LLMs Struggle with Discriminating Self-Generated Responses

AAAI 2025technical

Can LLMs consistently improve their previous outputs for better results? For this to be true, LLMs would need to be better at discriminating among previously-generated alternatives, than generating initial responses. We explore the validity of this hypothesis in practice. We first formulate a unifie…

2025

SIMPLEMIX: Frustratingly Simple Mixing of Off- and On-policy Data in Language Model Preference Learning

ICML 2025poster

Aligning language models with human preferences relies on pairwise preference datasets. While some studies suggest that on-policy data consistently outperforms off-policy data for preference learning, others indicate that the advantages of on-policy data are task-dependent, highlighting the need for…

Cited by 0SourcePDFScholar
2025

TurkingBench: A Challenge Benchmark for Web Agents

NAACL 2025long

Can advanced multi-modal models effectively tackle complex web-based tasks? Such tasks are often found on crowdsourcing platforms, where crowdworkers engage in challenging micro-tasks within web-based environments.Building on this idea, we present TurkingBench, a benchmark consisting of tasks presen…

2025

Upsample or Upweight? Balanced Training on Heavily Imbalanced Datasets

NAACL 2025long

Data abundance across different domains exhibits a long-tailed distribution: few domains have abundant data, while most face data scarcity. Our work focuses on a multilingual setting, where available data is heavily skewed toward high-resource languages, creating significant imbalances in training d…

Cited by 0SourcePDFScholar
2025

Verifiable by Design: Aligning Language Models to Quote from Pre-Training Data

NAACL 2025long

To trust the fluent generations of large language models (LLMs), humans must be able to _verify_ their correctness against trusted, external sources. Recent efforts, such as providing citations via retrieved documents or post-hoc provenance, enhance verifiability but provide no guarantees on their c…

2025

WorldAPIs: The World Is Worth How Many APIs? A Thought Experiment

AAAI 2025technical

AI systems make decisions in physical environments through primitive actions or affordances that are accessed via API calls. While deploying AI agents in the real world involves numerous high-level actions, existing embodied simulators offer a limited set of domain-salient APIs. This naturally bring…

Cited by 2SourcePDFScholar
2024

AnaloBench: Benchmarking the Identification of Abstract and Long-context Analogies

EMNLP 2024main

Humans regularly engage in analogical thinking, relating personal experiences to current situations (X is analogous to Y because of Z). Analogical thinking allows humans to solve problems in creative ways, grasp difficult concepts, and articulate ideas more effectively. Can language models (LMs) do…

Cited by 6SourcePDFScholar
2024

DiffNorm: Self-Supervised Normalization for Non-autoregressive Speech-to-speech Translation

NeurIPS 2024poster

Non-autoregressive Transformers (NATs) are recently applied in direct speech-to-speech translation systems, which convert speech across different languages without intermediate text data. Although NATs generate high-quality outputs and offer faster inference than autoregressive models, they tend to…

2024

Efficient Large Multi-modal Models via Visual Context Compression

NeurIPS 2024poster

While significant advancements have been made in compressed representations for text embeddings in large language models (LLMs), the compression of visual tokens in multi-modal LLMs (MLLMs) has remained a largely overlooked area. In this work, we present the study on the analysis of redundancy conce…

2024

Error Norm Truncation: Robust Training in the Presence of Data Noise for Text Generation Models

ICLR 2024spotlight

Text generation models are notoriously vulnerable to errors in the training data. With the wide-spread availability of massive amounts of web-crawled data becoming more commonplace, how can we enhance the robustness of models trained on a massive amount of noisy web-crawled text? In our work, we pro…

Cited by 4SourcePDFScholar
2024

Insights into LLM Long-Context Failures: When Transformers Know but Don’t Tell

EMNLP 2024finding

Large Language Models (LLMs) exhibit positional bias, struggling to utilize information from the middle or end of long contexts. Our study explores LLMs’ long-context reasoning by probing their hidden representations. We find that while LLMs encode the position of target information, they often fail…

2024

Position: Do pretrained Transformers Learn In-Context by Gradient Descent?

ICML 2024oral

The emergence of In-Context Learning (ICL) in LLMs remains a remarkable phenomenon that is partially understood. To explain ICL, recent studies have created theoretical connections to Gradient Descent (GD). We ask, do such connections hold up in actual pre-trained language models? We highlight the l…

Cited by 2SourcePDFScholar
2024

RORA: Robust Free-Text Rationale Evaluation

ACL 2024long

Free-text rationales play a pivotal role in explainable NLP, bridging the knowledge and reasoning gaps behind a model’s decision-making. However, due to the diversity of potential reasoning paths and a corresponding lack of definitive ground truth, their evaluation remains a challenge. Existing metr…

2024

SemStamp: A Semantic Watermark with Paraphrastic Robustness for Text Generation

NAACL 2024long

Existing watermarked generation algorithms employ token-level designs and therefore, are vulnerable to paraphrase attacks. To address this issue, we introduce watermarking on the semantic representation of sentences. We propose SemStamp, a robust sentence-level semantic watermarking algorithm that u…

2024

The Language Barrier: Dissecting Safety Challenges of LLMs in Multilingual Contexts

ACL 2024findings

As the influence of large language models (LLMs) spans across global communities, their safety challenges in multilingual settings become paramount for alignment research. This paper examines the variations in safety challenges faced by LLMs across different languages and discusses approaches to all…

Cited by 55SourcePDFScholar
2024

The Trickle-down Impact of Reward Inconsistency on RLHF

ICLR 2024poster

Standard practice within Reinforcement Learning from Human Feedback (RLHF) involves optimizing against a Reward Model (RM), which itself is trained to reflect human preferences for desirable generations. A notable subject that is understudied is the (in-)consistency of RMs --- whether they can recog…

2024

k-SemStamp: A Clustering-Based Semantic Watermark for Detection of Machine-Generated Text

ACL 2024findings

Recent watermarked generation algorithms inject detectable signatures during language generation to facilitate post-hoc detection. While token-level watermarks are vulnerable to paraphrase attacks, SemStamp (Hou et al., 2023) applies watermark on the semantic representation of sentences and demonstr…

2023

Flatness-Aware Prompt Selection Improves Accuracy and Sample Efficiency

EMNLP 2023long findings

With growing capabilities of large language models, prompting them has become the dominant way to access them. This has motivated the development of strategies for automatically selecting effective language prompts. In this paper, we introduce **pFlat** (prompt flatness), a new metric to quantify…

Cited by 0SourcecodeScholar
2023

Generating Sequences by Learning to Self-Correct

ICLR 2023poster

Sequence generation applications require satisfying semantic constraints, such as ensuring that programs are correct, using certain keywords, or avoiding undesirable content. Language models, whether fine-tuned or prompted with few-shot demonstrations, frequently violate these constraints, and lack…

Cited by 122SourcePDFScholar
2023

Representation Projection Invariance Mitigates Representation Collapse

EMNLP 2023long findings

Fine-tuning contextualized representations learned by pre-trained language models remains a prevalent practice in NLP. However, fine-tuning can lead to representation degradation (also known as representation collapse), which may result in instability, sub-optimal performance, and weak generalizatio…

Cited by 0SourceScholar
2023

Self-Instruct: Aligning Language Models with Self-Generated Instructions

ACL 2023long

Large “instruction-tuned” language models (i.e., finetuned to respond to instructions) have demonstrated a remarkable ability to generalize zero-shot to new tasks. Nevertheless, they depend heavily on human-written instruction data that is often limited in quantity, diversity, and creativity, theref…

2023

The Tail Wagging the Dog: Dataset Construction Biases of Social Bias Benchmarks

ACL 2023short

How reliably can we trust the scores obtained from social bias benchmarks as faithful indicators of problematic social biases in a given model? In this work, we study this question by contrasting social biases with non-social biases that stem from choices made during dataset construction (which migh…

2023

When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memories

ACL 2023long

Despite their impressive performance on diverse tasks, large language models (LMs) still struggle with tasks requiring rich world knowledge, implying the difficulty of encoding a wealth of world knowledge in their parameters. This paper aims to understand LMs’ strengths and limitations in memorizing…

2022

COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamics

NeurIPS 2022accept

Many applications of text generation require incorporating different constraints to control the semantics or style of generated text. These constraints can be hard (e.g., ensuring certain keywords are included in the output) and soft (e.g., contextualizing the output with the left- or right-hand con…

2022

Cross-Task Generalization via Natural Language Crowdsourcing Instructions

ACL 2022long

Humans (e.g., crowdworkers) have a remarkable ability in solving different tasks, by simply reading textual instructions that define them and looking at a few examples. Despite the success of the conventional supervised learning on individual datasets, such models often struggle with generalization…

Cited by 697SourcePDFScholar
2022

GENIE: Toward Reproducible and Standardized Human Evaluation for Text Generation

EMNLP 2022main

While often assumed a gold standard, effective human evaluation of text generation remains an important, open area for research.We revisit this problem with a focus on producing consistent evaluations that are reproducible—over time and across different populations. We study this goal in different s…

2022

Hey AI, Can You Solve Complex Tasks by Talking to Agents?

ACL 2022findings

Training giant models from scratch for each complex task is resource- and data-inefficient. To help develop models that can leverage existing systems, we propose a new challenge: Learning to solve complex tasks by communicating with existing agents (or models) in natural language. We design a synthe…

2022

NeuroLogic A*esque Decoding: Constrained Text Generation with Lookahead Heuristics

NAACL 2022long

The dominant paradigm for neural text generation is left-to-right decoding from autoregressive language models. Constrained or controllable generation under complex lexical constraints, however, requires foresight to plan ahead feasible future paths. Drawing inspiration from the A* search algorithm,…

2022

Prompt Waywardness: The Curious Case of Discretized Interpretation of Continuous Prompts

NAACL 2022long

Fine-tuning continuous prompts for target tasks has recently emerged as a compact alternative to full model fine-tuning. Motivated by these promising results, we investigate the feasibility of extracting a discrete (textual) interpretation of continuous prompts that is faithful to the problem they s…

2022

ProsocialDialog: A Prosocial Backbone for Conversational Agents

EMNLP 2022main

Most existing dialogue systems fail to respond properly to potentially unsafe user utterances by either ignoring or passively agreeing with them. To address this issue, we introduce ProsocialDialog, the first large-scale multi-turn dialogue dataset to teach conversational agents to respond to proble…

2022

Reframing Instructional Prompts to GPTk’s Language

ACL 2022findings

What kinds of instructional prompts are easier to follow for Language Models (LMs)? We study this question by conducting extensive empirical analysis that shed light on important features of successful instructional prompts. Specifically, we study several classes of reframing techniques for manual r…

Cited by 215SourcePDFScholar
2022

Time Waits for No One! Analysis and Challenges of Temporal Misalignment

NAACL 2022long

When an NLP model is trained on text data from one time period and tested or deployed on data from another, the resulting temporal misalignment can degrade end-task performance. In this work, we establish a suite of eight diverse tasks across different domains (social media, science papers, news, an…

Cited by 93SourcePDFScholar
2021

GooAQ: Open Question Answering with Diverse Answer Types

EMNLP 2021finding

While day-to-day questions come with a variety of answer types, the current question-answering (QA) literature has failed to adequately address the answer diversity of questions. To this end, we present GooAQ, a large-scale dataset with a variety of answer types. This dataset contains over 5 million…

2021

Text Modular Networks: Learning to Decompose Tasks in the Language of Existing Models

NAACL 2021long

We propose a general framework called Text Modular Networks(TMNs) for building interpretable systems that learn to solve complex tasks by decomposing them into simpler ones solvable by existing models. To ensure solvability of simpler tasks, TMNs learn the textual input-output behavior (i.e., langua…

2020

TransOMCS: From Linguistic Graphs to Commonsense Knowledge

IJCAI 2020poster

Commonsense knowledge acquisition is a key problem for artificial intelligence. Conventional methods of acquiring commonsense knowledge generally require laborious and costly human annotations, which are not feasible on a large scale. In this paper, we explore a practical way of mining commonsense k…