← Search

Xilun Chen

21 accepted papers

2026

Learning Facts at Scale with Active Reading

ICLR 2026poster

LLMs are known to store vast amounts of knowledge in their parametric memory. However, learning and recalling facts from this memory is known to be unreliable, depending largely on the prevalence of particular facts in the training data and other factors which are poorly understood. Practitioners ar…

Cited by 0SourceScholar
2026

Learning to Reason for Factuality

ICML 2026poster

Reasoning Large Language Models (R-LLMs) have significantly advanced complex reasoning tasks but often struggle with factuality, generating substantially more hallucinations than their non-reasoning counterparts on long-form factuality benchmarks. However, extending online Reinforcement Learning (RL…

Cited by 20SourceScholar
2026

SCRIBES: Web-Scale Script-Based Semi-Structured Data Extraction with Reinforcement Learning

ICLR 2026poster

Semi-structured content in HTML tables, lists, and infoboxes accounts for a substantial share of factual data on the web, yet the formatting complicates usage, and reliably extracting structured information from them remains challenging. Existing methods either lack generalization or are resource-in…

Cited by 0SourceScholar
2025

DRAMA: Diverse Augmentation from Large Language Models to Smaller Dense Retrievers

ACL 2025long

Large language models (LLMs) have demonstrated strong effectiveness and robustness when fine-tuned as dense retrievers.However, their large parameter size presents significant computational challenges at inference time.While smaller retrievers offer better efficiency, they often fail to generalize e…

2025

Extracting and Understanding the Superficial Knowledge in Alignment

NAACL 2025long

Alignment of large language models (LLMs) with human values and preferences, often achieved through fine-tuning based on human feedback, is essential for ensuring safe and responsible AI behaviors. However, the process typically requires substantial data and computation resources. Recent studies hav…

2024

FLAME : Factuality-Aware Alignment for Large Language Models

NeurIPS 2024poster

Alignment is a procedure to fine-tune pre-trained large language models (LLMs) to follow natural language instructions and serve as helpful AI assistants. We have observed, however, that the conventional alignment process fails to enhance the factual accuracy of LLMs, and often leads to the generat…

Cited by 24SourcePDFScholar
2024

Nearest Neighbor Speculative Decoding for LLM Generation and Attribution

NeurIPS 2024poster

Large language models (LLMs) often hallucinate and lack the ability to provide attribution for their generations. Semi-parametric LMs, such as kNN-LM, approach these limitations by refining the output of an LM for a given prompt using its nearest neighbor matches in a non-parametric data store. Howe…

2024

RA-DIT: Retrieval-Augmented Dual Instruction Tuning

ICLR 2024poster

Retrieval-augmented language models (RALMs) improve performance by accessing long-tail and up-to-date knowledge from external data stores, but are challenging to build. Existing approaches require either expensive retrieval-specific modifications to LM pre-training or use post-hoc integration of the…

Cited by 142SourcePDFScholar
2023

A Study on the Efficiency and Generalization of Light Hybrid Retrievers

ACL 2023short

Hybrid retrievers can take advantage of both sparse and dense retrievers. Previous hybrid retrievers leverage indexing-heavy dense retrievers. In this work, we study “Is it possible to reduce the indexing memory of hybrid retrievers without sacrificing performance”? Driven by this question, we lever…

Cited by 8SourcePDFScholar
2023

CITADEL: Conditional Token Interaction via Dynamic Lexical Routing for Efficient and Effective Multi-Vector Retrieval

ACL 2023long

Multi-vector retrieval methods combine the merits of sparse (e.g. BM25) and dense (e.g. DPR) retrievers and have achieved state-of-the-art performance on various retrieval tasks. These methods, however, are orders of magnitude slower and need much more space to store their indices compared to their…

2023

Hierarchical Video-Moment Retrieval and Step-Captioning

CVPR 2023poster

There is growing interest in searching for information from large video corpora. Prior works have studied relevant tasks, such as text-based video retrieval, moment retrieval, video summarization, and video captioning in isolation, without an end-to-end setup that can jointly search from video corpo…

2023

How to Train Your Dragon: Diverse Augmentation Towards Generalizable Dense Retrieval

EMNLP 2023long findings

Various techniques have been developed in recent years to improve dense retrieval (DR), such as unsupervised contrastive learning and pseudo-query generation. Existing DRs, however, often suffer from effectiveness tradeoffs between supervised and zero-shot retrieval, which some argue was due to the…

Cited by 0SourcecodeScholar
2023

Nonparametric Masked Language Modeling

ACL 2023findings

Existing language models (LMs) predict tokens with a softmax over a finite vocabulary, which can make it difficult to predict rare tokens or phrases. We introduce NPM, the first nonparametric masked language model that replaces this softmax with a nonparametric distribution over every phrase in a re…

2023

Task-aware Retrieval with Instructions

ACL 2023findings

We study the problem of retrieval with instructions, where users provide explicit descriptions of their intent along with their queries to guide a retrieval system. Our solution is a general-purpose task-aware retrieval system, trained using multi-task instruction tuning and can follow human-written…

2022

CCQA: A New Web-Scale Question Answering Dataset for Model Pre-Training

NAACL 2022findings

We propose a novel open-domain question-answering dataset based on the Common Crawl project. With a previously unseen number of around 130 million multilingual question-answer pairs (including about 60 million English data-points), we use our large-scale, natural, diverse and high-quality corpus to…

2022

Domain-matched Pre-training Tasks for Dense Retrieval

NAACL 2022findings

Pre-training on larger datasets with ever increasing model size isnow a proven recipe for increased performance across almost all NLP tasks.A notable exception is information retrieval, where additional pre-traininghas so far failed to produce convincing results. We show that, with theright pre-trai…

2022

Salient Phrase Aware Dense Retrieval: Can a Dense Retriever Imitate a Sparse One?

EMNLP 2022finding

Despite their recent popularity and well-known advantages, dense retrievers still lag behind sparse methods such as BM25 in their ability to reliably match salient phrases and rare entities in the query and to generalize to out-of-domain data. It has been argued that this is an inherent limitation o…

2022

Simple Local Attentions Remain Competitive for Long-Context Tasks

NAACL 2022long

Many NLP tasks require processing long contexts beyond the length limit of pretrained models. In order to scale these models to longer text sequences, many efficient long-range attention variants have been proposed. Despite the abundance of research along this direction, it is still difficult to gau…

2022

UniK-QA: Unified Representations of Structured and Unstructured Knowledge for Open-Domain Question Answering

NAACL 2022findings

We study open-domain question answering with structured, unstructured and semi-structured knowledge sources, including text, tables, lists and knowledge bases. Departing from prior work, we propose a unifying approach that homogenizes all sources by reducing them to text and applies the retriever-re…

2021

Learning Better Structured Representations Using Low-rank Adaptive Label Smoothing

ICLR 2021poster

Training with soft targets instead of hard targets has been shown to improve performance and calibration of deep neural networks. Label smoothing is a popular way of computing soft targets, where one-hot encoding of a class is smoothed with a uniform distribution. Owing to its simplicity, label smoo…

Cited by 21SourcePDFScholar
2021

Muppet: Massive Multi-task Representations with Pre-Finetuning

EMNLP 2021main

We propose pre-finetuning, an additional large-scale learning stage between language model pre-training and fine-tuning. Pre-finetuning is massively multi-task learning (around 50 datasets, over 4.8 million total labeled examples), and is designed to encourage learning of representations that genera…