Stratified Consistency Distillation for Natural Language Formalization

2026-08-31Computation and Language

Computation and LanguageArtificial Intelligence
AI summary

The authors worked on improving how computers turn natural language into logical formulas, which helps with complex reasoning tasks. Instead of just tweaking prompts for large language models, they created a method that generates multiple translations, groups similar ones, and uses different rules to pick the best labels. Then, they train a smaller model with these labels for better accuracy. Their tests showed this approach consistently improves performance on logical translation tasks.

neurosymbolic reasoninglarge language modelslogical translationprompt engineeringfine-tuningconsistency distillationsemantic clusteringpseudo-labelsmajority votingentropy
Authors
Zhichao Hou, Ferhat Erata, Joe Lilien, MohamadAli Torkamani
Abstract
Neurosymbolic reasoning has shown promising success in addressing complex reasoning tasks by combining large language models (LLMs) and symbolic solvers. While this approach shows promise, a fundamental challenge remains: improving the accuracy of translations from natural language to logical formulas. Current methods predominantly rely on prompt engineering, which is difficult to scale across different domains and input formats. Drawing inspiration from the success of fine-tuning in other model adaptation and alignment applications, we propose a fine-tuning-based Stratified Consistency Distillation approach: (1) We generate K logical translations per input using a frontier LLM and cluster them by semantic equivalence (2) Based on the entropy level, we apply majority voting (low entropy), LLM-as-a-Judge (medium entropy), or unification/abstention (high entropy), and (3) fine-tune a smaller model using the selected pseudo-labels. Our experiments show significant and consistent improvements in both Pass@K and our novel Equivalent Logical Similarity metrics, demonstrating the potential of advancing logical translation through consistency distillation.