Superposition in Graph Neural Networks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Pertl, Lukas, Xuanyuan, Han, Liò, Pietro
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917205991489536
author Pertl, Lukas
Xuanyuan, Han
Liò, Pietro
author_facet Pertl, Lukas
Xuanyuan, Han
Liò, Pietro
contents Interpreting graph neural networks (GNNs) is difficult because message passing mixes signals and internal channels rarely align with human concepts. We study superposition, the sharing of directions by multiple features, directly in the latent space of GNNs. Using controlled experiments with unambiguous graph concepts, we extract features as (i) class-conditional centroids at the graph level and (ii) linear-probe directions at the node level, and then analyze their geometry with simple basis-invariant diagnostics. Across GCN/GIN/GAT we find: increasing width produces a phase pattern in overlap; topology imprints overlap onto node-level features that pooling partially remixes into task-aligned graph axes; sharper pooling increases axis alignment and reduces channel sharing; and shallow models can settle into metastable low-rank embeddings. These results connect representational geometry with concrete design choices (width, pooling, and final-layer activations) and suggest practical approaches for more interpretable GNNs.
format Preprint
id arxiv_https___arxiv_org_abs_2509_00928
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Superposition in Graph Neural Networks
Pertl, Lukas
Xuanyuan, Han
Liò, Pietro
Machine Learning
Artificial Intelligence
Interpreting graph neural networks (GNNs) is difficult because message passing mixes signals and internal channels rarely align with human concepts. We study superposition, the sharing of directions by multiple features, directly in the latent space of GNNs. Using controlled experiments with unambiguous graph concepts, we extract features as (i) class-conditional centroids at the graph level and (ii) linear-probe directions at the node level, and then analyze their geometry with simple basis-invariant diagnostics. Across GCN/GIN/GAT we find: increasing width produces a phase pattern in overlap; topology imprints overlap onto node-level features that pooling partially remixes into task-aligned graph axes; sharper pooling increases axis alignment and reduces channel sharing; and shallow models can settle into metastable low-rank embeddings. These results connect representational geometry with concrete design choices (width, pooling, and final-layer activations) and suggest practical approaches for more interpretable GNNs.
title Superposition in Graph Neural Networks
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.00928