Preprint
Reinforcement Learning
Featured

Targeted Branching for the Maximum Independent Set Problem Using Graph Neural Networks

Silva, Gabriel(University of Aveiro), Rodrigues, Mário(University of Aveiro), Teixeira, António(University of Aveiro), Amorim, Marlene(University of Aveiro)
January 1, 2024DROPS (Schloss Dagstuhl – Leibniz Center for Informatics)5,185 citations

5.2k

Citations

0

Influential Citations

DROPS (Schloss Dagstuhl – Leibniz Center for Informatics)

Venue

2024

Year

Abstract

Identifying a maximum independent set is a fundamental NP-hard problem. This problem has several real-world applications and requires finding the largest possible set of vertices not adjacent to each other in an undirected graph. Over the past few years, branch-and-bound and branch-and-reduce algorithms have emerged as some of the most effective methods for solving the problem exactly. Specifically, the branch-and-reduce approach, which combines branch-and-bound principles with reduction rules, has proven particularly successful in tackling previously unmanageable real-world instances. This progress was largely made possible by the development of more effective reduction rules. Nevertheless, other key components that can impact the efficiency of these algorithms have not received the same level of interest. Among these is the branching strategy, which determines which vertex to branch on next. Until recently, the most widely used strategy was to choose the vertex of the highest degree. In this work, we present a graph neural network approach for selecting the next branching vertex. The intricate nature of current branch-and-bound solvers makes supervised and reinforcement learning difficult. Therefore, we use a population-based genetic algorithm to evolve the model’s parameters instead. Our proposed approach results in a speedup on 73% of the benchmark instances with a median speedup of 24%.

Analysis

Why This Paper Matters

Branch-and-bound and branch-and-reduce algorithms are among the most effective exact methods for solving the maximum independent set (MIS) problem, an NP-hard graph problem with numerous real-world applications. While significant progress has been made through improved reduction rules, other components such as the branching strategy—which determines the next vertex to branch on—have received comparatively little attention. Traditionally, the highest-degree heuristic has been widely used, but it may not be optimal. This paper introduces a graph neural network (GNN) approach to learn a better branching strategy, potentially leading to substantial performance gains in exact solvers.

The significance of this work lies in its demonstration that machine learning can enhance a critical component of exact combinatorial solvers. By leveraging a GNN to select branching vertices, the authors show that learned heuristics can outperform traditional handcrafted rules. This is particularly relevant as the AI community increasingly explores the intersection of machine learning and combinatorial optimization, where learning-based approaches have shown promise in approximate settings but are less explored in exact algorithms.

Moreover, the paper addresses a key challenge: the complexity of branch-and-bound solvers makes standard supervised and reinforcement learning difficult to apply. The authors' use of a genetic algorithm to evolve the GNN parameters offers a practical alternative, highlighting the potential of evolutionary strategies for training models in complex, non-differentiable environments.

Technical Contributions

  • GNN-based branching strategy: The paper proposes a graph neural network that takes a graph as input and outputs a probability distribution over vertices, guiding the selection of the next branching vertex.
  • Genetic algorithm training: Due to the intricate nature of branch-and-reduce solvers, the authors employ a population-based genetic algorithm to optimize the GNN's parameters, avoiding the need for differentiable solvers or reward shaping.
  • Integration with branch-and-reduce: The learned branching strategy is integrated into a state-of-the-art branch-and-reduce solver for MIS, demonstrating that the approach can be seamlessly combined with existing reduction rules.
  • Empirical evaluation: The method is evaluated on a set of benchmark instances, showing consistent speedups over the highest-degree heuristic.

Results

The proposed approach achieves a speedup on 73% of the benchmark instances, with a median speedup of 24% compared to the highest-degree branching strategy. This indicates that the GNN-based branching strategy is effective in reducing the search tree size and improving solver performance. The results are particularly noteworthy because they are obtained on a diverse set of instances, suggesting robustness. However, the paper does not provide detailed comparisons with other learning-based approaches or an in-depth analysis of the factors contributing to the speedup.

Significance

This work underscores the potential of combining graph neural networks with evolutionary strategies to enhance exact combinatorial solvers. It opens avenues for further research into learning-based branching strategies for other NP-hard problems, such as maximum clique or graph coloring. The use of genetic algorithms for training in complex solver environments also suggests a viable path for applying machine learning where gradient-based methods are impractical. Ultimately, this paper contributes to the growing body of work at the intersection of machine learning and combinatorial optimization, demonstrating that learned heuristics can yield tangible improvements in exact algorithms.