Adaptive node-level weighted learning for directed graph neural network.

Jincheng Huang, Xiaofeng Zhu
Author Information
  1. Jincheng Huang: School of Computer Science and Engineering, University of Electronic Science and Technology of China, Chengdu, 611731, China. Electronic address: huangjc0429@gmail.com.
  2. Xiaofeng Zhu: School of Computer Science and Engineering, University of Electronic Science and Technology of China, Chengdu, 611731, China; Shenzhen Institute for Advanced Study, University of Electronic Science and Technology of China, Shenzhen, 518000, China. Electronic address: seanzhuxf@gmail.com.

Abstract

Directed graph neural networks (DGNNs) have garnered increasing interest, yet few studies have focused on node-level representation in directed graphs. In this paper, we argue that different nodes rely on neighbor information from different directions. Furthermore, the commonly used mean aggregation for in-neighbor sets and out-neighbor sets may lose expressive power for certain nodes. To achieve this, first, we estimate the homophily of each node to neighbors in different directions by extending the Dirichlet energy. This approach allows us to assign larger weights to neighbors in directions exhibiting higher homophilic ratios for any node. Second, we introduce out-degree and in-degree information in the learning of weights to avoid the problem of weak expressive power ability of mean aggregation. Moreover, we theoretically demonstrate that our method enhances the expressive ability of directed graphs. Extensive experiments on seven real-world datasets demonstrate that our method outperforms state-of-the-art approaches in both node classification and link prediction tasks.

Keywords

Word Cloud

Created with Highcharts 10.0.0graphdirecteddifferentdirectionsexpressivenodelearningDirectedneuralnode-levelrepresentationgraphsnodesinformationmeanaggregationsetspowerneighborsweightsabilitydemonstratemethodGraphnetworksDGNNsgarneredincreasinginterestyetstudiesfocusedpaperarguerelyneighborFurthermorecommonlyusedin-neighborout-neighbormaylosecertainachievefirstestimatehomophilyextendingDirichletenergyapproachallowsusassignlargerexhibitinghigherhomophilicratiosSecondintroduceout-degreein-degreeavoidproblemweakMoreovertheoreticallyenhancesExtensiveexperimentssevenreal-worlddatasetsoutperformsstate-of-the-artapproachesclassificationlinkpredictiontasksAdaptiveweightednetworkheterophily

Similar Articles

Cited By

No available data.