让大模型学会读懂化学分子式SMILES,提升对分子结构的理解能力
Improving Chemical Understanding of LLMs via SMILES Parsing
- 将SMILES解析拆解为一系列明确的图结构任务,提供结构化监督
- 在分子指令基准上表现最佳,显著提升对环数等结构特征的识别能力
- 适合需要精准理解分子结构的研究者和开发者
大型语言模型(LLMs)在分子科学中的科研发现中日益成为强大工具。准确理解分子结构是这类模型的基本要求,而分子结构常以SMILES格式表示。然而当前的LLMs在解析SMILES方面仍存在困难,甚至无法完成计数分子环数等基础任务。为此,我们提出CLEANMOL框架,将SMILES解析转化为一系列清晰且确定性的任务,涵盖子图匹配到全局图匹配,旨在促进对分子图结构的全面理解。我们构建了一个带有自适应难度评分的分子预训练数据集,并在此基础上对开源LLMs进行预训练。实验表明,CLEANMOL不仅提升了模型对分子结构的理解能力,还在Mol-Instructions基准测试中达到最佳或与基线相当的表现。
原文摘要 · Abstract (English)
Large language models (LLMs) are increasingly recognized as powerful tools for scientific discovery, particularly in molecular science. A fundamental requirement for these models is the ability to accurately understand molecular structures, commonly encoded in the SMILES representation. However, current LLMs struggle to interpret SMILES, even failing to carry out basic tasks such as counting molecular rings. To address this limitation, we introduce CLEANMOL, a novel framework that formulates SMILES parsing into a suite of clean and deterministic tasks explicitly designed to promote graph-level molecular comprehension. These tasks span from subgraph matching to global graph matching, providing structured supervision aligned with molecular structural properties. We construct a molecular pretraining dataset with adaptive difficulty scoring and pre-train open-source LLMs on these tasks. Our results show that CLEANMOL not only enhances structural comprehension but also achieves the best or competes with the baseline on the Mol-Instructions benchmark.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。