Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Neural Bipartite Matching
(Submitted on 22 May 2020 (v1), last revised 2 Jun 2020 (this version, v3))
Abstract: Graph neural networks (GNNs) have found application for learning in the space of algorithms. However, the algorithms chosen by existing research (sorting, Breadth-First search, shortest path finding, etc.) usually align perfectly with a standard GNN architecture. This report describes how neural execution is applied to a complex algorithm, such as finding maximum bipartite matching by reducing it to a flow problem and using Ford-Fulkerson to find the maximum flow. This is achieved via neural execution based only on features generated from a single GNN. The evaluation shows strongly generalising results with the network achieving optimal matching almost 100% of the time.
Submission history
From: Dobrik Georgiev [view email][v1] Fri, 22 May 2020 17:50:38 GMT (307kb,D)
[v2] Fri, 29 May 2020 06:03:10 GMT (308kb,D)
[v3] Tue, 2 Jun 2020 13:21:41 GMT (565kb,D)
Link back to: arXiv, form interface, contact.