Multimodal Molecular Pretraining via Modality Blending
Qiying Yu, Yudi Zhang, Yuyan Ni, Shikun Feng, Yanyan Lan, Hao Zhou, Jingjing Liu
Abstract
Self-supervised learning has recently gained growing interest in molecular modeling for scientific tasks such as AI-assisted drug discovery. Current studies consider leveraging both 2D and 3D molecular structures for representation learning. However, relying on straightforward alignment strategies that treat each modality separately, these methods fail to exploit the intrinsic correlation between 2D and 3D representations that reflect the underlying structural characteristics of molecules, and only perform coarse-grained molecule-level alignment. To derive fine-grained alignment and promote structural molecule understanding, we introduce an atomic-relation level "blend-then-predict" self-supervised learning approach, MOLEBLEND, which first blends atom relations represented by different modalities into one unified relation matrix for joint encoding, then recovers modality-specific information for 2D and 3D structures individually. By treating atom relationships as anchors, MOLEBLEND organically aligns and integrates visually dissimilar 2D and 3D modalities of the same molecule at fine-grained atomic level, painting a more comprehensive depiction of each molecule. Extensive experiments show that MOLEBLEND achieves state-of-the-art performance across major 2D/3D molecular benchmarks. We further provide theoretical insights from the perspective of mutual-information maximization, demonstrating that our method unifies contrastive, generative (cross-modality prediction) and mask-then-predict (single-modality prediction) objectives into one single cohesive framework.
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.
Cited by top-tier papers14
- Mol-AE: Auto-Encoder Based Molecular Representation Learning With 3D Cloze Test ObjectiveJunwei Yang, Kangjie Zheng, Siyu Long, Zaiqing Nie et al.ICML 2024 · 19 citations
- MOL-Mamba: Enhancing Molecular Representation with Structural & Electronic InsightsJingjing Hu, Dan Guo, Zhan Si, Deguang Liu et al.AAAI 2025 · 9 citations
- MolParser: End-to-End Visual Recognition of Molecule Structures in the WildXi Fang, Jiankun Wang, Xiaochen Cai, Shangqian Chen et al.ICCV 2025 · 8 citations
- Learning 3D Anisotropic Noise Distributions Improves Molecular Force FieldsXixian Liu, Rui Jiao, Zhiyuan Liu, Yurou Liu et al.NeurIPS 2025 · 3 citations
- Self-supervised Blending Structural Context of Visual Molecules for Robust Drug Interaction PredictionTengfei Ma, Kun Chen, Yongsheng Zang, Yujie Chen et al.NeurIPS 2025 · 2 citations
Builds on23
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Graph Contrastive Learning with AugmentationsYuning You, Tianlong Chen, Yongduo Sui, Ting Chen et al.NeurIPS 2020 · 3,042 citations
- Strategies for Pre-training Graph Neural NetworksWeihua Hu, Bowen Liu, Joseph Gomes, Marinka Zitnik et al.ICLR 2020 · 1,744 citations
- Do Transformers Really Perform Badly for Graph Representation?Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng et al.NeurIPS 2021 · 1,632 citations
- Self-Supervised Graph Transformer on Large-Scale Molecular DataYu Rong, Yatao Bian, Tingyang Xu, Weiyang Xie et al.NeurIPS 2020 · 1,113 citations
Related papers
- Molecule Joint Auto-Encoding: Trajectory Pretraining with 2D and 3D DiffusionWeitao Du, Jiujiu Chen, Xuecang Zhang, Zhi-Ming Ma et al.NeurIPS 2023 · 15 citations
- One Transformer Can Understand Both 2D & 3D Molecular DataShengjie Luo, Tianlang Chen, Yixian Xu, Shuxin Zheng et al.ICLR 2023 · 4 citations
- Atomas: Hierarchical Adaptive Alignment on Molecule-Text for Unified Molecule Understanding and GenerationYikun Zhang, Geyan Ye, Chaohao Yuan, Bo Han et al.ICLR 2025
- Protein-ligand binding representation learning from fine-grained interactionsShikun Feng, Minghao Li, Yinjun Jia, Wei-Ying Ma et al.ICLR 2024 · 21 citations
- 3D Denoisers Are Good 2D Teachers: Molecular Pretraining via Denoising and Cross-Modal DistillationSungjun Cho, Dae-Woong Jeong, Sung Moon Ko, Jinwoo Kim et al.AAAI 2025 · 1 citation
