Saved in:
Bibliographic Details
Main Authors: Sim, Jaeyoon, Jeon, Sooyeon, Choi, InJun, Wu, Guorong, Kim, Won Hwa
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2401.11840
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914647433543680
author Sim, Jaeyoon
Jeon, Sooyeon
Choi, InJun
Wu, Guorong
Kim, Won Hwa
author_facet Sim, Jaeyoon
Jeon, Sooyeon
Choi, InJun
Wu, Guorong
Kim, Won Hwa
contents Various Graph Neural Networks (GNNs) have been successful in analyzing data in non-Euclidean spaces, however, they have limitations such as oversmoothing, i.e., information becomes excessively averaged as the number of hidden layers increases. The issue stems from the intrinsic formulation of conventional graph convolution where the nodal features are aggregated from a direct neighborhood per layer across the entire nodes in the graph. As setting different number of hidden layers per node is infeasible, recent works leverage a diffusion kernel to redefine the graph structure and incorporate information from farther nodes. Unfortunately, such approaches suffer from heavy diagonalization of a graph Laplacian or learning a large transform matrix. In this regards, we propose a diffusion learning framework, where the range of feature aggregation is controlled by the scale of a diffusion kernel. For efficient computation, we derive closed-form derivatives of approximations of the graph convolution with respect to the scale, so that node-wise range can be adaptively learned. With a downstream classifier, the entire framework is made trainable in an end-to-end manner. Our model is tested on various standard datasets for node-wise classification for the state-of-the-art performance, and it is also validated on a real-world brain network data for graph classifications to demonstrate its practicality for Alzheimer classification.
format Preprint
id arxiv_https___arxiv_org_abs_2401_11840
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning to Approximate Adaptive Kernel Convolution on Graphs
Sim, Jaeyoon
Jeon, Sooyeon
Choi, InJun
Wu, Guorong
Kim, Won Hwa
Machine Learning
Artificial Intelligence
Various Graph Neural Networks (GNNs) have been successful in analyzing data in non-Euclidean spaces, however, they have limitations such as oversmoothing, i.e., information becomes excessively averaged as the number of hidden layers increases. The issue stems from the intrinsic formulation of conventional graph convolution where the nodal features are aggregated from a direct neighborhood per layer across the entire nodes in the graph. As setting different number of hidden layers per node is infeasible, recent works leverage a diffusion kernel to redefine the graph structure and incorporate information from farther nodes. Unfortunately, such approaches suffer from heavy diagonalization of a graph Laplacian or learning a large transform matrix. In this regards, we propose a diffusion learning framework, where the range of feature aggregation is controlled by the scale of a diffusion kernel. For efficient computation, we derive closed-form derivatives of approximations of the graph convolution with respect to the scale, so that node-wise range can be adaptively learned. With a downstream classifier, the entire framework is made trainable in an end-to-end manner. Our model is tested on various standard datasets for node-wise classification for the state-of-the-art performance, and it is also validated on a real-world brain network data for graph classifications to demonstrate its practicality for Alzheimer classification.
title Learning to Approximate Adaptive Kernel Convolution on Graphs
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2401.11840