E(n) Equivariant Message Passing Cellular Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kovač, Veljko, Bekkers, Erik J., Liò, Pietro, Eijkelboom, Floor
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910532976508928
author Kovač, Veljko
Bekkers, Erik J.
Liò, Pietro
Eijkelboom, Floor
author_facet Kovač, Veljko
Bekkers, Erik J.
Liò, Pietro
Eijkelboom, Floor
contents This paper introduces E(n) Equivariant Message Passing Cellular Networks (EMPCNs), an extension of E(n) Equivariant Graph Neural Networks to CW-complexes. Our approach addresses two aspects of geometric message passing networks: 1) enhancing their expressiveness by incorporating arbitrary cells, and 2) achieving this in a computationally efficient way with a decoupled EMPCNs technique. We demonstrate that EMPCNs achieve close to state-of-the-art performance on multiple tasks without the need for steerability, including many-body predictions and motion capture. Moreover, ablation studies confirm that decoupled EMPCNs exhibit stronger generalization capabilities than their non-topologically informed counterparts. These findings show that EMPCNs can be used as a scalable and expressive framework for higher-order message passing in geometric and topological graphs
format Preprint
id arxiv_https___arxiv_org_abs_2406_03145
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle E(n) Equivariant Message Passing Cellular Networks
Kovač, Veljko
Bekkers, Erik J.
Liò, Pietro
Eijkelboom, Floor
Machine Learning
This paper introduces E(n) Equivariant Message Passing Cellular Networks (EMPCNs), an extension of E(n) Equivariant Graph Neural Networks to CW-complexes. Our approach addresses two aspects of geometric message passing networks: 1) enhancing their expressiveness by incorporating arbitrary cells, and 2) achieving this in a computationally efficient way with a decoupled EMPCNs technique. We demonstrate that EMPCNs achieve close to state-of-the-art performance on multiple tasks without the need for steerability, including many-body predictions and motion capture. Moreover, ablation studies confirm that decoupled EMPCNs exhibit stronger generalization capabilities than their non-topologically informed counterparts. These findings show that EMPCNs can be used as a scalable and expressive framework for higher-order message passing in geometric and topological graphs
title E(n) Equivariant Message Passing Cellular Networks
topic Machine Learning
url https://arxiv.org/abs/2406.03145