arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.12950q-bio.QMcs.AIcs.CEcs.CLcs.LG
收藏

MolecularGPT:用于少样本分子性质预测的开放大语言模型(LLM)

MolecularGPT: Open Large Language Model (LLM) for Few-Shot Molecular Property Prediction

New York University Shanghai(上海纽约大学);Xiamen University(厦门大学);University of California, San Francisco(加利福尼亚大学旧金山分校);Zhejiang University(浙江大学);The Hong Kong Polytechnic University(香港理工大学)

Yuyan Liu, Sirui Ding, Sheng Zhou, Wenqi Fan, Qiaoyu Tan

AI总结:

本文提出MolecularGPT,通过指令微调大语言模型并利用零样本/少样本上下文学习,在无需微调的情况下实现少样本分子性质预测,并在多个数据集上超越图神经网络和现有LLM基线。

摘要:

分子性质预测(MPP)是药物发现中一项基础且关键的任务。然而,以往方法受限于需要大量带标签的分子,并且其对未见任务和新任务的泛化能力有限,而这两点对实际应用都至关重要。为应对这些挑战,我们提出了用于少样本MPP的MolecularGPT。从指令微调的视角出发,我们基于覆盖超过1000个性质预测任务的人工整理分子指令对大语言模型(LLM)进行微调。这使得能够构建一个通用且专用的LLM,通过零样本和少样本上下文学习(ICL)即可适应新的MPP任务,而无需任何微调。MolecularGPT在10个下游评估数据集上展现出具有竞争力的上下文推理能力,为少样本分子预测任务设立了新的基准。更重要的是,仅使用两个示例(two-shot),MolecularGPT就能在7个数据集中的4个上超越标准的监督图神经网络方法。在零样本设置下,它还优于最先进的LLM基线,分类准确率提升最高达15.7%,回归指标(如RMSE)降低17.9。本研究证明了LLM作为有效的少样本分子性质预测器的潜力。代码可在 https://github.com/NYUSHCS/MolecularGPT 获取。

英文摘要:

Molecular property prediction (MPP) is a fundamental and crucial task in drug discovery. However, prior methods are limited by the requirement for a large number of labeled molecules and their restricted ability to generalize for unseen and new tasks, both of which are essential for real-world applications. To address these challenges, we present MolecularGPT for few-shot MPP. From a perspective on instruction tuning, we fine-tune large language models (LLMs) based on curated molecular instructions spanning over 1000 property prediction tasks. This enables building a versatile and specialized LLM that can be adapted to novel MPP tasks without any fine-tuning through zero- and few-shot in-context learning (ICL). MolecularGPT exhibits competitive in-context reasoning capabilities across 10 downstream evaluation datasets, setting new benchmarks for few-shot molecular prediction tasks. More importantly, with just two-shot examples, MolecularGPT can outperform standard supervised graph neural network methods on 4 out of 7 datasets. It also excels state-of-the-art LLM baselines by up to 15.7% increase on classification accuracy and decrease of 17.9 on regression metrics (e.g., RMSE) under zero-shot. This study demonstrates the potential of LLMs as effective few-shot molecular property predictors. The code is available at https://github.com/NYUSHCS/MolecularGPT.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑