Agentic Molecular Recovery via Molecule-Aware Exploration

Suwan Yoon; Changhee Lee

cs.AIJun 4, 2026

Agentic Molecular Recovery via Molecule-Aware Exploration

Authors: Suwan Yoon, Changhee Lee

Organizations: Department of Artificial Intelligence, Korea University / Seoul, Korea

Abstract

Text-guided molecular generation with LLMs often yields invalid SMILES. We argue that invalid drafts should be addressed through a shift from validity-oriented repair to identity-preserving molecular recovery: the objective is not only to restore chemical validity, but also to preserve target-relevant structural cues and recover the molecular identity implied by the description. This perspective reveals the limitations of existing correction strategies. Post-hoc repair can recover validity while distorting key structures, LLM-only correction can introduce unintended global drift, and generic agentic correction remains constrained by greedy single-candidate trajectories even when equipped with executable RDKit edit tools. To address these limitations, we propose AMREC, which couples molecule-aware mismatch tracking with expanded candidate exploration and trajectory-level selection. On invalid ChEBI-20 drafts from three backbone models, AMREC achieves the strongest overall recovery profile across structural, exact-match, and string-level metrics.

Explore similar work

MolE-RAG: Molecular Structure-Enhanced Retrieval-Augmented Generation for Chemistry
Jun 4, 2026Joey Chan, Wonbin Kweon, Ashley Shin +4Molecular Property Prediction Vast Chemical Space
MolGVR: A Chemistry-Grounded Framework for Text-to-Molecule Generation
Jul 31, 2026Qian Tan, Xuanyu Zhu, Lei Jiang +3Molecular Generation Text Generation
BiMol-Diff: A Unified Diffusion Framework for Molecular Generation and Captioning
Apr 27, 2026Aditya Hemant Shahane, Anuj Kumar Sirohi, Devansh Arora +3Molecular Generation

Jun 4, 2026 · cs.LGJ/K move · Enter open · S save

MolE-RAG: Molecular Structure-Enhanced Retrieval-Augmented Generation for Chemistry

Joey Chan, Wonbin Kweon, Ashley Shin, Niharika Bhattacharjee+3

University of Illinois Urbana-Champaign · University of California, San Diego

Large language models (LLMs) have shown promise for molecular property prediction, but their ability to reason over chemical structures remains limited, as molecular representations such as SMILES differ substantially from the natural language on which LLMs are primarily trained. To bridge this semantic and chemical knowledge gap, we propose MolE-RAG, a training-free, molecule-centric retrieval-augmented generation framework for LLM-based molecular property prediction. MolE-RAG augments each prediction with three complementary sources of inference-time context: retrieved chemistry literature, molecule-specific information including compound synonyms, identifiers, functional group annotations, and physicochemical descriptors, and structurally similar molecules retrieved from the training set. We evaluate MolE-RAG across nine molecular property prediction tasks using proprietary, chemistry-specialized, and open-source LLMs. Across general-purpose LLMs, MolE-RAG improves ROC-AUC by up to 28 percentage points on classification tasks and reduces regression RMSE by up to 67% relative to a SMILES-only baseline. We further find that the utility of each context source varies across models and tasks, with different models benefiting most from textual retrieval, molecular context, or structural retrieval. These results suggest that molecule-centric retrieval can improve LLM-based molecular property prediction without model fine-tuning while providing a flexible framework for integrating heterogeneous chemical knowledge at inference time.

Molecular Property Prediction Vast Chemical Space Hievi-Rag Annotations

Open paper Ask about it PDF

Abstract

Explore similar work

MolE-RAG: Molecular Structure-Enhanced Retrieval-Augmented Generation for Chemistry

MolGVR: A Chemistry-Grounded Framework for Text-to-Molecule Generation

BiMol-Diff: A Unified Diffusion Framework for Molecular Generation and Captioning