arXiv:2511.09209cs.LG2025-11AAAI被引 5

通过对比与竞争机制提升混合整数规划求解精度

CoCo-MILP: Inter-Variable Contrastive and Intra-Constraint Competitive MILP Solution Prediction

  • 引入变量间对比损失,强化重要变量的区分度
  • 设计约束内竞争GNN层,捕捉变量间的排他性关系
  • 在标准测试集上比传统求解器降低68.12%解差距

混合整数线性规划(MILP)是组合优化的核心,但大规模实例求解仍面临巨大计算挑战。近年来,图神经网络(GNN)在加速MILP求解器方面展现出潜力,通过预测高质量解来提升效率。然而,现有方法在两个层面与MILP问题内在结构存在偏差:在学习目标层面,二元交叉熵(BCE)损失将变量独立处理,忽略了其相对优先级,导致输出逻辑不具区分性;在模型架构层面,标准GNN消息传递机制会平滑变量表示,忽视了约束内部变量间的自然竞争关系。为此,我们提出CoCo-MILP,显式建模变量间的对比与约束内的竞争关系。在目标层面,引入变量对比损失(VCL),显式最大化赋值为1与0变量之间的嵌入距离;在架构层面,设计约束内竞争GNN层,不进行特征均质化,而是学习区分同一约束中相互竞争变量的表示,捕捉其互斥特性。在标准基准上的实验表明,CoCo-MILP显著优于现有基于学习的方法,相比传统求解器解差距最高降低68.12%。代码已开源:https://github.com/happypu326/CoCo-MILP。

原文摘要 · Abstract (English)

Mixed-Integer Linear Programming (MILP) is a cornerstone of combinatorial optimization, yet solving large-scale instances remains a significant computational challenge. Recently, Graph Neural Networks (GNNs) have shown promise in accelerating MILP solvers by predicting high-quality solutions. However, we identify that existing methods misalign with the intrinsic structure of MILP problems at two levels. At the leaning objective level, the Binary Cross-Entropy (BCE) loss treats variables independently, neglecting their relative priority and yielding plausible logits. At the model architecture level, standard GNN message passing inherently smooths the representations across variables, missing the natural competitive relationships within constraints. To address these challenges, we propose CoCo-MILP, which explicitly models inter-variable Contrast and intra-constraint Competition for advanced MILP solution prediction. At the objective level, CoCo-MILP introduces the Inter-Variable Contrastive Loss (VCL), which explicitly maximizes the embedding margin between variables assigned one versus zero. At the architectural level, we design an Intra-Constraint Competitive GNN layer that, instead of homogenizing features, learns to differentiate representations of competing variables within a constraint, capturing their exclusionary nature. Experimental results on standard benchmarks demonstrate that CoCo-MILP significantly outperforms existing learning-based approaches, reducing the solution gap by up to 68.12% compared to traditional solvers. Our code is available at https://github.com/happypu326/CoCo-MILP.

MILP图神经网络优化对比学习

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