Saved in:
Bibliographic Details
Main Authors: Zhou, Dongzhuoran, Yang, Hui, Xiong, Bo, Ma, Yue, Kharlamov, Evgeny
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2407.19231
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910544871555072
author Zhou, Dongzhuoran
Yang, Hui
Xiong, Bo
Ma, Yue
Kharlamov, Evgeny
author_facet Zhou, Dongzhuoran
Yang, Hui
Xiong, Bo
Ma, Yue
Kharlamov, Evgeny
contents Graph neural networks (GNNs) have achieved significant success in various applications. Most GNNs learn the node features with information aggregation of its neighbors and feature transformation in each layer. However, the node features become indistinguishable after many layers, leading to performance deterioration: a significant limitation known as over-smoothing. Past work adopted various techniques for addressing this issue, such as normalization and skip-connection of layer-wise output. After the study, we found that the information aggregations in existing work are all contracted aggregations, with the intrinsic property that features will inevitably converge to the same single point after many layers. To this end, we propose the aggregation over compacted manifolds method (ACM) that replaces the existing information aggregation with aggregation over compact manifolds, a special type of manifold, which avoids contracted aggregations. In this work, we theoretically analyze contracted aggregation and its properties. We also provide an extensive empirical evaluation that shows ACM can effectively alleviate over-smoothing and outperforms the state-of-the-art. The code can be found in https://github.com/DongzhuoranZhou/ACM.git.
format Preprint
id arxiv_https___arxiv_org_abs_2407_19231
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Alleviating Over-Smoothing via Aggregation over Compact Manifolds
Zhou, Dongzhuoran
Yang, Hui
Xiong, Bo
Ma, Yue
Kharlamov, Evgeny
Machine Learning
Graph neural networks (GNNs) have achieved significant success in various applications. Most GNNs learn the node features with information aggregation of its neighbors and feature transformation in each layer. However, the node features become indistinguishable after many layers, leading to performance deterioration: a significant limitation known as over-smoothing. Past work adopted various techniques for addressing this issue, such as normalization and skip-connection of layer-wise output. After the study, we found that the information aggregations in existing work are all contracted aggregations, with the intrinsic property that features will inevitably converge to the same single point after many layers. To this end, we propose the aggregation over compacted manifolds method (ACM) that replaces the existing information aggregation with aggregation over compact manifolds, a special type of manifold, which avoids contracted aggregations. In this work, we theoretically analyze contracted aggregation and its properties. We also provide an extensive empirical evaluation that shows ACM can effectively alleviate over-smoothing and outperforms the state-of-the-art. The code can be found in https://github.com/DongzhuoranZhou/ACM.git.
title Alleviating Over-Smoothing via Aggregation over Compact Manifolds
topic Machine Learning
url https://arxiv.org/abs/2407.19231