arXiv:2512.20915cs.LGmath.CO2025-12

用机器学习预测图上组合优化问题的难解性并解释原因。

Towards a General Framework for Predicting and Explaining the Hardness of Graph-based Combinatorial Optimization Problems using Machine Learning and Association Rule Mining

  • 基于图特征和硬度标签训练分类模型,仅用三个特征即达高精度。
  • 硬实例预测的加权F1达0.9921,解释规则支持率达88.29%。
  • 适合想理解算法性能瓶颈的研究者与工程优化人员。

本文提出GCO-HPIF框架,用于预测和解释可表示为图的组合优化问题的计算难度。该框架分两阶段:第一阶段构建包含问题无关图特征与实例硬度标签的数据集,训练机器学习分类模型将图特征映射至硬度类别;第二阶段利用关联规则挖掘算法解释预测结果,并训练回归模型预测算法计算时间。实验基于来自COLLAB、IMDB和TWITTER数据集的3287个最大团问题实例,采用五种先进算法(Gurobi、CliSAT、MOMC、EGN、HGS)。框架在预测硬度上表现优异,加权F1得分为0.9921,少数类F1为0.878,ROC-AUC为0.9083,仅使用三个图特征即达成。最佳关联规则在硬实例上的支持率为0.8829,整体准确率达87.64%,有效实现预测与解释。回归模型预测计算时间的百分比RMSE为5.12,R²值达0.991。

原文摘要 · Abstract (English)

This study introduces GCO-HPIF, a general machine-learning-based framework to predict and explain the computational hardness of combinatorial optimization problems that can be represented on graphs. The framework consists of two stages. In the first stage, a dataset is created comprising problem-agnostic graph features and hardness classifications of problem instances. Machine-learning-based classification algorithms are trained to map graph features to hardness categories. In the second stage, the framework explains the predictions using an association rule mining algorithm. Additionally, machine-learning-based regression models are trained to predict algorithmic computation times. The GCO-HPIF framework was applied to a dataset of 3287 maximum clique problem instances compiled from the COLLAB, IMDB, and TWITTER graph datasets using five state-of-the-art algorithms, namely three exact branch-and-bound-based algorithms (Gurobi, CliSAT, and MOMC) and two graph-neural-network-based algorithms (EGN and HGS). The framework demonstrated excellent performance in predicting instance hardness, achieving a weighted F1 score of 0.9921, a minority-class F1 score of 0.878, and an ROC-AUC score of 0.9083 using only three graph features. The best association rule found by the FP-Growth algorithm for explaining the hardness predictions had a support of 0.8829 for hard instances and an overall accuracy of 87.64 percent, underscoring the framework's usefulness for both prediction and explanation. Furthermore, the best-performing regression model for predicting computation times achieved a percentage RMSE of 5.12 and an R2 value of 0.991.

组合优化图神经网络机器学习解释预测建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。