Mol-AIR: Molecular Reinforcement Learning with Adaptive Intrinsic Rewards for Goal-directed Molecular Generation

2024-03-29 · Jinyeong Park, Jaegyoon Ahn, Jonghwan Choi, Jibum Kim

Optimizing techniques for discovering molecular structures with desired properties is crucial in artificial intelligence(AI)-based drug discovery. Combining deep generative models with reinforcement learning has emerged as an effective strategy for generating molecules with specific properties. Despite its potential, this approach is ineffective in exploring the vast chemical space and optimizing particular chemical properties. To overcome these limitations, we present Mol-AIR, a reinforcement learning-based framework using adaptive intrinsic rewards for effective goal-directed molecular generation. Mol-AIR leverages the strengths of both history-based and learning-based intrinsic rewards by exploiting random distillation network and counting-based strategies. In benchmark tests, Mol-AIR demonstrates superior performance over existing approaches in generating molecules with desired properties without any prior knowledge, including penalized LogP, QED, and celecoxib similarity. We believe that Mol-AIR represents a significant advancement in drug discovery, offering a more efficient path to discovering novel therapeutics.

📄 PDF Abstract BibTeX arXiv:2403.20109

Code (1)

DevSlem/Mol-AIR 공식 구현 pytorch

Tasks

Drug Discovery reinforcement-learning Reinforcement Learning

Automatic Reward Design via Learning Motivation-Consistent Intrinsic Rewards

2022-07-29 · Yixiang Wang, Yujing Hu, Feng Wu, Yingfeng Chen

Reward design is a critical part of the application of reinforcement learning, the performance of which strongly depends on how well the reward signal frames the goal of the designer and how well the signal assesses prog…

Adaptive Correlation-Weighted Intrinsic Rewards for Reinforcement Learning

2026-02-27 · Viet Bac Nguyen, Phuong Thai Nguyen arxiv

We propose ACWI (Adaptive Correlation Weighted Intrinsic), an adaptive intrinsic reward scaling framework designed to dynamically balance intrinsic and extrinsic rewards for improved exploration in sparse reward reinforc…

Computational Efficiency Reinforcement Learning

Feature Control as Intrinsic Motivation for Hierarchical Reinforcement Learning

2017-05-18 · Nat Dilokthanakul, Christos Kaplanis, Nick Pawlowski, Murray Shanahan

The problem of sparse rewards is one of the hardest challenges in contemporary reinforcement learning. Hierarchical reinforcement learning (HRL) tackles this problem by using a set of temporally-extended actions, or opti…

Hierarchical Reinforcement Learning Montezuma's Revenge reinforcement-learning Reinforcement Learning+1

Intrinsic Motivation in Reinforcement Learning: A Research Agenda for Adaptive Self-Organisation

2026-09-15 · Anatoly Belikov arxiv

Biological cells can be viewed as individual, interacting agents whose collective dynamics give rise to adaptive behaviour at multiple levels of organisation, from individual cells through tissues to whole multicellular …

Reinforcement Learning

LLM-Driven Intrinsic Motivation for Sparse Reward Reinforcement Learning

2025-08-25 · André Quadros, Cassio Silva, Ronnie Alves arxiv

This paper explores the combination of two intrinsic motivation strategies to improve the efficiency of reinforcement learning (RL) agents in environments with extreme sparse rewards, where traditional learning struggles…