Spectral Graph Pruning Against Over-Squashing and Over-Smoothing

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Jamadandi, Adarsh, Rubio-Madrigal, Celia, Burkholz, Rebekka
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909371085094912
author Jamadandi, Adarsh
Rubio-Madrigal, Celia
Burkholz, Rebekka
author_facet Jamadandi, Adarsh
Rubio-Madrigal, Celia
Burkholz, Rebekka
contents Message Passing Graph Neural Networks are known to suffer from two problems that are sometimes believed to be diametrically opposed: over-squashing and over-smoothing. The former results from topological bottlenecks that hamper the information flow from distant nodes and are mitigated by spectral gap maximization, primarily, by means of edge additions. However, such additions often promote over-smoothing that renders nodes of different classes less distinguishable. Inspired by the Braess phenomenon, we argue that deleting edges can address over-squashing and over-smoothing simultaneously. This insight explains how edge deletions can improve generalization, thus connecting spectral gap optimization to a seemingly disconnected objective of reducing computational resources by pruning graphs for lottery tickets. To this end, we propose a more effective spectral gap optimization framework to add or delete edges and demonstrate its effectiveness on large heterophilic datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2404_04612
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Spectral Graph Pruning Against Over-Squashing and Over-Smoothing
Jamadandi, Adarsh
Rubio-Madrigal, Celia
Burkholz, Rebekka
Machine Learning
Signal Processing
Message Passing Graph Neural Networks are known to suffer from two problems that are sometimes believed to be diametrically opposed: over-squashing and over-smoothing. The former results from topological bottlenecks that hamper the information flow from distant nodes and are mitigated by spectral gap maximization, primarily, by means of edge additions. However, such additions often promote over-smoothing that renders nodes of different classes less distinguishable. Inspired by the Braess phenomenon, we argue that deleting edges can address over-squashing and over-smoothing simultaneously. This insight explains how edge deletions can improve generalization, thus connecting spectral gap optimization to a seemingly disconnected objective of reducing computational resources by pruning graphs for lottery tickets. To this end, we propose a more effective spectral gap optimization framework to add or delete edges and demonstrate its effectiveness on large heterophilic datasets.
title Spectral Graph Pruning Against Over-Squashing and Over-Smoothing
topic Machine Learning
Signal Processing
url https://arxiv.org/abs/2404.04612