GTNet: A Tree-Based Deep Graph Learning Architecture
About
We propose Graph Tree Networks (GTNets), a deep graph learning architecture with a new general message passing scheme that originates from the tree representation of graphs. In the tree representation, messages propagate upward from the leaf nodes to the root node, and each node preserves its initial information prior to receiving information from its child nodes (neighbors). We formulate a general propagation rule following the nature of message passing in the tree to update a node's feature by aggregating its initial feature and its neighbor nodes' updated features. Two graph representation learning models are proposed within this GTNet architecture - Graph Tree Attention Network (GTAN) and Graph Tree Convolution Network (GTCN), with experimentally demonstrated state-of-the-art performance on several popular benchmark datasets. Unlike the vanilla Graph Attention Network (GAT) and Graph Convolution Network (GCN) which have the "over-smoothing" issue, the proposed GTAN and GTCN models can go deep as demonstrated by comprehensive experiments and rigorous theoretical analysis.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | Pubmed | -- | 742 | |
| Node Classification | Cora (semi-supervised) | Accuracy84.5 | 103 | |
| Semi-supervised node classification | Pubmed | Accuracy79.6 | 60 | |
| Semi-supervised node classification | Citeseer | Accuracy72.9 | 31 | |
| Node Classification | Coauthor CS | Macro F190.6 | 24 | |
| Node Classification | Citeseer | Macro F169.1 | 17 | |
| Node Classification | Cora | Macro F182.8 | 17 | |
| Node Classification | Coauthor-CS 2 | Macro F190.5 | 7 | |
| Node Classification | ogbn-arxiv (fixed) | Accuracy0.73 | 7 | |
| Semi-supervised node classification | Coauthor-CS (split 1) | Accuracy92.7 | 7 |