← Search

Zejiang Shen

7 accepted papers

2025

Position: Retrieval-augmented systems can be dangerous medical communicators

ICML 2025poster

Patients have long sought health information online, and increasingly, they are turning to generative AI to answer their health-related queries. Given the high stakes of the medical domain, techniques like retrieval-augmented generation and citation grounding have been widely promoted as methods to…

Cited by 0SourcePDFScholar
2025

SelfCite: Self-Supervised Alignment for Context Attribution in Large Language Models

ICML 2025poster

We introduce SelfCite, a novel self-supervised approach that aligns LLMs to generate high-quality, fine-grained, sentence-level citations for the statements in their generated responses. Instead of only relying on costly and labor-intensive annotations, SelfCite leverages a reward signal provided b…

2024

Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Research

ACL 2024long

Information about pretraining corpora used to train the current best-performing language models is seldom discussed: commercial models rarely detail their data, and even open models are often released without accompanying training data or recipes to reproduce them. As a result, it is challenging to…

2024

Learning to Decode Collaboratively with Multiple Language Models

ACL 2024long

We propose a method to teach multiple large language models (LLM) to collaborate by interleaving their generations at the token level. We model the decision of which LLM generates the next token as a latent variable. By optimizing the marginal likelihood of a training set under our latent variable m…

2023

American Stories: A Large-Scale Structured Text Dataset of Historical U.S. Newspapers

NeurIPS 2023poster

Existing full text datasets of U.S. public domain newspapers do not recognize the often complex layouts of newspaper scans, and as a result the digitized content scrambles texts from articles, headlines, captions, advertisements, and other layout regions. OCR quality can also be low. This study deve…

Cited by 29SourcePDFScholar
2023

Are Layout-Infused Language Models Robust to Layout Distribution Shifts? A Case Study with Scientific Documents

ACL 2023findings

Recent work has shown that infusing layout features into language models (LMs) improves processing of visually-rich documents such as scientific papers. Layout-infused LMs are often evaluated on documents with familiar layout features (e.g., papers from the same publisher), but in practice models en…

2022

Multi-LexSum: Real-world Summaries of Civil Rights Lawsuits at Multiple Granularities

NeurIPS 2022accept

With the advent of large language models, methods for abstractive summarization have made great strides, creating potential for use in applications to aid knowledge workers processing unwieldy document collections. One such setting is the Civil Rights Litigation Clearinghouse (CRLC, https://clearing…