Zhiyuan Liu, Sihang Li, Yanchen Luo, Hao Fei, Yixin Cao, Kenji Kawaguchi, Xiang Wang, Tat-Seng Chua
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.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Molecule Captioning | ChEBI-20 | BLEU-2 | 62 | MolCA, Galac1.3B |
| Molecule Captioning | ChEBI-20 | BLEU-4 | 53.1 | MolCA, Galac1.3B |
| Molecule Captioning | ChEBI-20 | METEOR | 65.1 | MolCA, Galac1.3B |
| Molecule Captioning | ChEBI-20 | ROUGE-1 | 68.1 | MolCA, Galac1.3B |
| Molecule Captioning | ChEBI-20 | ROUGE-2 | 53.7 | MolCA, Galac1.3B |
| Molecule Captioning | ChEBI-20 | ROUGE-L | 61.8 | MolCA, Galac1.3B |
| Molecule Captioning | ChEBI-20 | BLEU-2 | 61.6 | MolCA, Galac125M |
| Molecule Captioning | ChEBI-20 | BLEU-4 | 52.9 | MolCA, Galac125M |
| Molecule Captioning | ChEBI-20 | METEOR | 63.9 | MolCA, Galac125M |
| Molecule Captioning | ChEBI-20 | ROUGE-1 | 67.4 | MolCA, Galac125M |
| Molecule Captioning | ChEBI-20 | ROUGE-2 | 53.3 | MolCA, Galac125M |
| Molecule Captioning | ChEBI-20 | ROUGE-L | 61.5 | MolCA, Galac125M |