Skip to main content
SHARE
Publication

OpenGraphGym: A Parallel Reinforcement Learning Framework for Graph Optimization Problems...

by Weijian Zheng, Dali Wang
Publication Type
Conference Paper
Book Title
International Conference on Computational Science
Publication Date
Page Numbers
439 to 452
Volume
12141
Conference Name
International Conference on Computational Sciences (ICCS)
Conference Location
Amsterdam, Netherlands
Conference Sponsor
N/A
Conference Date

This paper presents an open-source, parallel AI environment (named OpenGraphGym) to facilitate the application of reinforcement learning (RL) algorithms to address combinatorial graph optimization problems. This environment incorporates a basic deep reinforcement learning method, and several graph embeddings to capture graph features, it also allows users to rapidly plug in and test new RL algorithms and graph embeddings for graph optimization problems. This new open-source RL framework is targeted at achieving both high performance and high quality of the computed graph solutions. This RL framework forms the foundation of several ongoing research directions, including 1) benchmark works on different RL algorithms and embedding methods for classic graph problems; 2) advanced parallel strategies for extreme-scale graph computations, as well as 3) performance evaluation on real-world graph solutions.