Topological Neural Networks go Persistent, Equivariant, and Continuous

Yogesh Verma,Amauri H Souza,Vikas Garg
2024-06-05
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.
Machine Learning
What problem does this paper attempt to address?
The paper proposes a framework called TopNets that combines the advantages of Topological Neural Networks (TNNs) and Persistent Homology (PH) to enhance the processing capability of high-order relational information and improve the expressiveness of Graph Neural Networks (GNNs). TNNs can handle richer representations than GNNs, while PH is a method for capturing important topological information in complex data. The paper points out that although some progress has been made in combining PH features with GNNs, the potential of integrating PH into TNNs has not been fully explored. TopNets not only unifies TNNs and PH methods, but also has the ability to handle complex geometric data, expanding their applications in spatial settings. Theoretically, the paper proves that PH descriptors can enhance the expressive power of simple propagation networks. Empirically, TopNets performs well in diverse tasks such as antibody design, molecular dynamics simulation, and drug attribute prediction, and proposes continuous and E(n)-covariant versions. In addition, the paper explores how to combine TopNets with Neural Ordinary Differential Equations (Neural ODEs) to achieve continuous systems and analyzes the error bounds between discrete and continuous systems. Through these contributions, TopNets demonstrates wide adaptability and high performance in various real-world tasks.