Learning to Rank the Initial Branching Order of SAT Solvers
Organizations: KTH Royal Institute of Technology · Kempner Institute at Harvard University · Mohamed Bin Zayed University of Artificial Intelligence
Abstract
Finding good branching orders is key to solving SAT problems efficiently, but finding such branching orders is a difficult problem. Using a learning based approach to predict a good branching order before solving, therefore, has potential. In this paper, we investigate predicting branching orders using graph neural networks as a preprocessing step to conflict-driven clause learning (CDCL) SAT solvers. We show that there are significant gains to be made in existing CDCL SAT solvers by providing a good initial branching. Further, we provide three labeling methods to find such initial branching orders in a tractable way. Finally, we train a graph neural network to predict these branching orders and show through our evaluations that a GNN-initialized ordering yields significant speedups on random 3-CNF and pseudo-industrial benchmarks, with generalization capabilities to instances much larger than the training set. However, we also find that the predictions fail at speeding up more difficult and industrial instances. We attribute this to the solver's dynamic heuristics, which rapidly overwrite the provided initialization, and to the complexity of these instances, making GNN prediction hard.
Figures & tables
| Instance | Variables | Clauses | Median (s) | Best speedup (%) | Top 10% speedup (%) |
|---|---|---|---|---|---|
| 3-CNF 100 | |||||
| 3-CNF 150 | |||||
| 3-CNF 200 | |||||
| 3-CNF 250 | |||||
| Ramsey | |||||
| Manthey |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | Value |
|---|---|
| Model Architecture | Graph Transformer from NeuroBack ( Wang et al., 2024 ) |
| Optimizer | AdamW ( Loshchilov and Hutter, 2018 ) |
| Adam | 0.9 |
| Adam | 0.999 |
| Learning Rate | 0.001 |
| Epochs | 10 |