arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.05075cs.LGq-bio.BM
收藏

面向分子预测任务的大语言模型基准测试

Benchmarking Large Language Models for Molecule Prediction Tasks

Aarhus University(奥胡斯大学);Microsoft(微软公司)

Zhiqiang Zhong, Kuangyu Zhou, Davide Mottin

AI总结:

该研究针对大语言模型在分子预测任务中的表现开展基准测试,通过6个标准数据集的多类任务对比现有机器学习模型,揭示其性能局限与协同提升潜力,并给出未来研究方向。

摘要:

大语言模型(LLMs)处于众多自然语言处理(NLP)任务的前沿。尽管LLMs在NLP领域已被广泛采用,但其在更广泛领域的大部分潜力仍未得到充分挖掘,且在设计与实现方面仍存在显著局限。值得注意的是,LLMs难以处理图这类结构化数据,在应对需要深厚专业知识的特定领域问题(如生物与化学领域问题)时往往表现不佳。\n本文探讨了一个核心问题:LLMs能否有效处理分子预测任务?我们的目标并非追求顶尖性能,而是评估LLMs对各类分子任务的助力作用。我们在6个标准分子数据集上确定了若干分类与回归预测任务。随后,我们精心设计了一组提示词以查询LLMs在这些任务上的表现,并将其性能与现有机器学习(ML)模型进行对比,这些模型包括基于文本的模型以及专门为分析分子几何结构设计的模型。\n我们的研究得出了几项关键结论:首先,LLMs在分子任务上的竞争力普遍落后于ML模型,尤其是与擅长捕捉分子几何结构的模型相比,凸显出LLMs理解图数据的能力有限。其次,当协同使用时,LLMs在提升ML模型性能方面展现出潜力。最后,我们讨论了利用LLMs开展分子预测任务面临的挑战与可行方向。代码与模型可在https://github.com/zhiqiangzhongddu/LLMaMol获取。

英文摘要:

Large Language Models (LLMs) stand at the forefront of a number of Natural Language Processing (NLP) tasks. Despite the widespread adoption of LLMs in NLP, much of their potential in broader fields remains largely unexplored, and significant limitations persist in their design and implementation. Notably, LLMs struggle with structured data, such as graphs, and often falter when tasked with answering domain-specific questions requiring deep expertise, such as those in biology and chemistry. In this paper, we explore a fundamental question: Can LLMs effectively handle molecule prediction tasks? Rather than pursuing top-tier performance, our goal is to assess how LLMs can contribute to diverse molecule tasks. We identify several classification and regression prediction tasks across six standard molecule datasets. Subsequently, we carefully design a set of prompts to query LLMs on these tasks and compare their performance with existing Machine Learning (ML) models, which include text-based models and those specifically designed for analysing the geometric structure of molecules. Our investigation reveals several key insights: Firstly, LLMs generally lag behind ML models in achieving competitive performance on molecule tasks, particularly when compared to models adept at capturing the geometric structure of molecules, highlighting the constrained ability of LLMs to comprehend graph data. Secondly, LLMs show promise in enhancing the performance of ML models when used collaboratively. Lastly, we engage in a discourse regarding the challenges and promising avenues to harness LLMs for molecule prediction tasks. The code and models are available at https://github.com/zhiqiangzhongddu/LLMaMol.

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

↑