Community Attention Network for Semi-supervised Node Classification.

Zhongjing Yu,Han Wang,Yang Liu,Christian Bohm,Junming Shao
DOI: https://doi.org/10.1109/icdm50108.2020.00181
2020-01-01
Abstract:Graph neural networks (GNNs) have achieved great success for semi-supervised node classification by embedding node representation into a low-dimensional space.. However, existing approaches usually ignore one intrinsic property of graphs: community structure, where the formation of distinct communities in graphs is often driven by different subset of attributes. In this paper, we introduce a new method, called Community Attention Network (CAT), aiming to extract community-specific features and then enhance node embeddings for classification. To learn such community-specific information, we design a new loss function to ensure the nodes in the same community should share similar attributes (i.e., low covariance), and any unlabelled node should belong to only one class with high probability i.e.,( low community distribution entropy) in a community attention network. Extensive experimental results demonstrate the effectiveness of CAT and its advantages over many state-of-the-art approaches. To further illustrate the benefits of CAT to capture the community information, a case study is given and discussed.
What problem does this paper attempt to address?