Empowering Molecule Discovery for Molecule-Caption Translation with Large Language Models: A ChatGPT Perspective

11 Jun 2023  ·  Jiatong Li, Yunqing Liu, Wenqi Fan, Xiao-Yong Wei, Hui Liu, Jiliang Tang, Qing Li ·

Molecule discovery plays a crucial role in various scientific fields, advancing the design of tailored materials and drugs. However, most of the existing methods heavily rely on domain experts, require excessive computational cost, or suffer from sub-optimal performance. On the other hand, Large Language Models (LLMs), like ChatGPT, have shown remarkable performance in various cross-modal tasks due to their powerful capabilities in natural language understanding, generalization, and in-context learning (ICL), which provides unprecedented opportunities to advance molecule discovery. Despite several previous works trying to apply LLMs in this task, the lack of domain-specific corpus and difficulties in training specialized LLMs still remain challenges. In this work, we propose a novel LLM-based framework (MolReGPT) for molecule-caption translation, where an In-Context Few-Shot Molecule Learning paradigm is introduced to empower molecule discovery with LLMs like ChatGPT to perform their in-context learning capability without domain-specific pre-training and fine-tuning. MolReGPT leverages the principle of molecular similarity to retrieve similar molecules and their text descriptions from a local database to enable LLMs to learn the task knowledge from context examples. We evaluate the effectiveness of MolReGPT on molecule-caption translation, including molecule understanding and text-based molecule generation. Experimental results show that compared to fine-tuned models, MolReGPT outperforms MolT5-base and is comparable to MolT5-large without additional training. To the best of our knowledge, MolReGPT is the first work to leverage LLMs via in-context learning in molecule-caption translation for advancing molecule discovery. Our work expands the scope of LLM applications, as well as providing a new paradigm for molecule discovery and design.

PDF Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Text-based de novo Molecule Generation ChEBI-20 MolReGPT (GPT-4-0413) Text2Mol 59.3 # 1
BLEU 85.7 # 3
Exact Match 28.0 # 6
Levenshtein 17.14 # 9
MACCS FTS 90.3 # 2
RDK FTS 80.5 # 3
Morgan FTS 73.9 # 4
Frechet ChemNet Distance (FCD) 0.41 # 6
Validity 89.9 # 9
Parameter Count None # 1
Molecule Captioning ChEBI-20 MolReGPT (GPT-4-0314) BLEU-2 60.7 # 6
BLEU-4 52.5 # 6
ROUGE-1 63.4 # 11
ROUGE-2 47.6 # 14
ROUGE-L 56.2 # 15
METEOR 61.0 # 9
Text2Mol 58.5 # 3
Text-based de novo Molecule Generation ChEBI-20 MolReGPT (GPT-3.5-turbo) Text2Mol 57.1 # 10
BLEU 79.0 # 12
Exact Match 13.9 # 15
Levenshtein 24.91 # 4
MACCS FTS 84.7 # 13
RDK FTS 70.8 # 13
Morgan FTS 62.4 # 15
Frechet ChemNet Distance (FCD) 0.57 # 10
Validity 88.7 # 11
Molecule Captioning ChEBI-20 MolReGPT (GPT-3.5-turbo) BLEU-2 56.5 # 12
BLEU-4 48.2 # 12
ROUGE-1 45.0 # 18
ROUGE-2 54.3 # 3
ROUGE-L 58.5 # 10
METEOR 62.3 # 7
Text2Mol 56.0 # 7

Methods


No methods listed for this paper. Add relevant methods here