人工知能の分野において、制約充足問題(CSP)の求解は中核的な課題であり続けている。数独は典型的な制約充足問題であり、厳格な離散制約の下での大域的な構造推論を必要とする。既存の解法は主に二つのアプローチに大別される。すなわち、伝統的なヒューリスティックに基づく記号ソルバーと、深層学習に基づくソルバーである。しかし、これらの手法には相補的な限界がある。学習ベースのソルバーは厳密な正しさの保証を欠き、完全な記号ソルバーは長尾の探索に陥りやすい。これらの欠点を解決するため、研究者らはDiBS(Diffusion-Informed Branch Selection)と呼ばれる新しい拡散モデル誘導アプローチを提案した。これは分枝選択探索プロセスを導くものである。
DiBSの中心的なアイデアは、記号ソルバーの完全性を維持しながら、拡散モデルを分枝の順序付けガイドとして利用することにある。具体的には、現在の部分割り当てと軽量な整合性信号の下で候補値をランク付けし、探索木における分枝の探索順序を決定する。この学習に基づくガイダンスにより、困難なインスタンスにおいて分枝順序の誤りが引き起こす指数関数的な探索爆発を効果的に回避できる。著者らは、拡散モデルが大域的な構造情報を捉えることで、分枝選択に対してより優れた優先順位を提供できると指摘している。
手法の有効性を検証するため、研究チームは挑戦的なRoyle 17手がかり数独ベンチマークで実験を行った。その結果、強力なヒューリスティックベースラインと比較して、DiBSは探索コストを大幅に削減し、特に探索ノード数、バックトラック数、長尾パーセンタイル(最も困難なインスタンスの探索時間分布)において優れた性能を示した。これらの結果は、学習された大域ガイダンスが、分枝順序の誤りが高コストとなる困難なインスタンスに対して特に有効であることを裏付けている。
さらに、論文はDiBSがなぜ、そしてどのように機能するかを明らかにする詳細な理論的証明を提供しており、この手法に堅固な理論的基盤を与えている。本研究はBo Liuら7名の著者によって共同で行われ、論文は全12ページで6図と3表を含み、arXivの人工知能(cs.AI)および機械学習(cs.LG)カテゴリに収録されている。すべてのコードはGitHubで公開されており、コミュニティによる利用とさらなる研究が可能である。