Poster
in
Workshop: Learning Meaningful Representations of Life (LMRL) Workshop @ ICLR 2025
Hierarchical Mixture of Topological Experts for Molecular Property Prediction
Kiwoong Yoo · Jaewoo Kang
Molecular property prediction enables rapid identification of promising drug candidates by forecasting key attributes such as bioactivity and toxicity. The relationship between molecular structure and properties spans multiple scales—from individual atoms to functional groups to the overall molecular framework. Depending on the property task and the target molecule’s scaffold, prediction may require focusing on specific substructures or the entire molecular configuration. This observation suggests that selectively attending to relevant structural features at different scales can improve prediction accuracy. In this light, we propose HierMolMoE, a hierarchical mixture-of-experts framework that learns specialized predictive models at three natural granularities of molecular graphs: atom-level, motif-level, and global-level. Our model integrates expert networks at each level with a high-level gating mechanism, and each expert is tailored to capture the unique topological semantics of molecular groups sharing similar scaffolds. Experiments on benchmark datasets demonstrate that HierMolMoE outperforms existing GNN-based mixture-of-experts approaches for molecular property prediction, highlighting its ability to learn robust structure–property relationships across scales.