{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/geometry-complete-diffusion-for-3d-molecule","title":"Geometry-Complete Diffusion for 3D Molecule Generation and Optimization","arxiv_id":"2302.04313","date":"2023-02-08","proceeding":null,"authors":["Alex Morehead","Jianlin Cheng"],"abstract":"Denoising diffusion probabilistic models (DDPMs) have pioneered new state-of-the-art results in disciplines such as computer vision and computational biology for diverse tasks ranging from text-guided image generation to structure-guided protein design. Along this latter line of research, methods have recently been proposed for generating 3D molecules using equivariant graph neural networks (GNNs) within a DDPM framework. However, such methods are unable to learn important geometric properties of 3D molecules, as they adopt molecule-agnostic and non-geometric GNNs as their 3D graph denoising networks, which notably hinders their ability to generate valid large 3D molecules. In this work, we address these gaps by introducing the Geometry-Complete Diffusion Model (GCDM) for 3D molecule generation, which outperforms existing 3D molecular diffusion models by significant margins across conditional and unconditional settings for the QM9 dataset and the larger GEOM-Drugs dataset, respectively, and generates more novel and unique unconditional 3D molecules for the QM9 dataset compared to previous methods. Importantly, we demonstrate that the geometry-complete denoising process of GCDM learned for 3D molecule generation enables the model to generate a significant proportion of valid and energetically-stable large molecules at the scale of GEOM-Drugs, whereas previous methods fail to do so with the features they learn. Additionally, we show that extensions of GCDM can not only effectively design 3D molecules for specific protein pockets but also that GCDM's geometric features can be repurposed to consistently optimize the geometry and chemical composition of existing 3D molecules for molecular stability and property specificity, demonstrating new versatility of molecular diffusion models. Our source code and data are freely available at https://github.com/BioinfoMachineLearning/Bio-Diffusion.","url_abs":"https://arxiv.org/abs/2302.04313v6","url_pdf":"https://arxiv.org/pdf/2302.04313v6.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"geometry-complete-diffusion-for-3d-molecule","repo_url":"https://github.com/bioinfomachinelearning/bio-diffusion","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"NOASSERTION"}},{"paper_slug":"geometry-complete-diffusion-for-3d-molecule","repo_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"jax","reach":{"status":"ok","spdx":"MIT"}},{"paper_slug":"geometry-complete-diffusion-for-3d-molecule","repo_url":"https://github.com/bioinfomachinelearning/gcpnet","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"jax","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"3d-molecule-generation","task_name":"3D Molecule Generation"},{"task_slug":"denoising","task_name":"Denoising"},{"task_slug":"graph-generation","task_name":"Graph Generation"},{"task_slug":"image-generation","task_name":"Image Generation"},{"task_slug":"molecular-graph-generation","task_name":"Molecular Graph Generation"},{"task_slug":"protein-design","task_name":"Protein Design"},{"task_slug":"representation-learning","task_name":"Representation Learning"},{"task_slug":"specificity","task_name":"Specificity"},{"task_slug":null,"task_name":"valid"}],"methods":[{"method_slug":"diffusion","method_name":"Diffusion"},{"method_slug":"fail","method_name":"fail"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2302.04313","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2302.04313"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/bioinfomachinelearning/bio-diffusion","reach":{"status":"ok","spdx":"NOASSERTION"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/bioinfomachinelearning/gcpnet","reach":{"status":"ok","spdx":"MIT"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/bioinfomachinelearning/gcdm-sbdd","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran":3,"unverified":7},"by_repo_kind":{"official":{"samples":9,"ran":3,"repositories":1},"listed":{"samples":1,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"c81e137b6d163914","entry":"get_bb_transform","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"geometry_utils.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/geometry_utils.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c81e137b6d163914"}},{"code_sha256_prefix":"be813c0c5a49cea2","entry":"get_n_nodes","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"process_crossdock.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/process_crossdock.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"be813c0c5a49cea2"}},{"code_sha256_prefix":"cf32ddf40be29175","entry":"rotation_matrix","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"geometry_utils.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/geometry_utils.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"cf32ddf40be29175"}},{"code_sha256_prefix":"7f62ceaa80a2427c","entry":"filter_and_flatten","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"process_bindingmoad.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/process_bindingmoad.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"7f62ceaa80a2427c"}},{"code_sha256_prefix":"b07de0bf12fe1cf0","entry":"get_GCP_with_custom_cfg","repo":"bioinfomachinelearning/gcpnet","repo_kind":"listed","path":"src/models/components/gcpnet.py","file_url":"https://github.com/bioinfomachinelearning/gcpnet/blob/HEAD/src/models/components/gcpnet.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"b07de0bf12fe1cf0"}},{"code_sha256_prefix":"975b957da34b7db7","entry":"get_grad_norm","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"utils.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"975b957da34b7db7"}},{"code_sha256_prefix":"667d353a0628f4a1","entry":"ligand_list_to_dict","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"process_finetuning_set.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/process_finetuning_set.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"667d353a0628f4a1"}},{"code_sha256_prefix":"a05fbc2a9d5ee01c","entry":"read_label_file","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"process_bindingmoad.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/process_bindingmoad.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a05fbc2a9d5ee01c"}},{"code_sha256_prefix":"04268029bfb9fddc","entry":"residues_to_atoms","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"utils.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"04268029bfb9fddc"}},{"code_sha256_prefix":"f69f2334c7ce9784","entry":"reverse_tensor","repo":"bioinfomachinelearning/gcdm-sbdd","repo_kind":"official","path":"utils.py","file_url":"https://github.com/bioinfomachinelearning/gcdm-sbdd/blob/HEAD/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f69f2334c7ce9784"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}