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

Developing ChemDFM as a large language foundation model for chemistry

About

Artificial intelligence (AI) has played an increasingly important role in chemical research. However, most models currently used in chemistry are specialist models that require training and tuning for specific tasks. A more generic and efficient solution would be an AI model that could address many tasks and support free-form dialogue in the broad field of chemistry. In its utmost form, such a generalist AI chemist could be referred to as Chemical General Intelligence. Large language models (LLMs) have recently logged tremendous success in the general domain of natural language processing, showing emerging task generalization and free-form dialogue capabilities. However, domain knowledge of chemistry is largely missing when training general-domain LLMs. The lack of such knowledge greatly hinders the performance of generalist LLMs in the field of chemistry. To this end, we develop ChemDFM, a pioneering LLM for chemistry trained on 34B tokens from chemical literature and textbooks, and fine-tuned using 2.7M instructions. As a result, it can understand and reason with chemical knowledge in free-form dialogue. Quantitative evaluations show that ChemDFM significantly surpasses most representative open-source LLMs. It outperforms GPT-4 on a great portion of chemical tasks, despite the substantial size difference. We have open-sourced the inference codes, evaluation datasets, and model weights of ChemDFM on Huggingface (https://huggingface.co/OpenDFM/ChemDFM-v1.0-13B).

Zihan Zhao, Da Ma, Lu Chen, Liangtai Sun, Zihao Li, Yi Xia, Bo Chen, Hongshen Xu, Zichen Zhu, Su Zhu, Shuai Fan, Guodong Shen, Kai Yu, Xin Chen• 2024

Related benchmarks

TaskDatasetResultRank
Molecule CaptioningChEBI-20 (test)
BLEU-40.265
107
Molecular Property ClassificationMoleculeNet BBBP
ROC AUC50.5
41
Molecular Property ClassificationMoleculeNet BACE
ROC AUC59.5
36
Molecular Property ClassificationMoleculeNet ClinTox
ROC-AUC60
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.51
21
Molecule GenerationChEBI-20 (test)
Exact Match1.8
14
Molecular OptimizationMolecular Pharmacology Optimization
Overall Optimization Score0.1997
13
Showing 10 of 23 rows

Other info

Follow for update