What are the Right Symmetries for Formal Theorem Proving?

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Olejniczak, Krzysztof, Dimitrov, Radoslav, Huang, Xingyue, Grau, Bernardo Cuenca, Kim, Jinwoo, Ceylan, İsmail İlkan
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910245603770368
author Olejniczak, Krzysztof
Dimitrov, Radoslav
Huang, Xingyue
Grau, Bernardo Cuenca
Kim, Jinwoo
Ceylan, İsmail İlkan
author_facet Olejniczak, Krzysztof
Dimitrov, Radoslav
Huang, Xingyue
Grau, Bernardo Cuenca
Kim, Jinwoo
Ceylan, İsmail İlkan
contents Formal theorem provers based on large language models (LLMs) are highly sensitive to superficial variations in problem representation: semantically equivalent statements can exhibit drastically different proof success rates, revealing a failure to respect structural symmetries inherent in formal mathematics. This raises a central question: what are the right symmetries for formal theorem proving? We introduce rewriting categories, a category-theoretic framework capturing the compositional, generally non-invertible transformations induced by proof tactics, and use it to formalize two symmetry notions: proof equivariance, governing how proof distributions transform under rewrites, and success invariance (i.e., invariance of success probability), requiring equivalent statements to be solved with the same probability. We observe that state-based next-tactic provers naturally satisfy proof equivariance by operating on proof states. In contrast, state-of-the-art LLM-based provers satisfy neither property, exhibiting large performance variation across equivalent formulations. To mitigate this, we propose test-time methods that aggregate over equivalent rewritings of the input, showing theoretically that they recover success invariance in the sampling limit, and empirically, that they improve robustness and performance under fixed inference budgets. Our results highlight symmetry as a key missing inductive bias in LLM-based theorem proving and suggest test-time computation as a practical route to approximate it.
format Preprint
id arxiv_https___arxiv_org_abs_2605_22257
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle What are the Right Symmetries for Formal Theorem Proving?
Olejniczak, Krzysztof
Dimitrov, Radoslav
Huang, Xingyue
Grau, Bernardo Cuenca
Kim, Jinwoo
Ceylan, İsmail İlkan
Machine Learning
Artificial Intelligence
Logic in Computer Science
Formal theorem provers based on large language models (LLMs) are highly sensitive to superficial variations in problem representation: semantically equivalent statements can exhibit drastically different proof success rates, revealing a failure to respect structural symmetries inherent in formal mathematics. This raises a central question: what are the right symmetries for formal theorem proving? We introduce rewriting categories, a category-theoretic framework capturing the compositional, generally non-invertible transformations induced by proof tactics, and use it to formalize two symmetry notions: proof equivariance, governing how proof distributions transform under rewrites, and success invariance (i.e., invariance of success probability), requiring equivalent statements to be solved with the same probability. We observe that state-based next-tactic provers naturally satisfy proof equivariance by operating on proof states. In contrast, state-of-the-art LLM-based provers satisfy neither property, exhibiting large performance variation across equivalent formulations. To mitigate this, we propose test-time methods that aggregate over equivalent rewritings of the input, showing theoretically that they recover success invariance in the sampling limit, and empirically, that they improve robustness and performance under fixed inference budgets. Our results highlight symmetry as a key missing inductive bias in LLM-based theorem proving and suggest test-time computation as a practical route to approximate it.
title What are the Right Symmetries for Formal Theorem Proving?
topic Machine Learning
Artificial Intelligence
Logic in Computer Science
url https://arxiv.org/abs/2605.22257