Symmetry-Aware Actor-Critic for 3D Molecular Design
Not on the map. Why?
Abstract (arXiv)
Automating molecular design using deep reinforcement learning (RL) has the potential to greatly accelerate the search for novel materials. Despite recent progress on leveraging graph representations to design molecules, such methods are fundamentally limited by the lack of three-dimensional (3D) information. In light of this, we propose a novel actor-critic architecture for 3D molecular design that can generate molecular structures unattainable with previous approaches. This is achieved by exploiting the symmetries of the design process through a rotationally covariant state-action representation based on a spherical harmonics series expansion. We demonstrate the benefits of our approach on several 3D molecular design tasks, where we find that building in such symmetries significantly improves generalization and the quality of generated molecules.
Key facts. Cited by 55 papers in the map (data through 7 Oct 2026). First posted on arXiv 25 Nov 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 18; 2023: 10; 2024: 5; 2025: 8; 2026: 2.
55 papers.
Sort:
By area:
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 666
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 511
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 438
- 2022Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pocketscited by 182
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Predictioncited by 176
- 2021Learning Gradient Fields for Molecular Conformation Generationcited by 162
- 2022A 3D Generative Model for Structure-Based Drug Designcited by 148
- 2021Inverse design of 3d molecular structures with conditional generative neural networkscited by 93
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2022MolGenSurvey: A Systematic Survey in Machine Learning Models for Molecule Designcited by 66
- 2021An End-to-End Framework for Molecular Conformation Generation via Bilevel Programmingcited by 59
- 2021Particle Cloud Generation with Message Passing Generative Adversarial Networkscited by 45
38 more
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 43
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 40
- 20233D molecule generation by denoising voxel gridscited by 38
- 2021Learning to design drug-like molecules in three-dimensional space using deep generative modelscited by 37
- 20223DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker Designcited by 33
- 2022SchNetPack 2.0: A neural network toolbox for atomistic machine learningcited by 29
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generationcited by 27
- 2021Multi-Agent MDP Homomorphic Networkscited by 24
- 2022Maximum Class Separation as Inductive Bias in One Matrixcited by 21
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structurescited by 20
- 2022Equivariant Shape-Conditioned Generation of 3D Molecules for Ligand-Based Drug Designcited by 19
- 2023De novo Drug Design using Reinforcement Learning with Multiple GPT Agentscited by 19
- 2023Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for 3D Molecule Generationcited by 17
- 2022Equivariant Networks for Zero-Shot Coordinationcited by 16
- 2023Generation of 3D Molecules in Pockets via Language Modelcited by 13
- 2022Scalable Fragment-Based 3D Molecular Design with Reinforcement Learningcited by 11
- 2023Deep Learning Methods for Small Molecule Drug Discovery: A Surveycited by 10
- 2025Geometry-aware RL for Manipulation of Varying Shapes and Deformable Objectscited by 10
- 2021Generating stable molecules using imitation and reinforcement learningcited by 9
- 2024Molecular relaxation by reverse diffusion with time step predictioncited by 9
- 2022Near-optimal Policy Identification in Active Reinforcement Learningcited by 7
- 2021Augmenting Molecular Deep Generative Models with Topological Data Analysis Representationscited by 6
- 2021Surrogate- and invariance-boosted contrastive learning for data-scarce applications in sciencecited by 6
- 2024Overcoming Order in Autoregressive Graph Generationcited by 3
- 2024Rectified Flow For Structure Based Drug Designcited by 3
- 2025Clifford Group Equivariant Diffusion Models for 3D Molecular Generationcited by 3
- 2022Atomistic structure search using local surrogate modecited by 2
- 2023Molecular Conformation Generation via Shifting Scorescited by 2
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generationcited by 2
- 2021An In-depth Summary of Recent Artificial Intelligence Applications in Drug Designcited by 1
- 2022Reinforcement Learning for Personalized Drug Discovery and Design for Complex Diseases: A Systems Pharmacology Perspectivecited by 1
- 20223D pride without 2D prejudice: Bias-controlled multi-level generative models for structure-based ligand designcited by 1
- 2022Automating Rigid Origami Designcited by 1
- 2024FREED++: Improving RL Agents for Fragment-Based Molecule Generation by Thorough Reproductioncited by 1
- 2025Symmetries-enhanced Multi-Agent Reinforcement Learningcited by 1
- 2025Learning Abstract World Models with a Group-Structured Latent Spacecited by 1
- 2025Gradient-based grand canonical optimization enabled by graph neural networks with fractional atomic existencecited by 1
- 2022Cluster Generation via Deep Energy-Based Modelcited by 0
50 most cited of 55 All 55 (JSON)
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning27 May 2026
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutions10 Feb 2026
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025FRAGMENTA: End-to-end Fragmentation-based Generative Model with Agentic Tuning for Drug Lead Optimization25 Nov 2025
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025Gradient-based grand canonical optimization enabled by graph neural networks with fractional atomic existence25 Jul 2025
- 2025Learning Abstract World Models with a Group-Structured Latent Space2 Jun 2025
- 2025Clifford Group Equivariant Diffusion Models for 3D Molecular Generation22 Apr 2025
- 2025Geometry-aware RL for Manipulation of Varying Shapes and Deformable Objects10 Feb 2025
- 2025Symmetries-enhanced Multi-Agent Reinforcement Learning2 Jan 2025
- 2024Rectified Flow For Structure Based Drug Design2 Dec 2024
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structures12 Jul 2024
38 more
- 2024Molecular relaxation by reverse diffusion with time step prediction16 Apr 2024
- 2024Overcoming Order in Autoregressive Graph Generation4 Feb 2024
- 2024FREED++: Improving RL Agents for Fragment-Based Molecule Generation by Thorough Reproduction18 Jan 2024
- 2023De novo Drug Design using Reinforcement Learning with Multiple GPT Agents21 Dec 2023
- 2023Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for 3D Molecule Generation27 Nov 2023
- 2023On Accelerating Diffusion-based Molecular Conformation Generation in SE(3)-invariant Space7 Oct 2023
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space4 Oct 2023
- 2023Molecular Conformation Generation via Shifting Scores12 Sep 2023
- 20233D molecule generation by denoising voxel grids13 Jun 2023
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generation21 May 2023
- 2023Generation of 3D Molecules in Pockets via Language Model17 May 2023
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Prediction6 Mar 2023
- 2023Deep Learning Methods for Small Molecule Drug Discovery: A Survey1 Mar 2023
- 2022Near-optimal Policy Identification in Active Reinforcement Learning19 Dec 2022
- 2022SchNetPack 2.0: A neural network toolbox for atomistic machine learning11 Dec 2022
- 2022Automating Rigid Origami Design20 Nov 2022
- 2022Equivariant Networks for Zero-Shot Coordination21 Oct 2022
- 2022Equivariant Shape-Conditioned Generation of 3D Molecules for Ligand-Based Drug Design6 Oct 2022
- 2022Atomistic structure search using local surrogate mode19 Aug 2022
- 2022Cluster Generation via Deep Energy-Based Model17 Jun 2022
- 2022Maximum Class Separation as Inductive Bias in One Matrix17 Jun 2022
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphs17 Jun 2022
- 20223DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker Design15 May 2022
- 2022Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pockets15 May 2022
- 20223D pride without 2D prejudice: Bias-controlled multi-level generative models for structure-based ligand design22 Apr 2022
- 2022Equivariant Diffusion for Molecule Generation in 3D31 Mar 2022
- 2022MolGenSurvey: A Systematic Survey in Machine Learning Models for Molecule Design28 Mar 2022
- 2022A 3D Generative Model for Structure-Based Drug Design20 Mar 2022
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation6 Mar 2022
- 2022Scalable Fragment-Based 3D Molecular Design with Reinforcement Learning1 Feb 2022
- 2022Reinforcement Learning for Personalized Drug Discovery and Design for Complex Diseases: A Systems Pharmacology Perspective21 Jan 2022
- 2021Surrogate- and invariance-boosted contrastive learning for data-scarce applications in science15 Oct 2021
- 2021An In-depth Summary of Recent Artificial Intelligence Applications in Drug Design10 Oct 2021
- 2021Multi-Agent MDP Homomorphic Networks9 Oct 2021
- 2021Inverse design of 3d molecular structures with conditional generative neural networks10 Sep 2021
- 2021Generating stable molecules using imitation and reinforcement learning11 Jul 2021
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Survey2 Jul 2021
- 2021Particle Cloud Generation with Message Passing Generative Adversarial Networks22 Jun 2021
50 newest of 55 All 55 (JSON)
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 666
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 511
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generationcited by 27
- 2024Rectified Flow For Structure Based Drug Designcited by 3
- 2025Clifford Group Equivariant Diffusion Models for 3D Molecular Generationcited by 3
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generationcited by 2
- 2023On Accelerating Diffusion-based Molecular Conformation Generation in SE(3)-invariant Spacecited by 0
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutionscited by 0
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutions10 Feb 2026
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025Clifford Group Equivariant Diffusion Models for 3D Molecular Generation22 Apr 2025
- 2024Rectified Flow For Structure Based Drug Design2 Dec 2024
- 2023On Accelerating Diffusion-based Molecular Conformation Generation in SE(3)-invariant Space7 Oct 2023
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generation21 May 2023
- 2022Equivariant Diffusion for Molecule Generation in 3D31 Mar 2022
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation6 Mar 2022
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 438
- 2022Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pocketscited by 182
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Predictioncited by 176
- 2021Learning Gradient Fields for Molecular Conformation Generationcited by 162
- 2022A 3D Generative Model for Structure-Based Drug Designcited by 148
- 2021Inverse design of 3d molecular structures with conditional generative neural networkscited by 93
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2022MolGenSurvey: A Systematic Survey in Machine Learning Models for Molecule Designcited by 66
- 2021An End-to-End Framework for Molecular Conformation Generation via Bilevel Programmingcited by 59
- 2021Particle Cloud Generation with Message Passing Generative Adversarial Networkscited by 45
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 43
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 40
12 most cited of 44 All 44 (JSON)
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning27 May 2026
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025Gradient-based grand canonical optimization enabled by graph neural networks with fractional atomic existence25 Jul 2025
- 2025Geometry-aware RL for Manipulation of Varying Shapes and Deformable Objects10 Feb 2025
- 2025Symmetries-enhanced Multi-Agent Reinforcement Learning2 Jan 2025
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structures12 Jul 2024
- 2024Molecular relaxation by reverse diffusion with time step prediction16 Apr 2024
- 2024Overcoming Order in Autoregressive Graph Generation4 Feb 2024
- 2024FREED++: Improving RL Agents for Fragment-Based Molecule Generation by Thorough Reproduction18 Jan 2024
- 2023De novo Drug Design using Reinforcement Learning with Multiple GPT Agents21 Dec 2023
- 2023Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for 3D Molecule Generation27 Nov 2023
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space4 Oct 2023
12 newest of 44 All 44 (JSON)