Reinforcement Learning for Molecular Design Guided by Quantum Mechanics
Not on the map. Why?
Abstract (arXiv)
Automating molecular design using deep reinforcement learning (RL) holds the promise of accelerating the discovery of new chemical compounds. Existing approaches work with molecular graphs and thus ignore the location of atoms in space, which restricts them to 1) generating single organic molecules and 2) heuristic reward functions. To address this, we present a novel RL formulation for molecular design in Cartesian coordinates, thereby extending the class of molecules that can be built. Our reward function is directly based on fundamental physical properties such as the energy, which we approximate via fast quantum-chemical methods. To enable progress towards de-novo molecular design, we introduce MolGym, an RL environment comprising several challenging molecular design tasks along with baselines. In our experiments, we show that our agent can efficiently learn to solve these tasks from scratch by working in a translation and rotation invariant state-action space.
Key facts. Cited by 51 papers in the map (data through 7 Oct 2026). First posted on arXiv 18 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 13; 2023: 13; 2024: 3; 2025: 8; 2026: 3.
51 papers.
Sort:
By area:
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Predictioncited by 176
- 2021Spherical Message Passing for 3D Graph Networkscited by 166
- 2022A 3D Generative Model for Structure-Based Drug Designcited by 148
- 2022Generating 3D Molecules for Target Protein Bindingcited by 104
- 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
- 2021Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial Spacescited by 45
- 20223DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker Designcited by 33
- 2021Artificial Intelligence in Drug Discovery: Applications and Techniquescited by 28
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generationcited by 27
38 more
- 2022Graph Neural Networks for Moleculescited by 21
- 2021Brick-by-Brick: Combinatorial Construction with Deep Reinforcement Learningcited by 19
- 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
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationcited by 17
- 2023MatSciML: A Broad, Multi-Task Benchmark for Solid-State Materials Modelingcited by 14
- 2025Equivariant Neural Diffusion for Molecule Generationcited by 13
- 2023Reinforcement Learning for Generative AI: A Surveycited by 11
- 2023DiffSDS: A language diffusion model for protein backbone inpainting under geometric conditions and constraintscited by 10
- 2023An Equivariant Generative Framework for Molecular Graph-Structure Co-Designcited by 10
- 2021Generating stable molecules using imitation and reinforcement learningcited by 9
- 2024Molecular relaxation by reverse diffusion with time step predictioncited by 9
- 2023ChemSpaceAL: An Efficient Active Learning Methodology Applied to Protein-Specific Molecular Generationcited by 8
- 2021Structure-aware generation of drug-like moleculescited by 7
- 2022Near-optimal Policy Identification in Active Reinforcement Learningcited by 7
- 2023PrefixMol: Target- and Chemistry-aware Molecule Design via Prefix Embeddingcited by 7
- 2021Augmenting Molecular Deep Generative Models with Topological Data Analysis Representationscited by 6
- 2025BARNN: A Bayesian Autoregressive and Recurrent Neural Networkcited by 5
- 2023StriderNET: A Graph Reinforcement Learning Approach to Optimize Atomic Structures on Rough Energy Landscapescited by 4
- 2023Automated patent extraction powers generative modeling in focused chemical spacescited by 4
- 20253DMolFormer: A Dual-channel Framework for Structure-based Drug Discoverycited by 4
- 2025De Novo Molecular Design Enabled by Direct Preference Optimization and Curriculum Learningcited by 4
- 20233D Molecular Generation via Virtual Dynamicscited by 3
- 2024Overcoming Order in Autoregressive Graph Generationcited by 3
- 2021Artificial Intelligence based Autonomous Molecular Design for Medical Therapeutic: A Perspectivecited 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
- 2022Automating Rigid Origami Designcited by 1
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinatescited by 1
- 2025Atomic Proximal Policy Optimization for Electric Robo-Taxi Dispatch and Charger Allocationcited by 1
- 2025Reinforcement Learning for Chemical Ordering in Alloy Nanoparticlescited by 1
- 2022Bayesian Sequential Stacking Algorithm for Concurrently Designing Molecules and Synthetic Reaction Networkscited by 0
- 2022Processing Network Controls via Deep Reinforcement Learningcited by 0
- 2022Cluster Generation via Deep Energy-Based Modelcited by 0
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learningcited by 0
- 2026PGFS++: Molecular Property Improvement under Synthesis and Diversity Constraintscited by 0
50 most cited of 51 All 51 (JSON)
- 2026Let CSP Be Your ANCHOR: Adaptive Crystal Search over Frozen Structure Priors27 Sep 2026
- 2026PGFS++: Molecular Property Improvement under Synthesis and Diversity Constraints19 Aug 2026
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning27 May 2026
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025Reinforcement Learning for Chemical Ordering in Alloy Nanoparticles15 Nov 2025
- 2025Equivariant Neural Diffusion for Molecule Generation12 Jun 2025
- 2025De Novo Molecular Design Enabled by Direct Preference Optimization and Curriculum Learning2 Apr 2025
- 2025Atomic Proximal Policy Optimization for Electric Robo-Taxi Dispatch and Charger Allocation19 Feb 2025
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generation18 Feb 2025
- 20253DMolFormer: A Dual-channel Framework for Structure-based Drug Discovery7 Feb 2025
- 2025BARNN: A Bayesian Autoregressive and Recurrent Neural Network30 Jan 2025
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinates2 Dec 2024
38 more
- 2024Molecular relaxation by reverse diffusion with time step prediction16 Apr 2024
- 2024Overcoming Order in Autoregressive Graph Generation4 Feb 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
- 2023MatSciML: A Broad, Multi-Task Benchmark for Solid-State Materials Modeling12 Sep 2023
- 2023ChemSpaceAL: An Efficient Active Learning Methodology Applied to Protein-Specific Molecular Generation11 Sep 2023
- 2023Reinforcement Learning for Generative AI: A Survey28 Aug 2023
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generation21 May 2023
- 2023An Equivariant Generative Framework for Molecular Graph-Structure Co-Design12 Apr 2023
- 2023Automated patent extraction powers generative modeling in focused chemical spaces14 Mar 2023
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Prediction6 Mar 2023
- 2023PrefixMol: Target- and Chemistry-aware Molecule Design via Prefix Embedding14 Feb 2023
- 20233D Molecular Generation via Virtual Dynamics12 Feb 2023
- 2023StriderNET: A Graph Reinforcement Learning Approach to Optimize Atomic Structures on Rough Energy Landscapes29 Jan 2023
- 2023DiffSDS: A language diffusion model for protein backbone inpainting under geometric conditions and constraints22 Jan 2023
- 2022Near-optimal Policy Identification in Active Reinforcement Learning19 Dec 2022
- 2022Automating Rigid Origami Design20 Nov 2022
- 2022Equivariant Shape-Conditioned Generation of 3D Molecules for Ligand-Based Drug Design6 Oct 2022
- 2022Graph Neural Networks for Molecules12 Sep 2022
- 2022Cluster Generation via Deep Energy-Based Model17 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
- 2022Processing Network Controls via Deep Reinforcement Learning1 May 2022
- 2022Generating 3D Molecules for Target Protein Binding19 Apr 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
- 2022Bayesian Sequential Stacking Algorithm for Concurrently Designing Molecules and Synthetic Reaction Networks1 Mar 2022
- 2022Reinforcement Learning for Personalized Drug Discovery and Design for Complex Diseases: A Systems Pharmacology Perspective21 Jan 2022
- 2021Structure-aware generation of drug-like molecules7 Nov 2021
- 2021Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial Spaces1 Nov 2021
- 2021Brick-by-Brick: Combinatorial Construction with Deep Reinforcement Learning29 Oct 2021
- 2021An In-depth Summary of Recent Artificial Intelligence Applications in Drug Design10 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
- 2021Artificial Intelligence in Drug Discovery: Applications and Techniques9 Jun 2021
- 2021Augmenting Molecular Deep Generative Models with Topological Data Analysis Representations8 Jun 2021
- 2021An End-to-End Framework for Molecular Conformation Generation via Bilevel Programming15 May 2021
- 2021Artificial Intelligence based Autonomous Molecular Design for Medical Therapeutic: A Perspective10 Feb 2021
50 newest of 51 All 51 (JSON)
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generationcited by 27
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationcited by 17
- 2025Equivariant Neural Diffusion for Molecule Generationcited by 13
- 2023DiffSDS: A language diffusion model for protein backbone inpainting under geometric conditions and constraintscited by 10
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generationcited by 2
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025Equivariant Neural Diffusion for Molecule Generation12 Jun 2025
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generation18 Feb 2025
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generation21 May 2023
- 2023DiffSDS: A language diffusion model for protein backbone inpainting under geometric conditions and constraints22 Jan 2023
- 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Predictioncited by 176
- 2021Spherical Message Passing for 3D Graph Networkscited by 166
- 2022A 3D Generative Model for Structure-Based Drug Designcited by 148
- 2022Generating 3D Molecules for Target Protein Bindingcited by 104
- 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
- 2021Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial Spacescited by 45
- 20223DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker Designcited by 33
- 2021Artificial Intelligence in Drug Discovery: Applications and Techniquescited by 28
- 2022Graph Neural Networks for Moleculescited by 21
12 most cited of 44 All 44 (JSON)
- 2026Let CSP Be Your ANCHOR: Adaptive Crystal Search over Frozen Structure Priors27 Sep 2026
- 2026PGFS++: Molecular Property Improvement under Synthesis and Diversity Constraints19 Aug 2026
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning27 May 2026
- 2025Reinforcement Learning for Chemical Ordering in Alloy Nanoparticles15 Nov 2025
- 2025Atomic Proximal Policy Optimization for Electric Robo-Taxi Dispatch and Charger Allocation19 Feb 2025
- 20253DMolFormer: A Dual-channel Framework for Structure-based Drug Discovery7 Feb 2025
- 2025BARNN: A Bayesian Autoregressive and Recurrent Neural Network30 Jan 2025
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinates2 Dec 2024
- 2024Molecular relaxation by reverse diffusion with time step prediction16 Apr 2024
- 2024Overcoming Order in Autoregressive Graph Generation4 Feb 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
12 newest of 44 All 44 (JSON)