Retrieval-guided Twin Fusion with Similarity-aware Contrast for Molecule-Text Alignment
Authors: Shunshun Gu, Shengqi Qiu, Hang Zhou, Xiao Luo
Organizations: University of Wisconsin–Madison, Madison, Wisconsin, USA · University of North Carolina at Chapel Hill, Chapel Hill, North Carolina, USA
This paper studies the problem of molecule-text alignment, which aims to project molecules and their textual descriptions into a joint latent space for downstream tasks including molecule search and molecular property prediction. Previous approaches typically combine graph structure mining with contrastive learning to enhance joint representation learning. However, they typically neglect fine-grained semantic relationships between substructures and texts, leading to suboptimal performance on downstream tasks. Towards this end, we propose a novel approach named Retrieval-guided Twin Fusion with Similarity-aware Contrast (RISEN) for molecule-text alignment. The core idea of RISEN is to construct a latent twin molecule for each substructure with cross-modal retrieval for semantic enhancement. In particular, for each substructure query, we retrieve relevant textual descriptions and sample several molecules that share similar descriptions of substructures. Then, we aggregate their representations via attention pooling for a twin latent representation, which would be further fused with the original substructure for representation enrichment. In addition, we measure the similarity across substructures and texts, which would further guide cross-modal contrastive learning with soft thresholding. Extensive experiments on benchmark datasets validate the superiority of the proposed RISEN in comparison with existing baselines.
Figures & tables
Text to Molecule
Molecule to Text
Methods
# Params
R@1
R@5
R@10
MRR
R@1
R@5
R@10
MRR
1D SMILES + 2D Graph
MoMu ( Su et al., 2022 )
111M
4.90
14.48
20.69
10.33
5.08
12.82
18.93
9.89
MolFM ( Luo et al., 2023 )
138M
16.14
30.67
39.54
23.63
13.90
28.69
36.21
21.42
MolFM (fine-tuned) ( Luo et al., 2023 )
138M
29.39
50.26
58.49
39.34
29.76
50.53
58.63
39.56
MolCA ( Liu et al., 2023b )
111M
35.09
62.14
69.77
47.33
37.95
66.81
74.48
50.80
Table 1. Performance comparison for zero-shot molecule-text retrieval on PCDes test set (scaffold split). Baseline results are quoted from Park et al. (2025) . We bold the best results and underline the second-best.
Text to Molecule
Molecule to Text
Methods
R@1
R@20
R@1
R@20
1D SMILES + 2D Graph
MoMu-S ( Su et al., 2022 )
40.8
86.1
40.9
86.2
MoMu-K ( Su et al., 2022 )
41.6
87.8
41.8
87.5
MoleculeSTM ( Liu et al., 2023a )
44.3
90.3
45.8
88.4
MolCA ( Liu et al., 2023b )
66.0
93.5
66.6
94.6
Table 2. Performance comparison for zero-shot molecule-text retrieval on PubChem324kV2 test set. Baseline results are quoted from Park et al. (2025) . We bold the best results and underline the second-best.
Methods
BBBP
Tox21
ToxCast
ClinTox
MUV
HIV
BACE
SIDER
Avg.
MoleculeSTM ( Liu et al., 2023a )
70.6
75.7
65.2
86.6
65.7
77.0
82.0
63.7
73.3
MolFM ( Luo et al., 2023 )
72.9
77.2
64.4
79.7
76.0
78.8
83.9
64.2
74.6
MoMu ( Su et al., 2022 )
70.5
75.6
63.4
79.9
70.6
75.9
76.7
60.5
71.6
MolCA-SMILES ( Liu et al., 2023b )
70.8
76.0
56.2
89.0
–
–
79.3
61.1
–
Atomas ( Zhang et al., 2025 )
73.7
77.9
66.9
93.2
76.3
80.6
83.1
64.4
77.0
MolBridge ( Park et al., 2025 )
77.6
84.7
70.3
94.8
76.8
77.8
84.5
66.9
79.2
Table 3. Results for molecular property prediction tasks (ROC-AUC %) on the MoleculeNet benchmark. We bold the best results and underline the second-best.
Figure 1. Sensitivity analysis of the prototype fusion weight α (left) and the soft-label mixing weight β (right) on PCDes. Two line charts compare retrieval performance across prototype fusion weights and soft-label mixing weights on the PCDes benchmark.
Figure 2. Qualitative comparison of functional-group phrases identified by RISEN and MolBridge ( Park et al., 2025 ) against ground-truth phrases from PubChem descriptions. A qualitative case-study figure compares chemical functional-group phrases identified by RISEN and MolBridge against PubChem ground-truth phrases.
Molecular embedding models can serve as foundational infrastructure for computational chemistry and drug discovery, where reusable vector representations support property prediction, virtual screening, and retrieval. Most molecular encoders are specialist models built around a single molecular view, producing unconditional vectors with no language interface for varying the representation. We ask whether multimodal large language models (MLLMs), which natively process images, text, and symbolic inputs, can instead serve as general molecular embedding models that produce embeddings conditioned on both a molecular profile and a natural-language semantic context. We introduce MolEmb, a lightweight framework that adapts MLLMs by aligning molecular profiles with textual descriptions in a shared embedding space using a bidirectional contrastive objective. The resulting embedding model is competitive on molecular property prediction and supports cross-modal molecule--text retrieval in the same space. We further introduce MolCAR, a diagnostic benchmark for context-aware retrieval, and find that context-aware molecular embedding is primarily a data property of the supervision. These results suggest that MLLMs are not merely chemistry assistants or generators, but a viable and extensible route to general molecular embedding models.
Xinjian Zhao, Xiangru Jian, Yaoyao Xu +4
The Chinese University of Hong Kong, Shenzhen · Shanghai Artificial Intelligence Laboratory · Cheriton School of Computer Science, University of Waterloo
Bridging molecular structures and natural language is essential for controllable design. Autoregressive models struggle with long-range dependencies, while standard diffusion processes apply uniform corruption across positions, which can distort structurally informative tokens. We present BiMol-Diff, a unified diffusion framework for the paired tasks of text-conditioned molecule generation and molecule captioning. Our key component is a token-aware noise schedule that assigns position-dependent corruption based on token recovery difficulty, preserving harder-to-recover substructures during the forward process. On ChEBI-20 and M3-20M, BiMol-Diff improves molecule reconstruction with a 15.4% relative gain in Exact Match and achieves strong captioning results, attaining best BLEU and BERTScore among compared baselines. These results indicate token-aware noising improves fidelity in molecular structure-language modelling.
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.
Joey Chan, Wonbin Kweon, Ashley Shin +4
University of Illinois Urbana-Champaign · University of California, San Diego