Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Towards 3D Molecule-Text Interpretation in Language Models

About

Language Models (LMs) have greatly influenced diverse domains. However, their inherent limitation in comprehending 3D molecular structures has considerably constrained their potential in the biomolecular domain. To bridge this gap, we focus on 3D molecule-text interpretation, and propose 3D-MoLM: 3D-Molecular Language Modeling. Specifically, 3D-MoLM enables an LM to interpret and analyze 3D molecules by equipping the LM with a 3D molecular encoder. This integration is achieved by a 3D molecule-text projector, bridging the 3D molecular encoder's representation space and the LM's input space. Moreover, to enhance 3D-MoLM's ability of cross-modal molecular understanding and instruction following, we meticulously curated a 3D molecule-centric instruction tuning dataset -- 3D-MoIT. Through 3D molecule-text alignment and 3D molecule-centric instruction tuning, 3D-MoLM establishes an integration of 3D molecular encoder and LM. It significantly surpasses existing baselines on downstream tasks, including molecule-text retrieval, molecule captioning, and more challenging open-text molecular QA tasks, especially focusing on 3D-dependent properties. We release our codes and datasets at https://github.com/lsh0520/3D-MoLM.

Sihang Li, Zhiyuan Liu, Yanchen Luo, Xiang Wang, Xiangnan He, Kenji Kawaguchi, Tat-Seng Chua, Qi Tian• 2024

Related benchmarks

TaskDatasetResultRank
Molecule CaptioningChEBI-20 (test)
BLEU-40.171
107
Molecular Property ClassificationMoleculeNet BBBP
ROC AUC53.8
41
Molecular Property ClassificationMoleculeNet BACE
ROC AUC55.5
36
Molecule Description GenerationChEBI-20 (test)
BLEU-26.7
34
Molecular Property ClassificationMoleculeNet ClinTox
ROC-AUC53.7
27
Reagent PredictionMol-Instructions
Exact Match0.00e+0
24
Forward reaction predictionMol-Instructions
Exact Match0.00e+0
24
RetrosynthesisMol-Instructions
Exact Match0.00e+0
24
Molecular Property ClassificationMoleculeNet SIDER
ROC-AUC0.497
21
Molecule GenerationChEBI-20 (test)
Exact Match0.00e+0
14
Showing 10 of 24 rows

Other info

Follow for update