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

Computer Science > Machine Learning

Title: Distributed Training of Graph Convolutional Networks

Abstract: The aim of this work is to develop a fully-distributed algorithmic framework for training graph convolutional networks (GCNs). The proposed method is able to exploit the meaningful relational structure of the input data, which are collected by a set of agents that communicate over a sparse network topology. After formulating the centralized GCN training problem, we first show how to make inference in a distributed scenario where the underlying data graph is split among different agents. Then, we propose a distributed gradient descent procedure to solve the GCN training problem. The resulting model distributes computation along three lines: during inference, during back-propagation, and during optimization. Convergence to stationary solutions of the GCN training problem is also established under mild conditions. Finally, we propose an optimization criterion to design the communication topology between agents in order to match with the graph describing data relationships. A wide set of numerical results validate our proposal. To the best of our knowledge, this is the first work combining graph convolutional neural networks with distributed optimization.
Comments: Published on IEEE Transactions on Signal and Information Processing over Networks
Subjects: Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE); Machine Learning (stat.ML)
Journal reference: IEEE Transactions on Signal and Information Processing over Networks, vol. 7, pp. 87-100, 2021
DOI: 10.1109/TSIPN.2020.3046237
Cite as: arXiv:2007.06281 [cs.LG]
  (or arXiv:2007.06281v2 [cs.LG] for this version)

Submission history

From: Simone Scardapane [view email]
[v1] Mon, 13 Jul 2020 10:04:20 GMT (5265kb,D)
[v2] Thu, 7 Jan 2021 10:00:24 GMT (4876kb,D)

Link back to: arXiv, form interface, contact.