In plain words: They show a transformer can solve 3-SAT by writing out step-by-step guesses and deductions, like the classic backtracking solver, then train one on those reasoning traces. It handles new problems of familiar sizes well but struggles when problems get longer.
Abstract
We formally study the logical reasoning capabilities of decoder-only Transformers in the context of the boolean satisfiability (SAT) problem. First, we prove by construction that decoder-only Transformers can decide 3-SAT, in a non-uniform model of computation, using backtracking and deduction via Chain-of-Thought (CoT). %We prove its correctness by showing trace equivalence to the well-known DPLL SAT-solving algorithm. Second, we implement our construction as a PyTorch model with a tool (PARAT) that we designed to empirically demonstrate its correctness and investigate its properties. Third, rather than \textit{programming} a transformer to reason, we evaluate empirically whether it can be \textit{trained} to do so by learning directly from algorithmic traces (``reasoning paths'') from our theoretical construction. The trained models demonstrate strong out-of-distribution generalization on problem sizes seen during training but has limited length generalization, which is consistent with the implications of our theoretical result
Leyan Pan, Vijay Ganesh, Jacob Abernethy, Chris Esposo, Wenke Lee
arXiv:2410.07432 · cs.LG, cs.AI, cs.LO · submitted Oct 9, 2024 · updated Feb 8, 2025
abstract · pdf · html · 41 pages, 4 Figures