Cite Pretrain: Retrieval-Free Knowledge Attribution for Large Language Models
cs.AI, cs.CL, cs.LG
Submitted: 2025-06-21
Updated: 2026-10-11
Code: https://github.com/kkkevinkkkkk/CitePretrain
License: http://creativecommons.org/licenses/by/4.0/
The gist: Large language models can answer questions correctly and still hallucinate their citations.
Terminology
Abstract
Large language models can answer questions correctly and still hallucinate their citations. Most citation systems address this by retrieving documents at inference time. Retrieval adds latency and depends on external infrastructure. Retrieval noise can also affect the answer. We study whether LLMs can cite documents from their (continual) pretraining without test-time retrieval. We introduce CitePretrainBench to evaluate this setting. It includes Wikipedia, Common Crawl, and arXiv as real-world corpora and also includes novel, unseen documents. The benchmark includes short-form citation tasks for a single fact and long-form tasks for multiple facts. Continual-pretraining first binds factual knowledge to a persistent identifier for each document. Instruction tuning then teaches the model to put these identifiers in its answers as citations. Passive Indexing appends an identifier at the end of each document. Synthetic data used in Active Indexing restate facts in diverse, compositional forms. We train two mappings together. These mappings are source to fact and fact to source. The first direction involves the model generating content given a cited source. In the second, it learns to attribute its own answers when facts are paraphrased or combined. On Qwen-2.5-7B and 3B, Active Indexing improves citation precision over Passive Indexing on every evaluated citation task, with gains of up to 30.2 percentage points. In the scaling ablation, citation performance continues to improve as we add augmented data, including at 16 times the original token count. Combining internal and external citations also improves robustness to retrieval noise.
Sources
- Generative Retrieval with Few-shot Indexing
- CopyBench: Measuring Literal and Non-Literal Reproduction of Copyright-Protected Text in Language Model Generation
- SelfCite: Self-Supervised Alignment for Context Attribution in Large Language Models
- Rethinking with Retrieval: Faithful Large Language Model Inference
- Training Language Models to Generate Text with Citations via Fine-grained Rewards
- HAGRID: A Human-LLM Collaborative Dataset for Generative Information-Seeking with Attribution
- UniGen: A Unified Generative Framework for Retrieval and Question Answering with Large Language Models
- Teaching language models to support answers with verified quotes
- WebGPT: Browser-assisted question-answering with human feedback
- How Context Affects Language Models' Factual Predictions
- The Web Is Your Oyster - Knowledge-Intensive NLP against a Very Large Web Corpus
- A Survey of Query Optimization in Large Language Models
- SciQAG: A Framework for Auto-Generated Science Question Answering Dataset with Fine-grained Evaluation
- Survey on Factuality in Large Language Models: Knowledge, Retrieval and Domain-Specificity
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems