MolWorld: Molecule World Models for Actionable Molecular Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qiao, Yang, Pan, Bo, Pang, Hao-Wei, Zhang, Peter Zhiping, Zhang, Liying, Zhao, Liang
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911667302957056
author Qiao, Yang
Pan, Bo
Pang, Hao-Wei
Zhang, Peter Zhiping
Zhang, Liying
Zhao, Liang
author_facet Qiao, Yang
Pan, Bo
Pang, Hao-Wei
Zhang, Peter Zhiping
Zhang, Liying
Zhao, Liang
contents Molecular optimization in drug discovery aims to discover molecules with improved target properties, but practical lead optimization often requires more than high predicted scores. A useful candidate should also be actionable: it should be reachable from known molecules through valid local structural transformations, so that it can be interpreted as a plausible revision within an evolving chemical series. Existing de novo and single-molecule optimization methods do not explicitly model such reachability, especially when both the target molecules and the intermediate molecules connecting them to known compounds are unknown. In this work, we formulate actionable molecular optimization as sequential expansion of a molecule-transfer graph, where nodes are molecules and edges encode valid local transformations. We propose MolWorld, a molecule world model-guided framework that treats the current molecule-transfer graph as an evolving search state. At each iteration, MolWorld selects local anchor contexts, generates candidate molecules conditioned on these contexts, evaluates their properties, and uses a learned world model to update the evolving molecule world by retaining admissible candidates and inserting them into the molecule-transfer graph. The expanded molecule world then guides subsequent optimization. Experiments on property optimization and docking-based tasks show that MolWorld discovers high-property molecules while maintaining substantially stronger structural connectivity, supporting actionable and sequential molecular design.
format Preprint
id arxiv_https___arxiv_org_abs_2605_08954
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle MolWorld: Molecule World Models for Actionable Molecular Optimization
Qiao, Yang
Pan, Bo
Pang, Hao-Wei
Zhang, Peter Zhiping
Zhang, Liying
Zhao, Liang
Machine Learning
Artificial Intelligence
Molecular optimization in drug discovery aims to discover molecules with improved target properties, but practical lead optimization often requires more than high predicted scores. A useful candidate should also be actionable: it should be reachable from known molecules through valid local structural transformations, so that it can be interpreted as a plausible revision within an evolving chemical series. Existing de novo and single-molecule optimization methods do not explicitly model such reachability, especially when both the target molecules and the intermediate molecules connecting them to known compounds are unknown. In this work, we formulate actionable molecular optimization as sequential expansion of a molecule-transfer graph, where nodes are molecules and edges encode valid local transformations. We propose MolWorld, a molecule world model-guided framework that treats the current molecule-transfer graph as an evolving search state. At each iteration, MolWorld selects local anchor contexts, generates candidate molecules conditioned on these contexts, evaluates their properties, and uses a learned world model to update the evolving molecule world by retaining admissible candidates and inserting them into the molecule-transfer graph. The expanded molecule world then guides subsequent optimization. Experiments on property optimization and docking-based tasks show that MolWorld discovers high-property molecules while maintaining substantially stronger structural connectivity, supporting actionable and sequential molecular design.
title MolWorld: Molecule World Models for Actionable Molecular Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2605.08954