Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Topological Neural Networks go Persistent, Equivariant, and Continuous

About

Topological Neural Networks (TNNs) incorporate higher-order relational information beyond pairwise interactions, enabling richer representations than Graph Neural Networks (GNNs). Concurrently, topological descriptors based on persistent homology (PH) are being increasingly employed to augment the GNNs. We investigate the benefits of integrating these two paradigms. Specifically, we introduce TopNets as a broad framework that subsumes and unifies various methods in the intersection of GNNs/TNNs and PH such as (generalizations of) RePHINE and TOGL. TopNets can also be readily adapted to handle (symmetries in) geometric complexes, extending the scope of TNNs and PH to spatial settings. Theoretically, we show that PH descriptors can provably enhance the expressivity of simplicial message-passing networks. Empirically, (continuous and E(n)-equivariant extensions of) TopNets achieve strong performance across diverse tasks, including antibody design, molecular dynamics simulation, and drug property prediction.

Yogesh Verma, Amauri H Souza, Vikas Garg• 2024

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy73.8
1252
Graph ClassificationMUTAG
Accuracy92.7
1103
Graph ClassificationNCI1
Accuracy79.1
658
Graph ClassificationNCI109
Accuracy78.4
267
Graph ClassificationIMDB-B
Mean Accuracy73.1
159
Graph ClassificationPTC
Accuracy65.7
46
Showing 6 of 6 rows

Other info

Follow for update