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

Download:

Current browse context:

cs.LG

Change to browse by:

cs

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: Label-GCN: An Effective Method for Adding Label Propagation to Graph Convolutional Networks

Abstract: We show that a modification of the first layer of a Graph Convolutional Network (GCN) can be used to effectively propagate label information across neighbor nodes, for binary and multi-class classification problems. This is done by selectively eliminating self-loops for the label features during the training phase of a GCN. The GCN architecture is otherwise unchanged, without any extra hyper-parameters, and can be used in both a transductive and inductive setting. We show through several experiments that, depending on how many labels are available during the inference phase, this strategy can lead to a substantial improvement in the model performance compared to a standard GCN approach, including with imbalanced datasets.
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2104.02153 [cs.LG]
  (or arXiv:2104.02153v1 [cs.LG] for this version)

Submission history

From: Claudio Bellei [view email]
[v1] Mon, 5 Apr 2021 21:02:48 GMT (393kb,D)

Link back to: arXiv, form interface, contact.