Can Transformers Learn to Verify During Backtracking Search?
Not on the map. Why?
Abstract (arXiv)
Backtracking search underlies classical constraint solvers, planners, and theorem provers. Recent transformer-based reasoning systems explore search trees over their own intermediate steps. A common training recipe fits an autoregressive next-token loss on offline solver traces. The model's input at each step is a cumulative trace of all prior decisions. The optimal continue-or-backtrack predictor depends only on the current search state, since two trajectories reaching the same state admit the same viable continuations. We show that decoder-only transformers trained on cumulative traces fail this requirement in two ways: the trace can scatter state features across many positions (scattered retrieval), and the predictor can condition on the trajectory rather than the state (history entanglement). We address scattered retrieval with localization, a trace-level fix that rewrites each decision block to expose state features locally. We address history entanglement with Selective State Attention (SSA), a fixed attention mask that enforces state-based decisions structurally without modifying training data, objective, or parameters. We focus on reactive verification, after propagation has exposed a contradiction. We test SSA on 3-SAT, graph coloring, Blocks World, and backtracking parsing. On same-state pairs that differ only in prior history, SSA emits identical decisions while a cumulative-trained causal baseline does not. Our contribution is a diagnostic of transformer behavior on serialized trajectory data, paired with a structural fix. Pretrained language models that search over their own reasoning steps may face the same failure. Our analysis opens up inference-time context clearing as a candidate way to apply the same isolation without retraining.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). First posted on arXiv 21 May 2026.
Builds on
0 of 0 references.
No references here yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Trace as State: Reasoning Traces as Conditional States for Long-Context Transformers
- 2025To Backtrack or Not to Backtrack: When Sequential Search Limits Model Reasoning
- 2026The Shape of Overthinking: Backtracking Bursts in Long Reasoning Traces
- 2024Beyond A*: Better Planning with Transformers via Search Dynamics Bootstrapping
- 2025How Much Backtracking is Enough? Exploring the Interplay of SFT and RL in Enhancing LLM Reasoning
- 2026Boosting Inference with Guided Reasoning: Stochastic Exploration for Recursive Models
- 2026Transformers Linearly Represent Highly Structured World Models
- 2025Reasoning-Finetuning Repurposes Latent Representations in Base Models