Node classification framework
- GNN

GCN is a recent effective algorithm which effectively learns a function incorporate both graph structure and node features for semisupervised graph based node classification. Although GCN exceeds other state-of-the-art methods, the number of parameters that need to be learned are still significantly more than the number of used samples. Because only labeled nodes and their 2-hop neighbors are used for learning parameters in GCN. As a consequence the accuracy of node classification declines sharply when decreasing the size of train data. In order to reducing parameters and making better use of unlabeled nodes, we proposes to use unsupervised representation algorithm such as autoencoder to pretrain a embedding for each node, which captures effective information from input features and reduce parameters in graph convolution layers. Experiments show that our model has increased by 0.87%, 1.96%, 1.87% compared to baseline on Cora, Citeseer and Pubmed data respectively. The performance of our model declines slower than GCN when decreasing the train samples.
View on arXiv