REVIEW 2 major objections 5 minor 1 cited by
The paper argues that locating knowledge in language-model training data is best done by entity identifier, not by text string, and builds a full suite — an entity-annotated Wikipedia corpus, a QID-based retrieval index, and 12 models with
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
LMEnt is an open suite of entity-annotated Wikipedia data, 12 pretrained LMs, and an entity-QID retrieval index that substantially outperforms string-based search and enables per-step knowledge attribution.
T0 review reviewed 2026-08-05 challenge →
load-bearing objection A carefully engineered entity-annotated corpus and 12 models are a real contribution, but the 80.4% retrieval headline rests on approximated baselines and should be softened. the 2 major comments →
LMEnt: A Suite for Analyzing Knowledge in Language Models from Pretraining Data to Representations
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
Core claim
On its own terms, the paper establishes that entity-grounded bookkeeping of pretraining data changes what can be said about knowledge in language models. It builds a corpus of 10.5M chunks annotated with 400M mentions of 7.3M entities, where each mention carries up to three confidence scores (from hyperlinks, an entity linker, and coreference resolution) pointing to a Wikidata QID, and it shows that retrieving chunks by QID outperforms the string-matching approach of prior tools on 66.3–80.4% of a stratified sample of 1,000 entities, with precision at depth staying above 97% where string methods drop to 84% (case-sensitive) and 27% (case-insensitive). It further shows that models trained on
What carries the argument
The load-bearing object is the entity-mention annotation: every token span in every chunk is mapped to a ranked list of candidate Wikidata QIDs, each backed by up to three scores — H from Wikipedia hyperlinks (binary and most trusted), EL from the ReFinED entity linker (model confidence), and C/CC from Maverick coreference clusters, where a longest-common-substring similarity weights which entity a cluster supports. Retrieval then queries an Elasticsearch index by QID with score thresholds (H=1, EL≥0.6, C≥0.6), and each chunk carries its training-step ID, so every retrieved chunk maps to the exact step at which a model saw it. The three-source design is what lets an indirect mention like 'th
Load-bearing premise
The headline win-rate claim rests on the paper's own simplified stand-ins for the string-matching tools WIMBD and Infini-gram, which are never actually run; if those stand-ins overstate how noisy the real tools are, the 66–80% advantage is not established.
What would settle it
Run the actual WIMBD and Infini-gram code on the same 1,000-entity stratified test set with the same LLM judge and sampling protocol, and compare per-entity win rates and precision-at-depth curves; if the genuine tools match or beat LMEnt on most entities, the central retrieval claim fails. A cheaper check: have human annotators re-judge a random sample of 200 chunks scored by the LLM judge, to verify the reported above-97% precision is not an artifact of the judge.
If this is right
- Any question of the form 'did the model see entity X, and at which step?' becomes answerable for 7.3M entities and 10.5M chunks, at training-step granularity.
- Entity-based retrieval holds precision above 97% even when up to 100K chunks are retrieved per entity, while case-insensitive string search falls to 27%; deeper retrieval is only trustworthy for the entity-based index.
- Models trained on a 3.6B-token Wikipedia-only corpus reach knowledge-recall accuracy on popular PopQA entities comparable to Pythia-1.4B and OLMo-1B, indicating that knowledge-dense data can substitute for a large share of scale.
- Fact frequency correlates with net knowledge gain, but both learning and forgetting rates increase with frequency, so frequency alone does not explain when knowledge is acquired; the released checkpoints put that question within reach.
- The entity-to-step mapping enables controlled causal interventions in pretraining data — reordering, upweighting, or editing mentions — whose effects on downstream knowledge can be measured directly.
Where Pith is reading between the lines
- The paper does not test that the Subject+Answer co-occurrence count, shown to track accuracy better than pageview popularity, is itself a training signal; reordering or upweighting chunks where subject and answer co-occur could become a cheap fact-recall intervention, testable directly on the released corpus.
- The finding that forgetting as well as learning rises with fact frequency suggests spaced-repetition-style deduplication of repeated entity mentions might reduce net forgetting; the released 1,000-step checkpoints make this measurable without any new training runs.
- Because every chunk maps to the exact training step that introduced it, the suite doubles as a contamination audit for knowledge benchmarks: one could check whether a PopQA answer is 'learned' only after the model saw the answer chunk, an application the paper does not discuss.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces LMEnt, a suite for studying knowledge acquisition in LMs. It comprises (1) a Wikipedia-based pretraining corpus annotated with fine-grained entity mentions derived from hyperlinks, entity linking, and coreference resolution; (2) an Elasticsearch index for retrieving chunks by Wikidata QID; and (3) twelve pretrained OLMo-2-style models (170M, 600M, 1B) with checkpoints. The main empirical claims are that entity-based retrieval outperforms string-based search methods similar to WIMBD and Infini-gram on 66.3%–80.4% of entities at high precision, that the models achieve knowledge-recall performance comparable to larger open models, and that a learning-dynamics analysis shows fact frequency correlates with learning and forgetting. The authors release the corpus, annotations, models, and code.
Significance. If the retrieval and resource claims hold, LMEnt would be a valuable community asset for studying knowledge acquisition, editing, attribution, and interpretability. The engineering is careful in several respects: the annotation pipeline includes an error analysis on 112 sampled mentions, the LLM-as-a-judge is validated with the alternative annotator test, and retrieval thresholds are tuned on a dev set before evaluation. The release of pretraining data with entity-level traceability and 4K checkpoints is a substantive contribution. However, the headline quantitative claim about outperforming previous retrieval tools is not yet established because the comparison class consists of hand-rolled approximations, and the win-rate protocol measures a fixed-sample precision contest rather than total relevant-chunk counts. The resource itself is promising, but the central retrieval claim needs to be either re-run against the actual tools or properly qualified.
major comments (2)
- [§5.2, Fig. 5, Abstract] The strongest quantitative claim, 'outperforms previous approaches by as much as 80.4%,' is not established by the experiments as presented. Section 5.2 states that the CS-SS Canonical baseline 'approximates' Infini-gram and CI-SS Canonical 'approximates' WIMBD, but the actual tools are never run. The approximations are not faithful in ways that matter: Infini-gram is an arbitrary n-gram matcher, so a query for 'Buffalo Bills' also surfaces unigrams such as 'Buffalo'—exactly the noise source motivating LMEnt—whereas CS-SS Canonical only matches the full canonical name. Similarly, CI-SS Canonical is a single exact-string Elasticsearch query, not WIMBD's own index and query semantics. No evidence is provided that these stand-ins reproduce the retrieval sets of the real tools. The abstract and conclusion should either compare against the actual WIMBD/Infini-gram implementations or be reword
- [§5.2, Fig. 5, Fig. 7] The win-rate protocol does not measure what the prose claims. For each method and entity, the authors 'randomly sample 100 chunks from the set retrieved per method and entity' and then measure the absolute number of chunks judged as relevant. When both methods retrieve more than 100 chunks, this is a comparison of precision on a fixed 100-chunk sample, not a comparison of the total number of relevant chunks retrieved. A method that retrieves 1,000 relevant chunks at 90% precision can appear worse than a method that retrieves 100 relevant chunks at 100% precision. The claim that 'LMEnt returns more relevant document chunks' for 80% of entities is therefore not directly supported. The precision@k curves in Fig. 7 mitigate this only for the top-k ranking, not for the total-relevant-chunk claim. The protocol should be revised to compare total relevant counts (e.g., by estimating recall via s
minor comments (5)
- [§4, Table 2] The mention-count statistics are internally inconsistent. Section 4 says the 400M mentions 'are composed of' 115M hyperlink mentions, 203M entity-linking mentions, and 310M coreference mentions, which sum to 628M. Table 2 also lists 151M coreference mentions and 310M coreference cluster mentions. Please clarify whether these categories overlap and what the unique mention count is.
- [Abstract, §2] The corpus is described as 'fully annotated with entity mentions,' but the annotation pipeline has known error rates (2.7% errors in a 112-mention sample) and the retrieval thresholds (H=1, EL≥0.6, C≥0.6) intentionally exclude low-confidence mentions. Consider using 'fine-grained' or 'densely annotated' to avoid overstating coverage.
- [§B.3] The error analysis evaluates precision on retrieved mentions, but does not measure recall of the annotation pipeline. Since the suite's value depends on locating mentions, a statement about annotation recall (or a discussion of why recall is hard to measure) would strengthen the resource description.
- [§6] The term 'fact' is used for a (Subject, Answer) entity-pair co-occurrence, not for a relation-specific statement. This is acknowledged in the text, but the finding 'fact frequency is key' in the abstract and conclusion should be phrased as 'entity co-occurrence frequency' to avoid implying that the relation itself is measured.
- [§5.2, Baselines paragraph] The sentence 'we show in §5.2 that shorter references ... introduce noise' is self-referential because it appears within §5.2. Presumably it refers to experiments described later in the section; please rephrase.
Circularity Check
No circularity found: the retrieval and knowledge-acquisition claims are empirical comparisons against external benchmarks and an external LLM judge; the only self-citation is motivational and not load-bearing.
full rationale
LMEnt is primarily a resource paper: its central claims are that (1) entity-based chunk retrieval outperforms string-based retrieval, (2) the released models reach competitive knowledge-recall on PopQA/PAQ, and (3) co-occurrence frequency correlates with, but does not fully explain, knowledge acquisition. None of these claims is derived by construction from its own inputs. The retrieval win-rates are computed by comparing LMEnt's retrieved chunks against chunks retrieved by string-matching baselines, with relevance judged by an external LLM (Gemini 2.5 Flash) whose agreement with human annotators is separately tested. The retrieval thresholds were tuned on a 60-entity dev set and then evaluated on a disjoint 1K-entity test set, so the headline 80.4% win rate is not a re-statement of the tuning criterion. The learning-curve analysis correlates independently measured PopQA accuracy with corpus co-occurrence counts; although both quantities come from the same Wikipedia-derived corpus, the accuracy values are not algebraically or definitionally determined by the co-occurrence counts. The single self-citation (Gottesman & Geva, 2024) appears in the introduction as motivation for focusing on entity names and is not used to justify any quantitative result. The paper's main weakness is external-validity, not circularity: the WIMBD and Infini-gram comparisons use hand-rolled approximations ('CS-SS Canonical baseline approximates the Infinigram tool', 'CI-SS Canonical baseline approximates WIMBD'), so the 'outperforms previous approaches' claim depends on whether those approximations are faithful. That is a legitimate correctness risk, but it is not a case of a prediction being equivalent to a fitted input or of a self-citation chain forcing the conclusion. Consequently, no circular step is present and the score is 0.
Axiom & Free-Parameter Ledger
free parameters (2)
- Retrieval score thresholds (H=1, EL>=0.6, C>=0.6) =
H=1; EL=0.6; C=0.6
- Model hyperparameters =
batch size 32,768; rank batch 8,192; peak LR 5e-3; weight decay 0.05; 1,000 warmup steps
axioms (5)
- domain assumption Wikipedia is a knowledge base structured around entities, providing a consistent snapshot and a good proxy for factual pretraining data.
- domain assumption Off-the-shelf ReFinED and Maverick annotations are accurate enough at scale to support entity-level tracing.
- ad hoc to paper The hand-rolled string-matching baselines faithfully approximate WIMBD and Infini-gram.
- domain assumption Entity names, rather than individual tokens, are central to knowledge representations.
- ad hoc to paper Co-occurrence of subject and answer entities in a chunk defines fact exposure.
Cite this review
Pith. "Pith review of LMEnt: A Suite for Analyzing Knowledge in Language Models from Pretraining Data to Representations." pith.science (2026). https://pith.science/paper/O34I3AHS
@misc{pith2026250903405,
author = {Pith},
title = {Pith review of: LMEnt: A Suite for Analyzing Knowledge in Language Models from Pretraining Data to Representations},
year = {2026},
howpublished = {\url{https://pith.science/paper/O34I3AHS}},
note = {Machine review of arXiv:2509.03405}
}
read the original abstract
Language models (LMs) increasingly drive real-world applications that require world knowledge. However, the internal processes through which models turn data into representations of knowledge and beliefs about the world, are poorly understood. Insights into these processes could pave the way for developing LMs with knowledge representations that are more consistent, robust, and complete. To facilitate studying these questions, we present LMEnt, a suite for analyzing knowledge acquisition in LMs during pretraining. LMEnt introduces: (1) a knowledge-rich pretraining corpus, fully annotated with entity mentions, based on Wikipedia, (2) an entity-based retrieval method over pretraining data that outperforms previous approaches by as much as 80.4%, and (3) 12 pretrained models with up to 1B parameters and 4K intermediate checkpoints, with comparable performance to popular open-sourced models on knowledge benchmarks. Together, these resources provide a controlled environment for analyzing connections between entity mentions in pretraining and downstream performance, and the effects of causal interventions in pretraining data. We show the utility of LMEnt by studying knowledge acquisition across checkpoints, finding that fact frequency is key, but does not fully explain learning trends. We release LMEnt to support studies of knowledge in LMs, including knowledge representations, plasticity, editing, attribution, and learning dynamics.
Figures
Forward citations
Cited by 1 Pith paper
-
LLM generation novelty through the lens of semantic similarity
An output is called novel when no semantically similar pretraining context can be retrieved; using this embedding-based measure, SmolLM models show longer-range reuse and instruction tuning increases novelty.
Reference graph
Works this paper leans on
- [1]
-
[2]
the Wahhabi school of Saudi Arabia
The mention must refer to the Entity itself. Merely using the Entity's name as a descriptive modifier for something else associated with the Entity (e.g., "the Wahhabi school of Saudi Arabia" referring to the school, not the country) does not count as directly mentioning or discussing the Entity, unless the descriptive phrase is the Entity (e.g., "Univers...
-
[3]
Exclusions (Do NOT count as direct mention/discussion): Mentions of individuals or things related to the Entity but not the Entity itself (e.g., family members, associates, parts or products of the entity unless the mention clearly refers back to the entity)
-
[4]
Inferences based on context, related events, locations, or concepts if the Entity's name is not used to refer to the Entity itself according to Rule 3
-
[5]
DeepSeekMoE: Towards ultimate ex- pert specialization in mixture-of-experts lan- guage models. In Proceedings of the 62nd An- nual Meeting of the Association for Compu- tational Linguistics (V olume 1: Long Papers) , pages 1280–1297, Bangkok, Thailand. Associ- ation for Computational Linguistics. Jacob Devlin, Ming-Wei Chang, Kenton Lee et al
-
[6]
Possessives, actions, or attributes must be logically consistent with the entity’s type. Implausible associations, like assigning a teacher to Ennis, a town in County Clare, Ireland (e.g., “Ennis’s teacher”), are invalid. Consider each rule individually and assign an intermediate judgment of Pass or Fail. If any rule receives a Fail, the final decision mu...
-
[9]
Between circuits and Chomsky: Pre- pretraining on formal languages imparts lin- guistic biases. In Proceedings of the 63rd An- nual Meeting of the Association for Compu- tational Linguistics (V olume 1: Long Papers) , pages 9691–9709, Vienna, Austria. Association for Computational Linguistics. Binyuan Hui, Jian Yang, Zeyu Cui et al. 2024. Qwen2.5-coder te...
work page 2024
-
[11]
In The Thirteenth Interna- tional Conference on Learning Representations
Knowledge entropy decay during lan- guage model pretraining hinders new knowl- edge acquisition. In The Thirteenth Interna- tional Conference on Learning Representations. Knowledgator. 2025. Flashdeberta. https: //github.com/Knowledgator/ FlashDeBERTa. Accessed: 2025-08-01. Oleksii Kuchaiev, Jason Li, Daniil Shleifer et al
work page 2025
-
[12]
arXiv preprint arXiv:1909.09577
Nemo: a toolkit for building ai appli- cations using neural modules. arXiv preprint arXiv:1909.09577. Komal Kumar, Tajamul Ashraf, Omkar Thawakar et al. 2025. Llm post-training: A deep dive into reasoning large language models. Patrick Lewis, Yuxiang Wu, Linqing Liu et al
Pith/arXiv arXiv 1909
-
[13]
Transac- tions of the Association for Computational Lin- guistics, 9:1098–1115
PAQ: 65 million probably-asked ques- tions and what you can do with them. Transac- tions of the Association for Computational Lin- guistics, 9:1098–1115. Belinda Z. Li, Maxwell Nye and Jacob Andreas
-
[14]
Implicit representations of meaning in neural language models. In Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th Inter- national Joint Conference on Natural Language Processing (V olume 1: Long Papers) , pages 1813–1827, Online. Association for Computa- tional Linguistics. D Li, B Jiang, L Huang et al....
work page 2024
-
[15]
How pre-trained language models cap- ture factual knowledge? a causal-inspired anal- ysis. In Findings of the Association for Com- putational Linguistics: ACL 2022 , pages 1720– 1732, Dublin, Ireland. Association for Compu- tational Linguistics. Jiacheng Liu, Taylor Blanton, Yanai Elazar et al
work page 2022
-
[16]
OLMoTrace: Tracing language model outputs back to trillions of training tokens. In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (V ol- ume 3: System Demonstrations) , pages 178– 188, Vienna, Austria. Association for Compu- tational Linguistics. Jiacheng Liu, Sewon Min, Luke Zettlemoyer et al. 2024a. Infini-gram: ...
Pith/arXiv arXiv 2019
-
[17]
When not to trust language models: Investigating effectiveness of parametric and non-parametric memories. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (V olume 1: Long Pa- pers), pages 9802–9822, Toronto, Canada. As- sociation for Computational Linguistics. Giuliano Martinelli, Edoardo Barba and Roberto Navigl...
work page 2024
-
[18]
In Advances in Neural Information Processing Systems
Locating and editing factual associations in GPT. In Advances in Neural Information Processing Systems. Niklas Muennighoff, Luca Soldaini, Dirk Groen- eveld et al. 2025. OLMoe: Open mixture-of- experts language models. In The Thirteenth In- ternational Conference on Learning Represen- tations. NVIDIA. 2025. Nvidia nemotron nano 2: An ac- curate and effici...
Pith/arXiv arXiv 2025
-
[19]
In Proceedings of NAACL- HLT 2019: Demonstrations
fairseq: A fast, extensible toolkit for se- quence modeling. In Proceedings of NAACL- HLT 2019: Demonstrations. Matthew E. Peters, Mark Neumann, Robert Lo- gan et al. 2019. Knowledge enhanced contex- tual word representations. In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th Inter- national Joint Confer...
work page 2019
-
[20]
How much knowledge can you pack into the parameters of a language model? In Proceedings of the 2020 Conference on Empir- ical Methods in Natural Language Processing (EMNLP), pages 5418–5426, Online. Nicholas Roberts, Niladri S. Chatterji, Sharan Narang et al. 2025. Compute optimal scaling of skills: Knowledge vs reasoning. In Findings of the Association f...
Pith/arXiv arXiv 2020
-
[26]
Distinguishing Similar Names: If the Text contains a name that is similar to the Entity's name but refers to a different person, place, or thing (e.g., a different historical figure, a different organization with a similar name, or differing spellings like "Elisabeth" vs. "Elizabeth" referring to different individuals), it does not count as a direct menti...
-
[28]
The cloze statement must preserve the full meaning and specificity of the original question
Do not lose any context from the question. The cloze statement must preserve the full meaning and specificity of the original question. For example, Question: What TV network does Funny or Die Presents air on? Answer: HBO Cloze Statement: Funny or Die Presents airs on the TV network
-
[29]
Rephrase the question into a natural, grammatically correct sentence. If the question is incomplete or unclear, you may add minimal language to clarify the intended answer. For example, Question: stevie cameron worked as a confidential informant for the rcmp during which controversial Cloze Statement: Stevie Cameron worked as a confidential informant for ...
-
[30]
End the statement right before the answer
-
[31]
Use any specific labels or terms from the question. For example, if the question specifies "capital city" or "TV network," include that exact phrase in the cloze. Output ONLY the Cloze Statement at the end of thinking. Question: {question} Answer: {answer} Cloze Statement: Figure 19: Prompt given to Gemini 2.5 Flash Preview 6-17 to convert PAQ questions t...
-
[2016]
Badr AlKhamissi, Millicent Li, Asli Celikyilmaz et al
A Neural Knowledge Language Model. Badr AlKhamissi, Millicent Li, Asli Celikyilmaz et al. 2022. A review on language models as knowledge bases. Loubna Ben Allal, Anton Lozhkov, Elie Bakouch et al. 2025. Smollm2: When smol goes big – data-centric training of a small language model. Zeyuan Allen-Zhu and Yuanzhi Li. 2024. Physics of language models: part 3.1...
work page 2022
-
[2017]
Reference-aware language models. In Proceedings of the 2017 Conference on Empir- ical Methods in Natural Language Processing , pages 1850–1859, Copenhagen, Denmark. As- sociation for Computational Linguistics. Linxi Zhao, Sofian Zalouk, Christian K. Belardi et al. 2025. Pre-training large memory language models with internal and external knowledge. Vilém ...
work page 2017
-
[2019]
BERT: Pre-training of deep bidirec- tional transformers for language understand- ing. In Proceedings of the 2019 Conference of the North American Chapter of the Associ- ation for Computational Linguistics: Human Language Technologies, V olume 1 (Long and Short Papers), pages 4171–4186, Minneapolis, Minnesota. Association for Computational Lin- guistics. Y...
work page 2019
-
[2020]
Vladimir Karpukhin, Barlas Oguz, Sewon Min et al
How can we know what language mod- els know? Transactions of the Association for Computational Linguistics, 8:423–438. Vladimir Karpukhin, Barlas Oguz, Sewon Min et al. 2020. Dense passage retrieval for open- domain question answering. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 6769–6781, Onlin...
work page 2020
-
[2021]
Transactions of the Association for Computational Linguistics , 9:1012–1031
Measuring and improving consistency in pretrained language models. Transactions of the Association for Computational Linguistics , 9:1012–1031. Ronen Eldan and Yuanzhi Li. 2023. Tinysto- ries: How small can language models be and still speak coherent english? arXiv preprint arXiv:2305.07759. Mor Geva, Jasmijn Bastings, Katja Filippova et al
Pith/arXiv arXiv 2023
-
[2022]
ReFinED: An efficient zero-shot-capable approach to end-to-end entity linking. In Pro- ceedings of the 2022 Conference of the North American Chapter of the Association for Com- putational Linguistics: Human Language Tech- nologies: Industry Track , pages 209–220, Hy- brid: Seattle, Washington + Online. Association for Computational Linguistics. Elie Bakou...
work page 2022
-
[2023]
Dissecting recall of factual associations in auto-regressive language models. In Pro- ceedings of the 2023 Conference on Empiri- cal Methods in Natural Language Processing , pages 12216–12235, Singapore. Association for Computational Linguistics. Abbas Ghaddar and Phillippe Langlais. 2016. Wi- kiCoref: An English coreference-annotated cor- pus of Wikipedi...
Pith/arXiv arXiv 2023
-
[2024]
How do large language models acquire factual knowledge during pretraining? In Ad- vances in Neural Information Processing Sys- tems, volume 37, pages 60626–60668. Curran Associates, Inc. Roi Cohen, Eden Biran, Ori Yoran et al. 2024. Evaluating the ripple effects of knowledge edit- ing in language models. Transactions of the Association for Computational L...
work page 2024
-
[2025]
The alternative annotator test for LLM- as-a-judge: How to statistically justify replac- ing human annotators with LLMs. In Proceed- ings of the 63rd Annual Meeting of the As- sociation for Computational Linguistics (V ol- ume 1: Long Papers) , pages 16051–16081, Vi- enna, Austria. Association for Computational Linguistics. Hoyeon Chang, Jinho Park, Seong...
This paper was first reviewed by deepseek-v4-flash on August 5, 2026.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.