Ricci Curvature-Based Semi-Supervised Learning on an Attributed Network
Open Access
- 27 February 2021
- Vol. 23 (3), 292
- https://doi.org/10.3390/e23030292
Abstract
In recent years, on the basis of drawing lessons from traditional neural network models, people have been paying more and more attention to the design of neural network architectures for processing graph structure data, which are called graph neural networks (GNN). GCN, namely, graph convolution networks, are neural network models in GNN. GCN extends the convolution operation from traditional data (such as images) to graph data, and it is essentially a feature extractor, which aggregates the features of neighborhood nodes into those of target nodes. In the process of aggregating features, GCN uses the Laplacian matrix to assign different importance to the nodes in the neighborhood of the target nodes. Since graph-structured data are inherently non-Euclidean, we seek to use a non-Euclidean mathematical tool, namely, Riemannian geometry, to analyze graphs (networks). In this paper, we present a novel model for semi-supervised learning called the Ricci curvature-based graph convolutional neural network, i.e., RCGCN. The aggregation pattern of RCGCN is inspired by that of GCN. We regard the network as a discrete manifold, and then use Ricci curvature to assign different importance to the nodes within the neighborhood of the target nodes. Ricci curvature is related to the optimal transport distance, which can well reflect the geometric structure of the underlying space of the network. The node importance given by Ricci curvature can better reflect the relationships between the target node and the nodes in the neighborhood. The proposed model scales linearly with the number of edges in the network. Experiments demonstrated that RCGCN achieves a significant performance gain over baseline methods on benchmark datasets.This publication has 16 references indexed in Scilit:
- GraRepPublished by Association for Computing Machinery (ACM) ,2015
- LINEPublished by Association for Computing Machinery (ACM) ,2015
- DeepWalkPublished by Association for Computing Machinery (ACM) ,2014
- Ollivier’s Ricci Curvature, Local Clustering and Curvature-Dimension Inequalities on GraphsDiscrete & Computational Geometry, 2013
- Ollivier’s Ricci curvature and the coloring of graphsEuropean Journal of Combinatorics, 2013
- Ollivier–Ricci curvature and the spectrum of the normalized graph Laplace operatorMathematical Research Letters, 2012
- Temporal link prediction by integrating content and structure informationPublished by Association for Computing Machinery (ACM) ,2011
- Ricci curvature of graphsTohoku Mathematical Journal, 2011
- Ricci curvature of Markov chains on metric spacesJournal of Functional Analysis, 2009
- Transport inequalities, gradient estimates, entropy and Ricci curvatureCommunications on Pure and Applied Mathematics, 2004