arXiv · 2110.09039
Capsule Graph Neural Networks with EM Routing
Abstract
To effectively classify graph instances, graph neural networks need to have the capability to capture the part-whole relationship existing in a graph. A capsule is a group of neurons representing complicated properties of entities, which has shown its advantages in traditional convolutional neural networks. This paper proposed novel Capsule Graph Neural Networks that use the EM routing mechanism (CapsGNNEM) to generate high-quality graph embeddings. Experimental results on a number of real-world graph datasets demonstrate that the proposed CapsGNNEM outperforms nine state-of-the-art models in graph classification tasks.
Explore related subjects
Keep this discovery
Yu Lei, Jing Zhang. 2021-10-18. Capsule Graph Neural Networks with EM Routing. https://doi.org/10.1145/3459637.3482069
Cite the original work for its findings. Save a collection to share your selection of sources.