arXiv:2412.04075cs.AI2024-12被引 4

构建基因属性基准测试,评估多种模型在生物任务中的表现差异。

Does your model understand genes? A benchmark of gene properties for biological and text models

  • 用统一的向量表示和简单预测模型,跨架构公平比较不同生物模型。
  • 蛋白语言模型在基因组与调控功能上表现更优,表达模型在定位任务中领先。
  • 适合生物信息学与AI交叉研究者,用于指导模型选型与研发策略。

近年来,深度学习特别是基础模型在生物研究中应用激增,涵盖文本模型与基于组学数据的模型。但因训练数据和下游任务差异,模型性能难以一致比较。为此,我们提出一种与架构无关的基准评估方法:不直接评估模型,而是利用各模型的实体表示向量,为每个任务训练简单的预测模型,确保输入输出一致。聚焦专业生物数据库中的基因属性,分为基因组特性、调控功能、定位、生物过程和蛋白特性五类,共定义数百个任务,涵盖二分类、多标签与多分类。应用于表达模型、大语言模型、蛋白语言模型、DNA模型及传统基线。结果表明,文本模型与蛋白语言模型在基因组与调控功能任务中普遍优于表达模型;而表达模型在定位任务中表现更佳。为保证可复现性,代码与数据已公开于 github.com/BiomedSciAI/gene-benchmark。

原文摘要 · Abstract (English)

The application of deep learning methods, particularly foundation models, in biological research has surged in recent years. These models can be text-based or trained on underlying biological data, especially omics data of various types. However, comparing the performance of these models consistently has proven to be a challenge due to differences in training data and downstream tasks. To tackle this problem, we developed an architecture-agnostic benchmarking approach that, instead of evaluating the models directly, leverages entity representation vectors from each model and trains simple predictive models for each benchmarking task. This ensures that all types of models are evaluated using the same input and output types. Here we focus on gene properties collected from professionally curated bioinformatics databases. These gene properties are categorized into five major groups: genomic properties, regulatory functions, localization, biological processes, and protein properties. Overall, we define hundreds of tasks based on these databases, which include binary, multi-label, and multi-class classification tasks. We apply these benchmark tasks to evaluate expression-based models, large language models, protein language models, DNA-based models, and traditional baselines. Our findings suggest that text-based models and protein language models generally outperform expression-based models in genomic properties and regulatory functions tasks, whereas expression-based models demonstrate superior performance in localization tasks. These results should aid in the development of more informed artificial intelligence strategies for biological understanding and therapeutic discovery. To ensure the reproducibility and transparency of our findings, we have made the source code and benchmark data publicly accessible for further investigation and expansion at github.com/BiomedSciAI/gene-benchmark.

生物模型基因属性基准测试蛋白语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。