4 个月前

TOMG-Bench:基于文本的开放分子生成模型评估

TOMG-Bench:基于文本的开放分子生成模型评估

摘要

在本文中,我们提出了基于文本的开放分子生成基准(Text-based Open Molecule Generation Benchmark, TOMG-Bench),这是首个用于评估大语言模型(LLMs)在开放域内分子生成能力的基准。TOMG-Bench 包含了三个主要任务的数据集:分子编辑(MolEdit)、分子优化(MolOpt)和定制分子生成(MolCustom)。每个任务进一步包含三个子任务,每个子任务包含 5,000 个测试样本。鉴于开放分子生成的固有复杂性,我们还开发了一个自动化评估系统,该系统有助于测量生成分子的质量和准确性。我们对 25 个大语言模型进行了全面的基准测试,揭示了当前文本引导下的分子发现中的局限性和潜在改进领域。此外,在专门设计用于解决 TOMG-Bench 提出挑战的指令调优数据集 OpenMolIns 的帮助下,Llama3.1-8B 在 TOMG-Bench 上的表现超过了所有开源的大语言模型,甚至比 GPT-3.5-turbo 高出 46.5%。我们的代码和数据集可通过 https://github.com/phenixace/TOMG-Bench 获取。

代码仓库

phenixace/tomg-bench
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
description-guided-molecule-generation-onGalactica-125M (OpenMolIns-light)
wAcc: 13.136
description-guided-molecule-generation-onClaude-3
wAcc: 30.47
description-guided-molecule-generation-onMolT5-small
wAcc: 1.299
description-guided-molecule-generation-onGPT-4o
wAcc: 32.29
description-guided-molecule-generation-onClaude-3.5
wAcc: 35.92
description-guided-molecule-generation-onGPT-3.5-turbo
wAcc: 18.58
description-guided-molecule-generation-onchatglm-9B
wAcc: 13.137
description-guided-molecule-generation-onGPT-4-turbo
wAcc: 34.23
description-guided-molecule-generation-onQwen2-7B-Instruct
wAcc: 0.15
description-guided-molecule-generation-onBioT5-base
wAcc: 4.21
description-guided-molecule-generation-onGalactica-125M (OpenMolIns-large)
wAcc: 23.42
description-guided-molecule-generation-onMolT5-large
wAcc: 2.89
description-guided-molecule-generation-onMolT5-base
wAcc: 1.300
description-guided-molecule-generation-onLlama-3.1-1B-Instruct
wAcc: 1.99
description-guided-molecule-generation-onGalactica-125M (OpenMolIns-xlarge)
wAcc: 25.73
description-guided-molecule-generation-onLlama-3.1-8B (OpenMolIns-large)
wAcc: 27.22
description-guided-molecule-generation-onMistral-7B-Instruct-v0.2
wAcc: 4.81
description-guided-molecule-generation-onGalactica-125M (OpenMolIns-small)
wAcc: 15.18
description-guided-molecule-generation-onyi-1.5-9B
wAcc: 7.32
description-guided-molecule-generation-onLlama3.2-1B (OpenMolIns-large)
wAcc: 8.10
description-guided-molecule-generation-onLlama3.1-8B-Instruct
wAcc: 14.09
description-guided-molecule-generation-onLlama3-8B-Instruct
wAcc: 13.75
description-guided-molecule-generation-onGemini-1.5-pro
wAcc: 34.80
description-guided-molecule-generation-onLlama3-70B-Instruct (INT4)
wAcc: 23.93
description-guided-molecule-generation-onGalactica-125M (OpenMolIns-medium)
wAcc: 19.89

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
TOMG-Bench:基于文本的开放分子生成模型评估 | 论文 | HyperAI超神经