arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2502.03499q-bio.GNcs.AIcs.LG
收藏

Omni-DNA:面向跨模态与多任务学习的统一基因组基础模型

Omni-DNA: A Unified Genomic Foundation Model for Cross-Modal and Multi-Task Learning

Imperial College London(帝国理工学院);Microsoft Research(微软研究院);Vector Institute(向量研究所);University Health Network(大学健康网络)

Zehui Li, Vallijah Subasri, Yifei Shen, Dongsheng Li, Yiren Zhao, Guy-Bart Stan, Caihua Shan

AI总结:

Omni-DNA通过两阶段训练构建了20M到1B参数的统一基因组基础模型,在26个任务中18个达到最优,并展示了DNA到文本和图像的跨模态能力。

摘要:

大型语言模型(LLMs)在多样化任务中展现出卓越的泛化能力,然而基因组基础模型(GFMs)仍需要为每个下游应用单独进行微调,随着模型规模的增长,这带来了显著的开销。此外,现有GFMs受限于固定的输出格式,限制了它们在各种基因组任务中的适用性。在本工作中,我们重新审视基于Transformer的自回归模型,并引入Omni-DNA,这是一个跨模态多任务模型家族,参数量从2000万到10亿不等。我们的方法包括两个阶段:(i)以下一token预测为目标在DNA序列上进行预训练,以及(ii)扩展多模态任务特定token,并同时对多个下游任务进行微调。在Nucleotide Transformer和GB基准上评估时,Omni-DNA在26个任务中的18个上取得了最先进的性能。通过多任务微调,Omni-DNA一次性处理10个乙酰化和甲基化任务,超越了在每个任务上单独训练的模型。最后,我们设计了两个复杂的基因组任务,DNA2Function和Needle-in-DNA,分别将DNA序列映射到文本功能描述和图像,表明Omni-DNA的跨模态能力能够拓宽基因组应用的范围。所有模型均可在https://huggingface.co/collections/zehui127获取。

英文摘要:

Large Language Models (LLMs) demonstrate remarkable generalizability across diverse tasks, yet genomic foundation models (GFMs) still require separate finetuning for each downstream application, creating significant overhead as model sizes grow. Moreover, existing GFMs are constrained by rigid output formats, limiting their applicability to various genomic tasks. In this work, we revisit the transformer-based auto-regressive models and introduce Omni-DNA, a family of cross-modal multi-task models ranging from 20 million to 1 billion parameters. Our approach consists of two stages: (i) pretraining on DNA sequences with next token prediction objective, and (ii) expanding the multi-modal task-specific tokens and finetuning for multiple downstream tasks simultaneously. When evaluated on the Nucleotide Transformer and GB benchmarks, Omni-DNA achieves state-of-the-art performance on 18 out of 26 tasks. Through multi-task finetuning, Omni-DNA addresses 10 acetylation and methylation tasks at once, surpassing models trained on each task individually. Finally, we design two complex genomic tasks, DNA2Function and Needle-in-DNA, which map DNA sequences to textual functional descriptions and images, respectively, indicating Omni-DNA's cross-modal capabilities to broaden the scope of genomic applications. All the models are available through https://huggingface.co/collections/zehui127

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

↑