We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Neural Bipartite Matching

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.
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:2005.11304 [cs.LG]
  (or arXiv:2005.11304v3 [cs.LG] for this version)

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.