Symmetry-Preserving Conformer Ensemble Networks for Molecular Representation Learning
Yanqiao Zhu, Yidan Shi, Yuanzhou Chen, Fang Sun, Yizhou Sun, Wei Wang
Abstract
Molecular representation learning has emerged as a promising approach for modeling molecules with deep learning in chemistry and beyond. While 3D geometric models effectively capture molecular structure, they typically process single static conformers, overlooking the inherent flexibility and dynamics of molecules. In reality, many molecular properties depend on distributions of thermodynamically accessible conformations rather than single structures. Recent works show that learning from conformer ensembles can improve molecular representations, but existing approaches either produce unphysical structures through averaging or require restrictive molecular alignment. In this paper, we propose Symmetry-Preserving Conformer Ensemble networks ( SPiCE ), which introduces two key innovations: (1) geometric mixture-of-experts for selective processing of scalar and vector features, and (2) hierarchical ensemble encoding that combines ensemble-level representation with cross-conformer integration. Crucially, SPiCE ensures physically meaningful representations by maintaining joint equivariance to geometric transformations of individual conformers and conformer permutations. Extensive experiments demonstrate that SPiCE consistently outperforms existing conformer ensemble methods and state-of-the-art structural aggregation models across quantum mechanical and biological property prediction tasks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 58eaf789-6ed7-4b00-81ab-fd811ed2caabBuilds on20
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- GShard: Scaling Giant Models with Conditional Computation and Automatic ShardingDmitry Lepikhin, HyoukJoong Lee, Yuanzhong Xu, Dehao Chen et al.ICLR 2021 · 1,954 citations
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 1,432 citations
- Recipe for a General, Powerful, Scalable Graph TransformerLadislav Rampásek, Michael Galkin, Vijay Prakash Dwivedi, Anh Tuan Luu et al.NeurIPS 2022 · 1,216 citations
- Directional Message Passing for Molecular GraphsJohannes Klicpera, Janek Groß, Stephan GünnemannICLR 2020 · 1,079 citations
Related papers
- Structure-Aware E(3)-Invariant Molecular Conformer Aggregation NetworksDuy Minh Ho Nguyen, Nina Lukashina, Tai Nguyen, An T. Le et al.ICML 2024 · 9 citations
- SaVeNet: A Scalable Vector Network for Enhanced Molecular Representation LearningSarp Aykent, Tian XiaNeurIPS 2023 · 6 citations
- Learning Over Molecular Conformer Ensembles: Datasets and BenchmarksYanqiao Zhu, Jeehyun Hwang, Keir Adams, Zhen Liu et al.ICLR 2024 · 13 citations
- GTMGC: Using Graph Transformer to Predict Molecule's Ground-State ConformationGuikun Xu, Yongquan Jiang, PengChuan Lei, Yan Yang et al.ICLR 2024 · 7 citations
- GeoMFormer: A General Architecture for Geometric Molecular Representation LearningTianlang Chen, Shengjie Luo, Di He, Shuxin Zheng et al.ICML 2024 · 9 citations
