Abstract:Graph partitioning aims to divide a graph into disjoint subsets while optimizing a specific partitioning objective. The majority of formulations related to graph partitioning exhibit NP-hardness due to their combinatorial nature. Conventional methods, like approximation algorithms or heuristics, are designed for distinct partitioning objectives and fail to achieve generalization across other important partitioning objectives. Recently machine learning-based methods have been developed that learn directly from data. Further, these methods have a distinct advantage of utilizing node features that carry additional information. However, these methods assume differentiability of target partitioning objective functions and cannot generalize for an unknown number of partitions, i.e., they assume the number of partitions is provided in advance. In this study, we develop NeuroCUT with two key innovations over previous methodologies. First, by leveraging a reinforcement learning-based framework over node representations derived from a graph neural network and positional features, NeuroCUT can accommodate any optimization objective, even those with non-differentiable functions. Second, we decouple the parameter space and the partition count making NeuroCUT inductive to any unseen number of partition, which is provided at query time. Through empirical evaluation, we demonstrate that NeuroCUT excels in identifying high-quality partitions, showcases strong generalization across a wide spectrum of partitioning objectives, and exhibits strong generalization to unseen partition count.

An approximation algorithm for graph partitioning via deterministic annealing neural network

Memoryless Cooperative Graph Search Based on the Simulated Annealing Algorithm

A deterministic annealing algorithm for approximating a solution of the min-bisection problem

An Effective Heuristic-Based Approach for Partitioning.

GAP: Generalizable Approximate Graph Partitioning Framework

Iterative algorithms for partitioned neural network approximation to partial differential equations

Graph Partitioning with Fujitsu Digital Annealer

NeuroCUT: A Neural Approach for Robust Graph Partitioning

Solving the production transportation problem via a deterministic annealing neural network method

System partitioning on MCM using a new neural network model

Probabilistic Framework For Network Partition

A hardware/software partitioning method based on graph convolution network

Algorithm for Road Network Graph Partitioning

A Feasible Graph Partition Framework for Random Walks Implemented by Parallel Computing in Big Graph

Graph Partition by Swendsen-Wang Cuts

An end-to-end bi-objective approach to deep graph partitioning

Enhancing Balanced Graph Edge Partition with Effective Local Search

Refining Graph Partitioning For Social Network Clustering

NC Algorithms for Partitioning Sparse Graphs into Induced Forests with an Application

A Feasible Graph Partition Framework for Parallel Computing of Big Graph

Partitioner Selection with EASE to Optimize Distributed Graph Processing