Saved in:
Bibliographic Details
Main Authors: Montagna, Marco, Scardapane, Simone, Telyatnikov, Lev
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2409.12033
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910610259705856
author Montagna, Marco
Scardapane, Simone
Telyatnikov, Lev
author_facet Montagna, Marco
Scardapane, Simone
Telyatnikov, Lev
contents Graph Neural Networks based on the message-passing (MP) mechanism are a dominant approach for handling graph-structured data. However, they are inherently limited to modeling only pairwise interactions, making it difficult to explicitly capture the complexity of systems with $n$-body relations. To address this, topological deep learning has emerged as a promising field for studying and modeling higher-order interactions using various topological domains, such as simplicial and cellular complexes. While these new domains provide powerful representations, they introduce new challenges, such as effectively modeling the interactions among higher-order structures through higher-order MP. Meanwhile, structured state-space sequence models have proven to be effective for sequence modeling and have recently been adapted for graph data by encoding the neighborhood of a node as a sequence, thereby avoiding the MP mechanism. In this work, we propose a novel architecture designed to operate with simplicial complexes, utilizing the Mamba state-space model as its backbone. Our approach generates sequences for the nodes based on the neighboring cells, enabling direct communication between all higher-order structures, regardless of their rank. We extensively validate our model, demonstrating that it achieves competitive performance compared to state-of-the-art models developed for simplicial complexes.
format Preprint
id arxiv_https___arxiv_org_abs_2409_12033
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Topological Deep Learning with State-Space Models: A Mamba Approach for Simplicial Complexes
Montagna, Marco
Scardapane, Simone
Telyatnikov, Lev
Machine Learning
Artificial Intelligence
Graph Neural Networks based on the message-passing (MP) mechanism are a dominant approach for handling graph-structured data. However, they are inherently limited to modeling only pairwise interactions, making it difficult to explicitly capture the complexity of systems with $n$-body relations. To address this, topological deep learning has emerged as a promising field for studying and modeling higher-order interactions using various topological domains, such as simplicial and cellular complexes. While these new domains provide powerful representations, they introduce new challenges, such as effectively modeling the interactions among higher-order structures through higher-order MP. Meanwhile, structured state-space sequence models have proven to be effective for sequence modeling and have recently been adapted for graph data by encoding the neighborhood of a node as a sequence, thereby avoiding the MP mechanism. In this work, we propose a novel architecture designed to operate with simplicial complexes, utilizing the Mamba state-space model as its backbone. Our approach generates sequences for the nodes based on the neighboring cells, enabling direct communication between all higher-order structures, regardless of their rank. We extensively validate our model, demonstrating that it achieves competitive performance compared to state-of-the-art models developed for simplicial complexes.
title Topological Deep Learning with State-Space Models: A Mamba Approach for Simplicial Complexes
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2409.12033