Agents on a Tree: Pathwise Coordination for Multi-Objective Molecular Optimization
Not on the map. Why?
Abstract (arXiv)
Multi-objective molecular optimization requires searching vast chemical spaces under conflicting objectives, where early design decisions strongly constrain downstream outcomes. Existing methods typically rely on a single policy or fixed scalarization, which limits their ability to represent diverse trade-offs and to explore multiple promising design trajectories. We propose ATOM, a multi-agent framework that formulates molecular optimization as a tree-structured search. Each node corresponds to an atomic operation and hosts an agent specialized for a particular objective or decision context. Agents coordinate along different paths of the tree rather than enforcing a global consensus, enabling the method to maintain and compare alternative molecular evolution trajectories. A global memory of past optimization behaviors further supports balanced exploration and exploitation across objectives. This tree-structured interaction enables reasoning over long-horizon dependencies inherent in molecular design. Experiments on challenging multi-objective benchmarks involving activity, synthesizability, and ADMET-related properties show that ATOM consistently achieves improved Pareto coverage and hypervolume over strong baselines. These results demonstrate the effectiveness of pathwise multi-agent coordination for molecular optimization. Code is available at https://anonymous.4open.science/r/ATOM-41CE.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 33 of its 63 references that are in the map. First posted on arXiv 27 Mar 2026.
Builds on
33 of 63 references.
References · 33
- 2023GPT-4 Technical Reportcited by 18,396
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023Qwen Technical Reportcited by 3,344
- 2023Large Language Models as Optimizerscited by 864
- 2016Automatic chemical design using a data-driven continuous representation of moleculescited by 765
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 677
- 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 652
- 2022BioGPT: Generative Pre-trained Transformer for Biomedical Text Generation and Miningcited by 538
- 2023ChatDoctor: A Medical Chat Model Fine-Tuned on a Large Language Model Meta-AI (LLaMA) Using Medical Domain Knowledgecited by 325
- 2020GraphAF: a Flow-based Autoregressive Model for Molecular Graph Generationcited by 305
- 2018GuacaMol: Benchmarking Models for De Novo Molecular Designcited by 297
- 2023MedAlpaca -- An Open-Source Collection of Medical Conversational AI Models and Training Datacited by 286
21 more
- 2022Structure-based Drug Design with Equivariant Diffusion Modelscited by 217
- 2022Sample Efficiency Matters: A Benchmark for Practical Molecular Optimizationcited by 155
- 2023Mol-Instructions: A Large-Scale Biomolecular Instruction Dataset for Large Language Modelscited by 133
- 2021MARS: Markov Molecular Sampling for Multi-objective Drug Discoverycited by 125
- 2017Objective-Reinforced Generative Adversarial Networks (ORGAN) for Sequence Generation Modelscited by 117
- 2019Mol-CycleGAN - a generative model for molecular optimizationcited by 66
- 2024Efficient Evolutionary Search Over Chemical Space with Large Language Modelscited by 61
- 2017SMILES Enumeration as Data Augmentation for Neural Network Modeling of Moleculescited by 59
- 2020Accelerating high-throughput virtual screening through molecular pool-based active learningcited by 59
- 2023DrugAssist: A Large Language Model for Molecule Optimizationcited by 59
- 2020Optimizing Molecules using Efficient Queries from Property Evaluationscited by 50
- 2023Sample-Efficient Multi-Objective Learning via Generalized Policy Improvement Prioritizationcited by 48
- 2022Reinforced Genetic Algorithm for Structure-based Drug Designcited by 46
- 2022Retrieval-based Controllable Molecule Generationcited by 44
- 2022Computer-Aided Multi-Objective Optimization in Small Molecule Discoverycited by 31
- 2024LICO: Large Language Models for In-Context Molecular Optimizationcited by 26
- 2024Large Language Models are In-Context Molecule Learnerscited by 18
- 2025Collaborative Expert LLMs Guided Multi-Objective Molecular Optimizationcited by 11
- 2020Graph Polish: A Novel Graph Generation Paradigm for Molecular Optimizationcited by 6
- 2025AMoPO: Adaptive Multi-objective Preference Optimization without Reward Models and Reference Modelscited by 6
- 2025Bridging Molecular Graphs and Large Language Modelscited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.