Neural Bipartite Matching

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Georgiev, Dobrik, Liò, Pietro
Natura: Preprint
Pubblicazione: 2020
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929416684175360
author Georgiev, Dobrik
Liò, Pietro
author_facet Georgiev, Dobrik
Liò, Pietro
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2005_11304
institution arXiv
publishDate 2020
record_format arxiv
spellingShingle Neural Bipartite Matching
Georgiev, Dobrik
Liò, Pietro
Machine Learning
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.
title Neural Bipartite Matching
topic Machine Learning
url https://arxiv.org/abs/2005.11304