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: A Capsule Network-based Model for Learning Node Embeddings

Abstract: In this paper, we focus on learning low-dimensional embeddings for nodes in graph-structured data. To achieve this, we propose Caps2NE -- a new unsupervised embedding model leveraging a network of two capsule layers. Caps2NE induces a routing process to aggregate feature vectors of context neighbors of a given target node at the first capsule layer, then feed these features into the second capsule layer to infer a plausible embedding for the target node. Experimental results show that our proposed Caps2NE obtains state-of-the-art performances on benchmark datasets for the node classification task. Our code is available at: \url{this https URL}.
Comments: Extended version of our CIKM 2020 paper, including inductive results
Subjects: Machine Learning (cs.LG); Computation and Language (cs.CL); Machine Learning (stat.ML)
Cite as: arXiv:1911.04822 [cs.LG]
  (or arXiv:1911.04822v2 [cs.LG] for this version)

Submission history

From: Dai Quoc Nguyen [view email]
[v1] Tue, 12 Nov 2019 12:44:26 GMT (178kb,D)
[v2] Tue, 18 Aug 2020 15:48:59 GMT (74kb,D)

Link back to: arXiv, form interface, contact.