NaturalPRISM: A Natural Language Premise Retrieval Task for Research-level Intermediate Proof States

Harris Proctor, Li An, Austin LaHue, Michael Burr, Vinita Gangaram Jansari, Carlos Toxtli Hernandez, Luis David Garcia Puente, Benjamin E Nye


Abstract
As AI use progresses in research-level mathematics, automated information retrieval systems for constructing relevant prompts are increasingly important. Existing benchmarks and datasets for identifying which prior results are necessary to prove a statement do not align well with current workflows in which a natural language proof is in progress, but it is unclear what the next step(s) should be.

We introduce NaturalPRISM, a benchmark task and dataset containing 1.7 million pairs of intermediate proof states and the premise invoked in the next step of the proof. All pairs are extracted from natural language research-level mathematics papers published on arXiv, and span all 32 subject categories.

We evaluate the effect of fine-tuning retrieval models for domain-specific vs. domain-agnostic use cases, and find that the combination of sufficient training data with alignment between the training and testing distributions allows smaller specialized models to dramatically outperform larger general models.
Anthology ID:
2026.mathnlp-main.13
Volume:
The Proceedings of the 4th Workshop on Mathematical Natural Language Processing (MathNLP 2026)
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Marco Valentino, Deborah Ferreira, Mokanarangan Thayaparan, Leonardo Ranaldi, Jo-Ku Cheng, Andre Freitas
Venues:
MathNLP | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
167–176
Language:
URL:
https://aclanthology.org/2026.mathnlp-main.13/
DOI:
Bibkey:
Cite (ACL):
Harris Proctor, Li An, Austin LaHue, Michael Burr, Vinita Gangaram Jansari, Carlos Toxtli Hernandez, Luis David Garcia Puente, and Benjamin E Nye. 2026. NaturalPRISM: A Natural Language Premise Retrieval Task for Research-level Intermediate Proof States. In The Proceedings of the 4th Workshop on Mathematical Natural Language Processing (MathNLP 2026), pages 167–176, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
NaturalPRISM: A Natural Language Premise Retrieval Task for Research-level Intermediate Proof States (Proctor et al., MathNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.mathnlp-main.13.pdf