WRAP++: Web discoveRy Amplified Pretraining
Abstract
Synthetic data rephrasing has emerged as a powerful technique for enhancing knowledge acquisition during large language model (LLM) pretraining. However, existing approaches operate at the single-document level, rewriting individual web pages in isolation. This confines synthesized examples to intra-document knowledge, missing cross-document relationships and leaving facts with limited associative context. We propose WRAP++ (Web discoveRy Amplified Pretraining), which amplifies the associative context of factual knowledge by discovering cross-document relationships from web hyperlinks and synthesizing joint QA over each discovered document pair. Concretely, WRAP++ discovers high-confidence relational motifs including dual-links and co-mentions, and synthesizes QA that requires reasoning across both documents. This produces relational knowledge absent from either source document alone, creating diverse entry points to the same facts. Because the number of valid entity pairs grows combinatorially, this discovery-driven synthesis also amplifies data scale far beyond single-document rewriting. Instantiating WRAP++ on Wikipedia, we amplify ~8.4B tokens of raw text into 80B tokens of cross-document QA data. On SimpleQA, OLMo-based models at both 7B and 32B scales trained with WRAP++ substantially outperform single-document approaches and exhibit sustained scaling gains, underscoring the advantage of cross-document knowledge discovery and amplification.
Figures & tables
| Data Recipe | OLMo-3-7B | OLMo-3-32B |
|---|---|---|
| Pretrained Base | 34.76 | 42.35 |
| + WRAP | 39.55 | 44.43 |
| + Extended WRAP | 43.69 | 47.91 |
| + WRAP++ | 49.13 | 53.97 |
| Ablation Axis | Variant | pass@128 |
| Pairing Strategy (§ 4.1 ) | Random entity pairing | 43.46 |
| Topological relation discovery | 45.11 | |
| Relation Topology (§ 4.2 ) | Dual-link only | 44.24 |
| Co-mention only | 44.36 | |
| Combined | 45.11 | |
| Synthesis Method (§ 4.3 ) | Raw concatenation (no QA) | 35.43 |
| Single-Document Component | Proportion of WRAP++ in Mixture | |
|---|---|---|
| 0% (Baseline Only) | 50% (1:1 Mix) | |
| Pretrained Model (No further training) | 34.76 | |
| Raw FineWiki | 39.23 | 41.80 |
| Distill | 38.12 | 41.59 |
| Extract Knowledge | 38.69 | 43.35 |
| Knowledge List | 38.60 | 42.03 |
| Setting | SimpleQA | Gen. Avg |
|---|---|---|
| pass@128 | (12 tasks) | |
| Pretrained Base | 34.76 | 57.79 |
| + Midtrain (100B) | 34.74 | 68.24 |
| + WRAP++ Mix (100B) | 37.58 | 68.16 |
Appendix figures & tables15 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | OLMo-3-7B (8B) | OLMo-3-7B (80B) | OLMo-3-32B |
|---|---|---|---|
| Sequence length | 8,192 | 8,192 | 8,192 |
| Global batch size (tokens) | 2M ( ) | 2M ( ) | 4M ( ) |
| Peak learning rate | |||
| LR schedule | Linear 0 | Linear 0 | Linear 0 |
| Warmup steps | 0 | 0 | 0 |
| Training steps | 4,000 | 40,000 | 10,000 |
| Statistic | Value |
|---|---|
| Source corpus | FineWiki (English) |
| Raw corpus tokens | 8.4B |
| Wikipedia articles processed | 6.7M |
| Dual-link pairs ( ) | 9.6M |
| Co-mention pairs ( , ) | 232M |
| Dual-link synthesized tokens | 3B |
| Category | Benchmark | Base | Midtrain | WRAP++ Mix |
|---|---|---|---|---|
| General Knowledge | ARC | 77.47 | 84.98 | 85.07 |
| MMLU Redux | 60.30 | 65.30 | 66.58 | |
| Reading Comprehension | SQuAD v2 | 42.27 | 48.59 | 49.99 |
| DROP | 40.51 | 66.51 | 65.36 | |
| Reasoning | OpenBookQA | 76.40 | 85.20 | 82.40 |
| CSQA | 72.15 | 74.20 | 72.97 |
| Co-mention Variant | SimpleQA pass@8 |
|---|---|
| 2-doc (entities , only) | 15.74 |
| 3-doc (entities , + bridge ) | 15.42 |
| Relation Type | QA Instances | Proportion |
|---|---|---|
| Co-mention | 231,292,954 | 96.1% |
| Dual-link | 9,365,111 | 3.9% |
| Total | 240,658,065 | 100.0% |
| Field | Mean | Std | Min | P5 | P25 | Median | P75 | P95 |
|---|---|---|---|---|---|---|---|---|
| Character-level | ||||||||
| Question | 207 | 43 | 45 | 143 | 177 | 203 | 233 | 283 |
| Answer | 1,424 | 624 | 139 | 563 | 938 | 1,386 | 1,817 | 2,463 |
| QA (combined) | 1,651 | 637 | 242 | 766 | 1,161 | 1,610 | 2,051 | 2,716 |
| Word-level | ||||||||
| Question | 33 | 7 | 7 | 23 | 28 | 32 | 37 | 45 |
| Relation Type | Question | Answer | QA Total |
|---|---|---|---|
| Co-mention | 204 (32 words) | 1,394 (213 words) | 1,618 |
| Dual-link | 197 (32 words) | 1,210 (189 words) | 1,428 |
| Overall | 203 (32 words) | 1,386 (212 words) | 1,610 |
| Length Range (chars) | QA (combined) | Answer only | ||
|---|---|---|---|---|
| Count | % | Count | % | |
| 0 | 0.0 | 2,479 | ||
| 456,399 | 0.2 | 6,771,177 | 2.8 | |
| 38,609,776 | 16.0 | 61,411,377 | 25.5 | |
| 135,127,249 | 56.1 | 132,242,235 | 55.0 | |
| 66,336,557 | 27.6 | 40,114,788 | 16.7 | |
| WRAP (single-document QA on entity A) | WRAP++ (cross-document QA on entities A+B) |
| Q: Who is Ludwig Göransson? A: Ludwig Göransson is a Swedish composer known for film scores. Q: What films has Göransson scored? A: He has scored Black Panther , Tenet , and Oppenheimer . | Q: What films earned Nolan’s original Tenet composer and his replacement their second Oscars? A: Hans Zimmer turned down Tenet for Dune, which earned his second Oscar (after The Lion King). His replacement, Ludwig Göransson, later won his second Oscar for Oppenheimer (after Black Panther). Therefore, the answer is Dune and Oppenheimer. |
Knowledge Augmentation for Language Models
Knowledge augmentation for language models adds external or newly learned information to improve factuality, reasoning, and domain performance.
See all 178 papers