Heterophilous distribution propagation for Graph Neural Networks.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Graph Neural Networks (GNNs) have achieved remarkable success in various graph mining tasks by aggregating information from neighborhoods for representation learning. The success relies on the homophily assumption that nearby nodes exhibit similar behaviors, while it may be violated in many real-world graphs. Recently, heterophilous graph neural networks (HeterGNNs) have attracted increasing attention by modifying the neural message passing schema for heterophilous neighborhoods. However, they suffer from insufficient neighborhood partition and heterophily modeling, both of which are critical but challenging to break through. To tackle these challenges, in this paper, we propose heterophilous distribution propagation (HDP) for graph neural networks. Instead of aggregating information from all neighborhoods, HDP adaptively separates the neighbors into homophilous and heterophilous parts based on the pseudo assignments during training. The heterophilous neighborhood distribution is learned with orthogonality-oriented constraint via a trusted prototype contrastive learning paradigm. Both the homophilous and heterophilous patterns are propagated with a novel semantic-aware message-passing mechanism. We conduct extensive experiments on 9 benchmark datasets with different levels of homophily. Experimental results show that our method outperforms representative baselines on heterophilous datasets.

Authors

  • Zhuonan Zheng
    College of Computer Science, Zhejiang University, Hangzhou, 310027, China; Zhejiang Key Laboratory of Accessible Perception and Intelligent Systems, Zhejiang University, Hangzhou, 310027, China. Electronic address: zhengzn@zju.edu.cn.
  • Sheng Zhou
    Department of The First Clinical Medical College of Gansu, University of Chinese Medicine, Lanzhou, Gansu, China.
  • Hongjia Xu
    College of Computer Science, Zhejiang University, Hangzhou, 310027, China; Zhejiang Key Laboratory of Accessible Perception and Intelligent Systems, Zhejiang University, Hangzhou, 310027, China. Electronic address: xu_hj@zju.edu.cn.
  • Ming Gu
    School of Software, Tsinghua University, Beijing, China.
  • Yilun Xu
    Department of Gastroenterology, The First Affiliated Hospital of Wenzhou Medical University, Wenzhou, 325000, Zhejiang, China.
  • Ao Li
    Beijing University of Chinese Medicine, Beijing, China.
  • Yuhong Li
    Shanghai Wision AI Co Ltd, Shanghai, China.
  • Jingjun Gu
    College of Computer Science, Zhejiang University, Hangzhou, P. R. China.
  • Jiajun Bu