sudoku
sudoku 是一个探索卷积神经网络(CNN)能否独立解决数独谜题的开源实验项目。传统数独求解通常依赖复杂的规则引擎或回溯算法,而 sudoku 尝试让深度学习模型在不借助任何基于规则的后处理步骤下,直接通过图像识别般的逻辑“看”出答案。
该项目核心解决了如何利用纯数据驱动的方式处理强逻辑约束问题。它使用了一百万个生成的数独样本进行训练,构建了一个包含 10 个卷积层的简单网络模型。其独特的技术亮点在于推理策略:模型并非一次性填满所有空格,而是采用迭代方式,每次仅填入预测概率最高的一个数字,逐步推导直至完成。这种“分步填充”的技巧显著提升了准确率,在测试中整体准确度达到了 86%,且在部分高难度关卡中实现了完美求解。
sudoku 非常适合对深度学习、计算机视觉及逻辑推理交叉领域感兴趣的研究人员和开发者使用。它不仅提供了完整的代码实现、数据集生成脚本及预训练模型,还展示了基础 CNN 架构在处理离散逻辑任务上的潜力。对于希望了解神经网络如何学习规则,或打算在此基础上探索更复杂关系推理(如结合可微分优化层)的技术人员来说,这是一个极具参考价值的入门案例。
使用场景
某教育科技公司的算法团队正在开发一款智能数独辅导 App,需要为不同难度的题目提供实时解题提示和自动批改功能。
没有 sudoku 时
- 团队必须手动编写复杂的回溯算法和逻辑规则引擎,代码量大且难以维护,稍有疏漏就会导致解题失败。
- 面对“专家”或“恶魔”级的高难度题目,传统规则匹配容易陷入死循环,导致服务器响应超时,用户体验极差。
- 每次更新题库或调整难度策略,都需要重新测试大量边界案例,迭代周期长,无法快速验证新想法。
- 系统缺乏泛化能力,一旦遇到非标准变体或噪声干扰(如用户手写识别误差),整个求解流程就会崩溃。
使用 sudoku 后
- 直接加载预训练的卷积神经网络模型,无需编写任何基于规则的后期处理代码,大幅简化了后端架构。
- 利用模型“逐步填充最高概率空格”的推理技巧,即使面对极高难度题目也能保持高准确率(部分类别达 100%),确保响应流畅。
- 通过简单的
test.py脚本即可批量评估新题库表现,训练曲线直观展示收敛情况,让算法迭代效率提升数倍。 - 模型展现出强大的容错性与泛化力,能够直接从数据中学习数独的内在逻辑,有效应对输入中的轻微噪声或不完整信息。
sudoku 证明了深度学习可以替代繁琐的传统逻辑编程,用端到端的神经网络高效解决复杂的约束满足问题。
运行环境要求
未说明(基于 TensorFlow 1.1,通常支持 CPU 或任意兼容的 NVIDIA GPU,无具体型号要求)
未说明

快速开始
卷积神经网络能否破解数独谜题?
数独是一种广受欢迎的数字谜题,要求在9×9的网格中填入数字,使得每一列、每一行以及九个3×3的小宫格内都包含1至9的所有数字。解决数独的方法多种多样,其中也包括基于计算的方法。在本项目中,我展示了简单的卷积神经网络无需任何基于规则的后处理即可破解数独的潜力。
需求
- NumPy >= 1.11.1
- TensorFlow == 1.1
背景
- 如需了解数独是什么,请参阅维基百科
- 若要全面研究此任务,请阅读McGuire等,2013年。
数据集
- 使用
generate_sudoku.py生成了100万局用于训练的数据。这些数据已上传至Kaggle数据集存储,可在此下载:sudoku.zip。 - 另外,我还从http://1sudoku.com收集了30局真实的数独游戏。
模型描述
- 10个卷积层块,卷积核大小为3。
文件说明
generate_sudoku.py用于生成数独游戏。您无需运行此文件,可以直接下载预生成的游戏。hyperparams.py包含了所有可调的超参数。data_load.py负责加载数据并将其放入队列中,以便并行生成多个小批量数据。modules.py包含一些封装函数。train.py用于模型训练。test.py用于模型测试。
训练
测试
- 运行
python test.py。
评估指标
准确率定义为:
预测与答案完全一致的空格数 / 总空格数。
结果
经过几个小时的训练,训练曲线似乎已达到最优状态。
在推理过程中,我采用了一个简单技巧:不是一次性填完整个数独,而是在所有预测中选择置信度最高的一个空格进行填充。如下所示,我的模型准确率达到0.86。详细结果请参见`results`文件夹。
| 难度级别 | 准确率(正确数/空格数=准确率) |
|---|---|
| 简单 | 47/47 = 1.00 |
| 简单 | 45/45 = 1.00 |
| 简单 | 47/47 = 1.00 |
| 简单 | 45/45 = 1.00 |
| 简单 | 47/47 = 1.00 |
| 简单 | 46/46 = 1.00 |
| 中等 | 33/53 = 0.62 |
| 中等 | 55/55 = 1.00 |
| 中等 | 55/55 = 1.00 |
| 中等 | 53/53 = 1.00 |
| 中等 | 33/52 = 0.63 |
| 中等 | 51/56 = 0.91 |
| 困难 | 29/56 = 0.52 |
| 困难 | 55/55 = 1.00 |
| 困难 | 27/55 = 0.49 |
| 困难 | 57/57 = 1.00 |
| 困难 | 35/55 = 0.64 |
| 困难 | 15/56 = 0.27 |
| 专家 | 56/56 = 1.00 |
| 专家 | 55/55 = 1.00 |
| 专家 | 54/54 = 1.00 |
| 专家 | 55/55 = 1.00 |
| 专家 | 17/55 = 0.31 |
| 专家 | 54/54 = 1.00 |
| 恶魔 | 50/50 = 1.00 |
| 恶魔 | 50/50 = 1.00 |
| 恶魔 | 49/49 = 1.00 |
| 恶魔 | 28/53 = 0.53 |
| 恶魔 | 51/51 = 1.00 |
| 恶魔 | 51/51 = 1.00 |
| 总体准确率 | 1345/1568 = 0.86 |
参考文献
如果您将此代码用于研究,请引用以下内容:
@misc{sudoku2018,
author = {Park, Kyubyong},
title = {卷积神经网络能否破解数独谜题?},
year = {2018},
publisher = {GitHub},
journal = {GitHub仓库},
howpublished = {\url{https://github.com/Kyubyong/sudoku}}
}
引用本仓库的相关论文
相似工具推荐
ML-For-Beginners
ML-For-Beginners 是由微软推出的一套系统化机器学习入门课程,旨在帮助零基础用户轻松掌握经典机器学习知识。这套课程将学习路径规划为 12 周,包含 26 节精炼课程和 52 道配套测验,内容涵盖从基础概念到实际应用的完整流程,有效解决了初学者面对庞大知识体系时无从下手、缺乏结构化指导的痛点。 无论是希望转型的开发者、需要补充算法背景的研究人员,还是对人工智能充满好奇的普通爱好者,都能从中受益。课程不仅提供了清晰的理论讲解,还强调动手实践,让用户在循序渐进中建立扎实的技能基础。其独特的亮点在于强大的多语言支持,通过自动化机制提供了包括简体中文在内的 50 多种语言版本,极大地降低了全球不同背景用户的学习门槛。此外,项目采用开源协作模式,社区活跃且内容持续更新,确保学习者能获取前沿且准确的技术资讯。如果你正寻找一条清晰、友好且专业的机器学习入门之路,ML-For-Beginners 将是理想的起点。
funNLP
funNLP 是一个专为中文自然语言处理(NLP)打造的超级资源库,被誉为"NLP 民工的乐园”。它并非单一的软件工具,而是一个汇集了海量开源项目、数据集、预训练模型和实用代码的综合性平台。 面对中文 NLP 领域资源分散、入门门槛高以及特定场景数据匮乏的痛点,funNLP 提供了“一站式”解决方案。这里不仅涵盖了分词、命名实体识别、情感分析、文本摘要等基础任务的标准工具,还独特地收录了丰富的垂直领域资源,如法律、医疗、金融行业的专用词库与数据集,甚至包含古诗词生成、歌词创作等趣味应用。其核心亮点在于极高的全面性与实用性,从基础的字典词典到前沿的 BERT、GPT-2 模型代码,再到高质量的标注数据和竞赛方案,应有尽有。 无论是刚刚踏入 NLP 领域的学生、需要快速验证想法的算法工程师,还是从事人工智能研究的学者,都能在这里找到急需的“武器弹药”。对于开发者而言,它能大幅减少寻找数据和复现模型的时间;对于研究者,它提供了丰富的基准测试资源和前沿技术参考。funNLP 以开放共享的精神,极大地降低了中文自然语言处理的开发与研究成本,是中文 AI 社区不可或缺的宝藏仓库。
cs-video-courses
cs-video-courses 是一个精心整理的计算机科学视频课程清单,旨在为自学者提供系统化的学习路径。它汇集了全球知名高校(如加州大学伯克利分校、新南威尔士大学等)的完整课程录像,涵盖从编程基础、数据结构与算法,到操作系统、分布式系统、数据库等核心领域,并深入延伸至人工智能、机器学习、量子计算及区块链等前沿方向。 面对网络上零散且质量参差不齐的教学资源,cs-video-courses 解决了学习者难以找到成体系、高难度大学级别课程的痛点。该项目严格筛选内容,仅收录真正的大学层级课程,排除了碎片化的简短教程或商业广告,确保用户能接触到严谨的学术内容。 这份清单特别适合希望夯实计算机基础的开发者、需要补充特定领域知识的研究人员,以及渴望像在校生一样系统学习计算机科学的自学者。其独特的技术亮点在于分类极其详尽,不仅包含传统的软件工程与网络安全,还细分了生成式 AI、大语言模型、计算生物学等新兴学科,并直接链接至官方视频播放列表,让用户能一站式获取高质量的教育资源,免费享受世界顶尖大学的课堂体验。
ragflow
RAGFlow 是一款领先的开源检索增强生成(RAG)引擎,旨在为大语言模型构建更精准、可靠的上下文层。它巧妙地将前沿的 RAG 技术与智能体(Agent)能力相结合,不仅支持从各类文档中高效提取知识,还能让模型基于这些知识进行逻辑推理和任务执行。 在大模型应用中,幻觉问题和知识滞后是常见痛点。RAGFlow 通过深度解析复杂文档结构(如表格、图表及混合排版),显著提升了信息检索的准确度,从而有效减少模型“胡编乱造”的现象,确保回答既有据可依又具备时效性。其内置的智能体机制更进一步,使系统不仅能回答问题,还能自主规划步骤解决复杂问题。 这款工具特别适合开发者、企业技术团队以及 AI 研究人员使用。无论是希望快速搭建私有知识库问答系统,还是致力于探索大模型在垂直领域落地的创新者,都能从中受益。RAGFlow 提供了可视化的工作流编排界面和灵活的 API 接口,既降低了非算法背景用户的上手门槛,也满足了专业开发者对系统深度定制的需求。作为基于 Apache 2.0 协议开源的项目,它正成为连接通用大模型与行业专有知识之间的重要桥梁。
PaddleOCR
PaddleOCR 是一款基于百度飞桨框架开发的高性能开源光学字符识别工具包。它的核心能力是将图片、PDF 等文档中的文字提取出来,转换成计算机可读取的结构化数据,让机器真正“看懂”图文内容。 面对海量纸质或电子文档,PaddleOCR 解决了人工录入效率低、数字化成本高的问题。尤其在人工智能领域,它扮演着连接图像与大型语言模型(LLM)的桥梁角色,能将视觉信息直接转化为文本输入,助力智能问答、文档分析等应用场景落地。 PaddleOCR 适合开发者、算法研究人员以及有文档自动化需求的普通用户。其技术优势十分明显:不仅支持全球 100 多种语言的识别,还能在 Windows、Linux、macOS 等多个系统上运行,并灵活适配 CPU、GPU、NPU 等各类硬件。作为一个轻量级且社区活跃的开源项目,PaddleOCR 既能满足快速集成的需求,也能支撑前沿的视觉语言研究,是处理文字识别任务的理想选择。
awesome-machine-learning
awesome-machine-learning 是一份精心整理的机器学习资源清单,汇集了全球优秀的机器学习框架、库和软件工具。面对机器学习领域技术迭代快、资源分散且难以甄选的痛点,这份清单按编程语言(如 Python、C++、Go 等)和应用场景(如计算机视觉、自然语言处理、深度学习等)进行了系统化分类,帮助使用者快速定位高质量项目。 它特别适合开发者、数据科学家及研究人员使用。无论是初学者寻找入门库,还是资深工程师对比不同语言的技术选型,都能从中获得极具价值的参考。此外,清单还延伸提供了免费书籍、在线课程、行业会议、技术博客及线下聚会等丰富资源,构建了从学习到实践的全链路支持体系。 其独特亮点在于严格的维护标准:明确标记已停止维护或长期未更新的项目,确保推荐内容的时效性与可靠性。作为机器学习领域的“导航图”,awesome-machine-learning 以开源协作的方式持续更新,旨在降低技术探索门槛,让每一位从业者都能高效地站在巨人的肩膀上创新。