mlsanityData to 7 Oct 2026

One Sequence, Many Decodings: CAGenMol-2 Recasts Drug Design as Masked Molecular Inference

arXiv 2609.34301 ↗First posted 28 Sep 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Drug design couples property evaluation, conditional generation, structure-based design, and local optimization, yet machine learning systems typically address these capabilities with separate task-specific models. We introduce CAGenMol-2, a masked diffusion molecular language model that represents molecules, continuous scalar properties, and 3D protein pockets within a single wrapped sequence. Within this pretrained interface, downstream operations are selected by which sequence regions are observed or masked at inference, allowing one checkpoint to perform property prediction, property- and pocket-conditioned generation, and partial-constraint design without task-specific architectures or backbone fine-tuning. We further propose Adaptive Fragment Optimization (AdaFO), a gradient-free mask-and-refill search that turns the masked decoder into an iterative local molecular optimizer. On CrossDocked2020, AdaFO increases Success Rate from 30.2% to 70.8%, the best reported under this protocol, while largely preserving drug-likeness and diversity. Finally, scaffold-preserving directional editing and CRBN/VHL case studies demonstrate its use in compound design workflows spanning local molecular editing, structure-based prioritization, and downstream simulation-based screening.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 34 of its 68 references that are in the map. First posted on arXiv 28 Sep 2026.

Builds on

34 of 68 references.

References · 34

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2017Proximal Policy Optimization Algorithmscited by 14,441
  4. 2022Classifier-Free Diffusion Guidancecited by 5,717
  5. 2015Trust Region Policy Optimizationcited by 2,503
  6. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
  7. 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
  8. 2017Selective Classification for Deep Neural Networkscited by 799
  9. 2016Automatic chemical design using a data-driven continuous representation of moleculescited by 762
  10. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
  11. 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 647
  12. 2019Deep Ensembles: A Loss Landscape Perspectivecited by 438
22 more
  1. 2019Deep Evidential Regressioncited by 354
  2. 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 332
  3. 2018GuacaMol: Benchmarking Models for De Novo Molecular Designcited by 294
  4. 2018Molecular Sets (MOSES): A Benchmarking Platform for Molecular Generation Modelscited by 270
  5. 2017Molecular De Novo Design through Deep Reinforcement Learningcited by 264
  6. 2021Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Developmentcited by 258
  7. 2022Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pocketscited by 182
  8. 20233D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Predictioncited by 176
  9. 2024Simple Guidance Mechanisms for Discrete Diffusion Modelscited by 148
  10. 2024LlaSMol: Advancing Large Language Models for Chemistry with a Large-Scale, Comprehensive, High-Quality Instruction Tuning Datasetcited by 124
  11. 2022Generating 3D Molecules for Target Protein Bindingcited by 104
  12. 2022Multi-Objective GFlowNetscited by 100
  13. 2024DecompDiff: Diffusion Models with Decomposed Priors for Structure-Based Drug Designcited by 87
  14. 2022Regression Transformer: Concurrent sequence regression and generation for molecular language modelingcited by 49
  15. 2025GenMol: A Drug Discovery Generalist with Discrete Diffusioncited by 47
  16. 2023Gotta be SAFE: A New Framework for Molecular Designcited by 45
  17. 2023DrugCLIP: Contrastive Protein-Molecule Representation Learning for Virtual Screeningcited by 28
  18. 2024DecompOpt: Controllable and Decomposed Diffusion Models for Structure-based Molecular Optimizationcited by 27
  19. 2024The Prevalence of Neural Collapse in Neural Multivariate Regressioncited by 21
  20. 2025FoNE: Precise Single-Token Number Embeddings via Fourier Featurescited by 21
  21. 2025MolChord: Structure-Sequence Alignment for Protein-Guided Drug Designcited by 2
  22. 2026CAGenMol: Condition-Aware Diffusion Language Model for Goal-Directed Molecular Generationcited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Ligand-Conditioned Discrete Diffusion for Protein Sequence-Structure Co-Design
  2. 2026Scaffold-Conditioned Preference Triplets for Controllable Molecular Optimization with Large Language Models
  3. 2026Empowering LLMs for Structure-Based Drug Design via Exploration-Augmented Latent Inference
  4. 2025SynCraft: Guiding Large Language Models to Predict Edit Sequences for Molecular Synthesizability Optimization
  5. 2026MolWorld: Molecule World Models for Actionable Molecular Optimization
  6. 2025Graph Diffusion Transformers are In-Context Molecular Designers
  7. 2026SEISMO: Explanation-Aware, Trajectory-Conditioned LLM Agents for Sample-Efficient Molecular Optimisation
  8. 2026Purely Agent-Driven Black-Box Optimization for Biological Design