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

MolCA: Molecular Graph-Language Modeling with Cross-Modal Projector and Uni-Modal Adapter

About

Language Models (LMs) have demonstrated impressive molecule understanding ability on various 1D text-related tasks. However, they inherently lack 2D graph perception - a critical ability of human professionals in comprehending molecules' topological structures. To bridge this gap, we propose MolCA: Molecular Graph-Language Modeling with Cross-Modal Projector and Uni-Modal Adapter. MolCA enables an LM (e.g., Galactica) to understand both text- and graph-based molecular contents via the cross-modal projector. Specifically, the cross-modal projector is implemented as a Q-Former to connect a graph encoder's representation space and an LM's text space. Further, MolCA employs a uni-modal adapter (i.e., LoRA) for the LM's efficient adaptation to downstream tasks. Unlike previous studies that couple an LM with a graph encoder via cross-modal contrastive learning, MolCA retains the LM's ability of open-ended text generation and augments it with 2D graph information. To showcase its effectiveness, we extensively benchmark MolCA on tasks of molecule captioning, IUPAC name prediction, and molecule-text retrieval, on which MolCA significantly outperforms the baselines. Our codes and checkpoints can be found at https://github.com/acharkq/MolCA.

Zhiyuan Liu, Sihang Li, Yanchen Luo, Hao Fei, Yixin Cao, Kenji Kawaguchi, Xiang Wang, Tat-Seng Chua• 2023

Related benchmarks

TaskDatasetResultRank
Molecular property predictionMoleculeNet BBBP (scaffold)
ROC AUC70.8
117
Molecule CaptioningChEBI-20 (test)
BLEU-40.54
107
Graph ClassificationHIV
ROC-AUC0.727
104
Graph property predictionTox21
ROC-AUC0.685
101
Molecular Property ClassificationMoleculeNet BBBP
ROC AUC70
41
Molecular Property ClassificationMoleculeNet BACE
ROC AUC79.8
36
Molecule Description GenerationChEBI-20 (test)
BLEU-262
34
Molecular Property ClassificationBACE (MoleculeNet) scaffold (test)
ROC-AUC0.798
30
Quantitative Solute-Solvent InteractionFreeSolv (test)
RMSE1.271
29
Molecular Property ClassificationMoleculeNet ClinTox
ROC-AUC89.5
27
Showing 10 of 47 rows

Other info

Code

Follow for update