核心发现
方法论
论文将GPT-2式因果Transformer训练为可搜索的组合问题求解器。模型先模仿四条Sudoku基础规则,再在规则无法推进时进行知情猜测;验证器检查部分赋值,若出现冲突则依据DPLL式Depth-First Search回溯。动作级编码把(r,c,v)压缩为三位数token,并用多目标损失−∑_{i∈S}log p_i学习多个合法下一步。
关键结果
- 在100K随机生成Sudoku上,完整方法达到约99%棋盘准确率;在1-in-3 SAT上也达到99%。GPT-4o、Gemini-1.5 Pro、o3 mini、Gemini-2.5 Flash和DeepSeek-R1在9×9 Sudoku上的棋盘准确率均为0%。
- 方法在Random、Kaggle unfiltered、Kaggle filtered和RRN四类数据上测试。图1显示其准确率约99%,超过RRN、Recurrent Transformer、MDM等神经求解器;随机生成训练还保持了对Kaggle和RRN分布的泛化。
- 深度1猜测实验表明,约99.8%的随机Sudoku在基础规则后只需一个正确猜测即可完成。与交叉熵相比,源于contextual MIN-SUM SET COVER的新损失更直接优化解答长度。
研究意义
研究说明,Transformer不必一次性生成完整答案,也可以在验证器监督下逐步尝试、纠错和回溯。这缓解了LLM在算术、SAT和Sudoku中“会生成但不会验证”的问题,并把语言模型研究连接到算法搜索、随机优化与可解释推理。由于框架只依赖问题验证器,理论上可迁移到SAT、Clique和TSP等NP问题。
技术贡献
核心工程贡献包括动作级tokenization、多合法标签监督、规则执行与DFS搜索的统一轨迹,以及C语言实现的高效SudokuPy生成器。理论上,作者把单层、非自适应猜测形式化为contextual MIN-SUM SET COVER,并据此设计直接刻画solution length的损失。模型是42M参数、8层8头、576维嵌入的vanilla GPT-2,而非定制神经架构。
新颖性
区别于直接预测完整棋盘或只模仿七种人工策略的方法,本文让普通decoder-only Transformer学习“规则—猜测—验证—回溯”的闭环。其新颖之处还在于把减少猜测次数转化为MIN-SUM SET COVER目标,并用只含成功轨迹的训练超越传统模仿学习。
局限性
- 实验主体是9×9 Sudoku及1-in-3 SAT;深度1猜测依赖随机Sudoku中约99.8%的实例存在backdoor,不能证明对大规模NP问题同样高效。
- DFS在搜索树增大时仍可能指数爆炸;模型错误猜测、序列长度上限和验证器设计也会造成失败。论文尚未给出大规模工业实例或严格多项式时间保证。
未来方向
未来可扩展到更大Sudoku、SAT、Clique和TSP,研究多层猜测、记忆历史的自适应搜索及更强验证器。还应比较不同Transformer规模、损失和搜索策略,并建立统一的NP问题流式基准。
AI 总览摘要
大型语言模型在文本生成上表现卓越,却常在Sudoku、SAT、TSP乃至基础算术中失效。GPT-4o、Gemini-1.5 Pro、o3 mini、Gemini-2.5 Flash和DeepSeek-R1在本文测试的9×9 Sudoku上棋盘准确率均为0%,主要原因是它们一旦作出错误推断,通常不能可靠回溯。
Giannoulis等提出一种“高效试错”框架,让42M参数的vanilla GPT-2先学习四条基础Sudoku规则,再在规则耗尽时进行知情猜测。验证器检查每一步;若冲突,模型按DPLL式DFS回溯并尝试另一候选值。动作级token和多目标损失允许多个合法下一步共同监督模型。结果显示,在100K随机Sudoku上准确率约99%,在1-in-3 SAT上也为99%,超过多种定制神经求解器。
更进一步,作者把减少解答长度形式化为contextual MIN-SUM SET COVER。深度1猜测实验发现,约99.8%的随机Sudoku在基础规则后只需一个正确猜测即可完成。研究意义不在于声称Transformer解决了NP难题,而在于展示一种可验证、可纠错、可迁移的神经搜索范式;不过其效率仍依赖实例分布、验证器和较小搜索深度。
深度分析
研究背景
LLM通常以next-token prediction学习语言,却缺少组合问题所需的精确约束维护。早期Sudoku方法使用Hopfield Network、Recurrent Relational Network、Recurrent Transformer或Masked Diffusion Model;SDWP24则用GPT式模型模仿七种人工策略。但直接生成答案或无错误轨迹训练缺乏恢复机制。本文转向“生成候选—验证—回溯”的算法式推理。
核心问题
目标是在不调用外部工具的情况下,让普通decoder-only Transformer逐步求解NP类问题。难点包括:多个下一步都可能合法;局部规则可能耗尽;错误猜测必须被识别并撤销;同时还要减少搜索时间,而非仅提高最终正确率。
核心创新
第一,使用四条基础规则而非七种复杂策略,并由DFS补足规则盲区。第二,以单个三位数token表示(r,c,v),将序列长度缩短约三倍。第三,用多目标损失监督所有合法动作。第四,把单猜测、非自适应设置连接到contextual MIN-SUM SET COVER,并优化solution length。
方法详解
- �� 输入初始线索与start token。
- �� 模型反复输出可由Lone/Hidden Single类规则推出的动作;无动作时输出rules-end。
- �� 输出guess level,并从候选值中选择猜测。
- �� 验证器检查行、列、3×3 box及部分赋值。
- �� 若出现dead-end,回到最近猜测层,换用同一格的另一候选,形成DPLL式DFS。
- �� 训练时对合法集合S使用−∑_{i∈S}log p_i;推理时取每格最后赋值。
实验设计
模型为42M参数GPT-2变体:8层、8个注意力头、576维embedding、3456维前馈层。数据包括100K Random测试题、Kaggle unfiltered、Kaggle filtered及RRN数据集,并比较RRN、Recurrent Transformer、Causal Transformer和MDM。消融比较三位token、单目标损失与多目标损失;另测1-in-3 SAT和深度1猜测。
结果分析
完整模型在Sudoku上约99%棋盘准确率,在1-in-3 SAT上99%。图1报告其约99%表现,明显高于若干神经基线;Random训练在Kaggle及RRN上也保持较高准确率。规则学习消融显示,动作级编码加多目标监督收敛更快。深度1实验中约99.8%随机题存在可行backdoor。
应用场景
可用于SAT可满足性检查、约束排程、组合设计、物流路线和游戏规划,前提是存在高效验证器。实际系统可将Transformer作为候选动作排序器,将符号验证器作为安全闸门;这比完全依赖自由生成更适合需要审计和纠错的流程。
局限与展望
Sudoku规模小且分布受随机生成器控制;99.8%的单猜测结果不能外推至任意NP实例。DFS仍可能指数级增长,模型也受最大序列长度和错误动作影响。1-in-3 SAT实验支持泛化方向,但论文尚未证明大规模SAT、TSP或工业约束问题上的成本优势。
通俗解读 非专业人士也能看懂
把模型想成一位解数独的学生。它先使用课本规则填空:某个格子只能放一个数字,某个数字在一行、一列或一个小宫格中只有一个位置。若这些规则都用完了,学生就试填一个看起来最有希望的数字。
关键是学生不会把第一次尝试当成事实。每填一步,就让“裁判”检查是否违反行、列或小宫格规则。如果发现矛盾,他退回到最后一次试填的位置,换一个数字继续。这就是一种有记录的试错,而不是盲猜。
训练时,老师不仅告诉他一个正确答案,也告诉他所有当时都可以接受的动作。研究发现,这样的普通GPT-2模型在十万道随机Sudoku上约99%解对,并且约99.8%的题在基础规则后只需要一次正确试填。它的价值是把“会说答案”变成“会尝试、检查和改正”。
简单解释 像给14岁少年讲一样
想象你在玩Sudoku闯关游戏。普通聊天机器人像一个反应很快但不存档的玩家:它能连续说出很多数字,可一旦某一步错了,后面的数字全乱,而且不知道从哪里重来。论文里的GPT-2则像会自动存档的玩家。
它先用简单规则清理棋盘。如果卡住,就选择一个可能的数字试试看。每次行动后都有裁判检查:行、列和小宫格不能重复。若裁判说“不行”,系统回到最近的存档点,换另一个数字。这种“试一下—检查—撤销—再试”的过程叫DFS回溯。
研究者还把每一步压缩成一个动作token,并允许多个答案同时作为正确训练目标。结果很惊人:在100K随机Sudoku上大约99%的棋盘被完整解出;1-in-3 SAT也达到99%。约99.8%的随机Sudoku只需一次猜测就能完成。
但这不代表AI已经轻松解决所有超级难题。棋盘变大后,可能的尝试会爆炸式增加,像游戏地图分支太多一样。真正厉害的地方是:模型不再只背答案,而是学会了带裁判的探索和纠错。
术语表
Depth-First Search(深度优先搜索)
沿一条候选路径尽可能深入,失败后退回最近分叉点。它保证系统能系统地探索候选空间。
模型猜测、检测dead end并回溯时采用DFS。
Verifier(验证器)
快速判断部分或完整候选解是否违反约束的程序。它不负责寻找答案,只负责检查。
验证Sudoku行、列、box及SAT赋值。
Backdoor(后门)
一个关键猜测,使后续简单规则足以完成实例。它不是安全漏洞,而是搜索意义上的突破点。
约99.8%随机Sudoku在深度1设置下存在backdoor。
Multiple-target loss(多目标损失)
当多个动作都合法时,同时提高它们的概率。论文使用−∑_{i∈S}log p_i。
用于学习规则阶段的多个可行下一步。
MIN-SUM SET COVER
以最小期望代价覆盖目标集合的优化问题。本文将猜测选择解释为上下文相关的集合覆盖。
用于推导直接优化解答长度的新损失。
开放问题 这项研究留下的未解疑问
- 1 尚不清楚多层猜测和长期记忆能否在大规模SAT、TSP中保持优势;需要跨实例分布的理论界限与真实工业基准。
- 2 验证器若昂贵或难以编写,框架成本会显著上升;如何自动学习可靠验证器仍是开放问题。
应用场景
近期应用
约束规划辅助
企业可将模型用于排班、资源分配或配置生成:Transformer提出候选动作,验证器即时拒绝违规方案,DFS负责回溯。前提是约束可程序化检查,预期收益是更好的可审计性。
SAT与规则测试
研究团队可用1-in-3 SAT轨迹训练候选解生成器,并把逻辑验证器置于闭环中。该设计适合测试神经模型是否真正遵守约束,而非只产生看似合理的文本。
远期愿景
可验证的神经搜索系统
未来可形成统一平台,让同一类Transformer通过替换验证器处理SAT、Clique、TSP和调度问题。主要障碍是搜索规模、验证成本和跨任务表示学习。
原文摘要
Despite their proficiency in various language tasks, Large Language Models (LLMs) struggle with combinatorial problems like Satisfiability, Traveling Salesman Problem, or even basic arithmetic. We address this gap through a novel trial & error approach for solving problems in the class NP, where candidate solutions are iteratively generated and efficiently validated using verifiers. We focus on the paradigmatic task of Sudoku and achieve state-of-the-art accuracy (99%) compared to prior neuro-symbolic approaches. Unlike prior work that used custom architectures, our method employs a vanilla decoder-only Transformer (GPT-2) without external tools or function calling. Our method integrates imitation learning of simple Sudoku rules with an explicit Depth-First Search (DFS) exploration strategy involving informed guessing and backtracking. Moving beyond imitation learning, we seek to minimize the number of guesses until reaching a solution. This is achieved using depth-1 guessing, showing empirically that almost all Sudoku can be solved using the puzzle's rules with at most one guess. We provide a rigorous analysis of this setup formalizing its connection to a contextual variant of Min-Sum Set Cover, a well-studied problem in algorithms and stochastic optimization.