arXiv:2508.04757q-bio.GNcs.LG2025-08被引 1

用固定嵌入替代微调,让基因组预测更高效环保

Embedding Is (Almost) All You Need: Retrieval-Augmented Inference for Generalizable Genomic Prediction Tasks

  • 直接提取预训练模型的固定嵌入,搭配轻量分类器推理
  • 跨数据分布场景下性能超越微调,推理速度提升10-22倍
  • 适合部署在多样或未知基因组环境,碳排放降低超8倍

DNABERT-2、Nucleotide Transformer和HyenaDNA等大型预训练DNA语言模型在多个基因组基准上表现优异。但多数应用依赖昂贵的微调,且仅在训练与测试数据分布相似时效果最佳。本文探究任务特定微调是否必要,发现仅提取这些模型的固定嵌入并输入轻量分类器即可达到竞争性性能。在不同数据分布的评估中,嵌入方法常优于微调,推理时间减少10至20倍。例如,在增强子分类中,HyenaDNA嵌入结合zCurve达0.68准确率(微调为0.58),推理时间减少88%,碳排放从0.17 kg降至0.02 kg CO2;在非TATA启动子分类中,DNABERT-2嵌入结合zCurve或GC含量达0.85准确率(微调为0.89),碳足迹降低至0.02 kg(原为0.44 kg)。结果表明,嵌入管道在保持强预测能力的同时,碳效率提升超10倍。代码已开源。

原文摘要 · Abstract (English)

Large pre-trained DNA language models such as DNABERT-2, Nucleotide Transformer, and HyenaDNA have demonstrated strong performance on various genomic benchmarks. However, most applications rely on expensive fine-tuning, which works best when the training and test data share a similar distribution. In this work, we investigate whether task-specific fine-tuning is always necessary. We show that simple embedding-based pipelines that extract fixed representations from these models and feed them into lightweight classifiers can achieve competitive performance. In evaluation settings with different data distributions, embedding-based methods often outperform fine-tuning while reducing inference time by 10x to 20x. Our results suggest that embedding extraction is not only a strong baseline but also a more generalizable and efficient alternative to fine-tuning, especially for deployment in diverse or unseen genomic contexts. For example, in enhancer classification, HyenaDNA embeddings combined with zCurve achieve 0.68 accuracy (vs. 0.58 for fine-tuning), with an 88% reduction in inference time and over 8x lower carbon emissions (0.02 kg vs. 0.17 kg CO2). In non-TATA promoter classification, DNABERT-2 embeddings with zCurve or GC content reach 0.85 accuracy (vs. 0.89 with fine-tuning) with a 22x lower carbon footprint (0.02 kg vs. 0.44 kg CO2). These results show that embedding-based pipelines offer over 10x better carbon efficiency while maintaining strong predictive performance. The code is available here: https://github.com/NIRJHOR-DATTA/EMBEDDING-IS-ALMOST-ALL-YOU-NEED.

基因组预测嵌入提取低碳推理

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