Topological Neural Networks go Persistent, Equivariant, and Continuous

Yogesh Verma*, Amauri H. Souza, Vikas Garg

*Corresponding author for this work

Research output: Contribution to journalConference articleScientificpeer-review

28 Downloads (Pure)

Abstract

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.

Original languageEnglish
Pages (from-to)49388-49407
Number of pages20
JournalProceedings of Machine Learning Research
Volume235
Publication statusPublished - 2024
MoE publication typeA4 Conference publication
EventInternational Conference on Machine Learning - Vienna, Austria
Duration: 21 Jul 202427 Jul 2024
Conference number: 41

Fingerprint

Dive into the research topics of 'Topological Neural Networks go Persistent, Equivariant, and Continuous'. Together they form a unique fingerprint.

Cite this