← Search

Rulin Shao

16 accepted papers

2026

DS SERVE: A Framework for Efficient and Scalable Neural Retrieval

AAAI 2026technical

We present DS SERVE, a framework that transforms large-scale text datasets—comprising half a trillion tokens—into a high-performance neural retrieval system. DS SERVE offers both a web interface and API endpoints, achieving low latency with modest memory overhead on a single node. The framework also

Cited by 0SourcePDFScholar
2026

Frustratingly Simple Retrieval Improves Challenging, Reasoning-Intensive Benchmarks

ICLR 2026poster

Retrieval augmentation has primarily been studied in limited settings, such as factoid question answering; more challenging, reasoning-intensive benchmarks have seen limited success from minimal RAG. In this work, we challenge this prevailing view on a set of established, reasoning-intensive benchma…

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

Reinforcement Learning with Evolving Rubrics for Deep Research

ICML 2026oral

Deep research agents perform multi-step research to produce long-form, well-attributed answers. However, most open deep research agents are trained on easily verifiable short-form QA tasks via reinforcement learning with verifiable rewards, which does not extend to realistic long-form tasks. We addr…

Cited by 0SourceScholar
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
2026

Spurious Rewards: Rethinking Training Signals in RLVR

ICML 2026poster

We show that reinforcement learning with verifiable rewards (RLVR) can elicit strong mathematical reasoning in certain language models even with spurious rewards that have little, no, or outright negative correlation with the correct answer. For example, RLVR training with GRPO improves MATH-500 per…

Cited by 0SourcecodeScholar
2026

TruthRL: Incentivizing Truthful LLMs via Reinforcement Learning

ICML 2026poster

While large language models (LLMs) have demonstrated strong performance on factoid question answering, they are still prone to hallucination and untruthful responses, particularly when tasks demand information outside their parametric knowledge. Indeed, truthfulness requires more than accuracy---mod…

Cited by 0SourceScholar
2025

Improving Factuality with Explicit Working Memory

ACL 2025long

Large language models can generate factually inaccurate content, a problem known as hallucination. Recent works have built upon retrieved-augmented generation to improve factuality through iterative prompting but these methods are limited by the traditional RAG design. To address these challenges, w…

2025

Language models scale reliably with over-training and on downstream tasks

ICLR 2025poster

Scaling laws are useful guides for derisking expensive training runs, as they predict performance of large models using cheaper, small-scale experiments. However, there remain gaps between current scaling studies and how language models are ultimately trained and evaluated. For instance, scaling is…

2025

LongLeader: A Comprehensive Leaderboard for Large Language Models in Long-context Scenarios

NAACL 2025long

Large Language Models (LLMs), exemplified by Claude and LLama, have exhibited impressive proficiency in tackling a myriad of Natural Language Processing (NLP) tasks. Yet, in pursuit of the ambitious goal of attaining Artificial General Intelligence (AGI), there remains ample room for enhancing LLM c…

Cited by 0SourcePDFScholar
2024

DataComp-LM: In search of the next generation of training sets for language models

NeurIPS 2024poster

We introduce DataComp for Language Models, a testbed for controlled dataset experiments with the goal of improving language models. As part of DCLM, we provide a standardized corpus of 240T tokens extracted from Common Crawl, effective pretraining recipes based on the OpenLM framework, and a broad s…

Cited by 64SourcePDFScholar
2024

Scaling Retrieval-Based Language Models with a Trillion-Token Datastore

NeurIPS 2024poster

Scaling laws with respect to the amount of training data and the number of parameters allow us to predict the cost-benefit trade-offs of pretraining language models (LMs) in different configurations. In this paper, we consider another dimension of scaling: the amount of data available at inference t…

2024

Vision-Flan: Scaling Human-Labeled Tasks in Visual Instruction Tuning

ACL 2024findings

Despite vision-language models’ (VLMs) remarkable capabilities as versatile visual assistants, two substantial challenges persist within the existing VLM frameworks: (1) lacking task diversity in pretraining and visual instruction tuning, and (2) annotation error and bias in GPT-4 synthesized instru…

Cited by 34SourcePDFScholar
2023

Cross-modal Attention Congruence Regularization for Vision-Language Relation Alignment

ACL 2023long

Despite recent progress towards scaling up multimodal vision-language models, these models are still known to struggle on compositional generalization benchmarks such as Winoground. We find that a critical component lacking from current vision-language models is relation-level alignment: the ability…

Cited by 13SourcePDFScholar
2023

MPCFORMER: FAST, PERFORMANT AND PRIVATE TRANSFORMER INFERENCE WITH MPC

ICLR 2023top-25%

Enabling private inference is crucial for many cloud inference services that are based on Transformer models. However, existing private inference solutions can increase the inference latency by more than 60$\times$ or significantly compromise the inference quality. In this paper, we design the frame…

2023

VisIT-Bench: A Dynamic Benchmark for Evaluating Instruction-Following Vision-and-Language Models

NeurIPS 2023poster

We introduce VisIT-Bench (Visual InsTruction Benchmark), a benchmark for evaluating instruction-following vision-language models for real-world use. Our starting point is curating 70 "instruction families" that we envision instruction tuned vision-language models should be able to address. Extending…