cs.LGMay 21, 2026

What are the Right Symmetries for Formal Theorem Proving?

Authors: Krzysztof OlejniczakRadoslav DimitrovXingyue HuangBernardo Cuenca GrauJinwoo Kimİsmail İlkan Ceylan

Organizations: University of Oxford · KAIST · TU Wien · AITHYRA

Abstract

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.

Explore similar work

CardsList
  1. Benchmarking Testing in Automated Theorem Proving

    Apr 26, 2026Jongyoon Kim, Hojae Han, Seung-won HwangTheoremProof

  2. FormalRewardBench: A Benchmark for Formal Theorem Proving Reward Models

    May 11, 2026Zeynel A. Uluşan, Burak S. Akbudak, Can S. Erer +1TheoremReasoning Benchmark