Demystifying Multimodal Biomolecular Co-design With Intrinsic Geodesic Coupling
Keyue Qiu, Xintong Wang, Zhilong Zhang, Hao Zhou, Wei-Ying Ma
Abstract
Biomolecules such as proteins and small-molecule ligands play a central role in biological systems, arising from the tight interplay between sequence and three-dimensional structure. Recent generative models for biomolecular co-design aim to capture this interplay by jointly modeling coupled modalities. However, existing approaches largely adopt a parallel execution of marginal generative processes, implicitly enforcing fixed synchronous coupling. We argue that a critical but overlooked degree of freedom lies in how these marginal processes are temporally coupled during training and generation, where inappropriate coupling can introduce high-variance supervision and inconsistent intermediate states, affecting modality consistency. To address this, we introduce GeoCoupling, a systematic framework that optimizes for temporal couplings between heterogeneous modalities. Empirical results across structure-based drug design and unconditional protein design demonstrate the learned couplings consistently outperform synchronous and randomly coupled baselines, yielding biomolecules with improved physical validity and diversity.
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 26d9616f-cae3-4ab3-b754-e4040f795440Builds on17
- Learning inverse folding from millions of predicted structuresChloe Hsu, Robert Verkuil, Jason Liu, Zeming Lin et al.ICML 2022 · 560 citations
- Understanding Diffusion Objectives as the ELBO with Simple Data AugmentationDiederik P. Kingma, Ruiqi GaoNeurIPS 2023 · 358 citations
- SE(3) diffusion model with application to protein backbone generationJason Yim, Brian L. Trippe, Valentin De Bortoli, Emile Mathieu et al.ICML 2023 · 313 citations
- Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein PocketsXingang Peng, Shitong Luo, Jiaqi Guan, Qi Xie et al.ICML 2022 · 291 citations
- Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-DesignAndrew Campbell, Jason Yim, Regina Barzilay, Tom Rainforth et al.ICML 2024 · 283 citations
Related papers
- Unified Guidance for Geometry-Conditioned Molecular GenerationSirine Ayadi, Leon Hetzel, Johanna Sommer, Fabian J. Theis et al.NeurIPS 2024 · 11 citations
- Geometric Representation Condition Improves Equivariant Molecule GenerationZian Li, Cai Zhou, Xiyuan Wang, Xingang Peng et al.ICML 2025
- SynCoGen: Synthesizable 3D Molecule Generation via Joint Reaction and Coordinate ModelingAndrei Rekesh, Miruna Cretu, Dmytro Shevchuk, Pietro Lio et al.ICLR 2026 · 6 citations
- 3DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker DesignYinan Huang, Xingang Peng, Jianzhu Ma, Muhan ZhangICML 2022 · 65 citations
- Unified Generative Modeling of 3D Molecules with Bayesian Flow NetworksYuxuan Song, Jingjing Gong, Hao Zhou, Mingyue Zheng et al.ICLR 2024 · 36 citations
