Command Palette
Search for a command to run...
CensNet: Convolution with Edge-Node Switching in Graph Neural Networks
{Sheng Li Pengsheng Ji Xiaodong Jiang}

Abstract
In this paper, we present CensNet, Convolution with Edge-Node Switching graph neural network, for semi-supervised classification and regression in graph-structured data with both node and edge features. CensNet is a general graph embedding framework, which embeds both nodes and edges to a latent feature space. By using line graph of the original undirected graph, the role of nodes and edges are switched, and two novel graph convolution operations are proposed for feature propagation. Experimental results on real-world academic citation networks and quantum chemistry graphs show that our approach has achieved or matched the state-of-the-art performance.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| graph-regression-on-lipophilicity | Random Forests | RMSE@80%Train: 1.16 |
| graph-regression-on-lipophilicity | CensNet | RMSE@80%Train: 0.93 |
| graph-regression-on-lipophilicity | Logistic Regression | RMSE@80%Train: 1.15 |
| graph-regression-on-tox21 | Logistic Regression | AUC@80%Train: 0.71 |
| graph-regression-on-tox21 | Random Forest | AUC@80%Train: 0.71 |
| graph-regression-on-tox21 | CensNet | AUC@80%Train: 0.78 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.