Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Graph Capsule Convolutional Neural Networks

About

Graph Convolutional Neural Networks (GCNNs) are the most recent exciting advancement in deep learning field and their applications are quickly spreading in multi-cross-domains including bioinformatics, chemoinformatics, social networks, natural language processing and computer vision. In this paper, we expose and tackle some of the basic weaknesses of a GCNN model with a capsule idea presented in \cite{hinton2011transforming} and propose our Graph Capsule Network (GCAPS-CNN) model. In addition, we design our GCAPS-CNN model to solve especially graph classification problem which current GCNN models find challenging. Through extensive experiments, we show that our proposed Graph Capsule Network can significantly outperforms both the existing state-of-art deep learning methods and graph kernels on graph classification benchmark datasets.

Saurabh Verma, Zhi-Li Zhang• 2018

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy76.4
1383
Graph ClassificationNCI1
Accuracy82.72
707
Graph ClassificationCOLLAB
Accuracy77.71
532
Graph ClassificationIMDB-B
Accuracy71.69
455
Graph ClassificationIMDB-M
Accuracy48.5
434
Graph ClassificationENZYMES
Accuracy61.83
419
Graph ClassificationDD
Accuracy77.62
309
Graph ClassificationNCI109
Accuracy81.12
275
Graph ClassificationD&D
Accuracy77.62
179
Graph ClassificationIMDB MULTI
Accuracy48.5
168
Showing 10 of 13 rows

Other info

Code

Follow for update