Teaching Probabilistic Logical Reasoning to Transformers
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
In this paper, we evaluate the capability of transformer-based language models in making inferences over uncertain text that includes uncertain rules of reasoning. We cover both Pre-trained Language Models (PLMs) and generative Large Language Models (LLMs). Our evaluation results show that both generations of language models struggle with reasoning over uncertain text. We propose a novel end-to-end fine-tuning approach, Probabilistic Constraint Training (PCT), that utilizes probabilistic logical rules as constraints in the fine-tuning phase without relying on these rules in the inference stage. To assess the effectiveness of PCT, we utilize the related corpora and, additionally, create a new and more challenging benchmark that, unlike the previous ones, uses instance-specific rules. Our study demonstrates that PCT improves the transformer-based language model's intrinsic reasoning and makes their probabilistic logical reasoning process more explicit and explainable. Furthermore, PCT equips these models to effectively handle novel situations, including higher reasoning depth, new domains, and complex probabilistic structures.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 26 of its 45 references that are in the map. First posted on arXiv 22 May 2023.
Builds on
26 of 45 references.
References · 26
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
- 2019Neural Probabilistic Logic Programming in DeepProbLogcited by 394
- 2020Generative Language Modeling for Automated Theorem Provingcited by 388
- 2020ProofWriter: Generating Implications, Proofs, and Abductive Statements over Natural Languagecited by 324
- 2015Towards AI-Complete Question Answering: A Set of Prerequisite Toy Taskscited by 314
- 2020Transformers as Soft Reasoners over Languagecited by 270
- 2017End-to-End Differentiable Provingcited by 161
- 2022Large Language Models are few(1)-shot Table Reasonerscited by 116
- 2022NumGLUE: A Suite of Fundamental yet Challenging Mathematical Reasoning Taskscited by 112
14 more
- 2019QuaRTz: An Open-Domain Dataset of Qualitative Relationship Questionscited by 89
- 2019Reasoning Over Paragraph Effects in Situationscited by 79
- 2020Logic-Guided Data Augmentation and Regularization for Consistent Question Answeringcited by 78
- 2019WIQA: A dataset for "What if..." reasoning over procedural textcited by 71
- 2022StepGame: A New Benchmark for Robust Multi-Hop Spatial Reasoning in Textscited by 68
- 2019A Logic-Driven Framework for Consistency of Neural Modelscited by 64
- 2023Scallop: A Language for Neurosymbolic Programmingcited by 63
- 2020PRover: Proof Generation for Interpretable Reasoning over Rulescited by 53
- 2020Learning to Prove Theorems by Learning to Generate Theoremscited by 43
- 2019NLProlog: Reasoning with Weak Unification for Question Answering in Natural Languagecited by 37
- 2021Measuring Sentence-Level and Aspect-Level (Un)certainty in Science Communicationscited by 27
- 2021RuleBert: Teaching Soft Rules to Pre-trained Language Modelscited by 25
- 2023Improved Logical Reasoning of Language Models via Differentiable Symbolic Programmingcited by 18
- 2023GLUECons: A Generic Benchmark for Learning Under Constraintscited by 9
Built on by
Citing papers per year: 2023: 1; 2024: 2; 2025: 3; 2026: 1.
7 papers.
- 2024Enhancing Reasoning Capabilities of LLMs via Principled Synthetic Logic Corpuscited by 29
- 2024A Survey on Compositional Learning of AI Models: Theoretical and Experimental Practicescited by 15
- 2025Extracting Probabilistic Knowledge from Large Language Models for Bayesian Network Parameterizationcited by 15
- 2025Reasoning Under Uncertainty: Exploring Probabilistic Reasoning Capabilities of LLMscited by 8
- 2025Bayesian Network Structure Discovery Using Large Language Modelscited by 5
- 2023Federated Neuro-Symbolic Learningcited by 2
- 2026AdaptFuse: Training-Free Sequential Preference Inference via Externalized Bayesian Computationcited by 2
- 2026AdaptFuse: Training-Free Sequential Preference Inference via Externalized Bayesian Computation5 Apr 2026
- 2025Bayesian Network Structure Discovery Using Large Language Models1 Nov 2025
- 2025Reasoning Under Uncertainty: Exploring Probabilistic Reasoning Capabilities of LLMs12 Sep 2025
- 2025Extracting Probabilistic Knowledge from Large Language Models for Bayesian Network Parameterization21 May 2025
- 2024Enhancing Reasoning Capabilities of LLMs via Principled Synthetic Logic Corpus19 Nov 2024
- 2024A Survey on Compositional Learning of AI Models: Theoretical and Experimental Practices13 Jun 2024
- 2023Federated Neuro-Symbolic Learning29 Aug 2023
Similar, not cited
Close in wording, no citation either way
- 2024Towards Logically Consistent Language Models via Probabilistic Reasoning
- 2024Reasoning over Uncertain Text by Generative Large Language Models
- 2026Using Probabilistic Programs to Train Inductive Reasoning in Large Language Models
- 2022ThinkSum: Probabilistic reasoning over sets using large language models
- 2023Transformers in the Service of Description Logic-based Contexts
- 2024Transformer-based Language Models for Reasoning in the Description Logic ALCQ
- 2024Reasoning in Transformers -- Mitigating Spurious Correlations and Reasoning Shortcuts
- 2025Enigme: Generative Text Puzzles for Evaluating Reasoning in Language Models