arXiv:2502.18405cs.LG2025-02被引 6

改进基因组模型的掩码机制,提升无微调场景下的特征提取性能。

Enhancing DNA Foundation Models to Address Masking Inefficiencies

  • 用编码器-解码器结构替代传统BERT,解决掩码任务与实际应用脱节问题
  • 在200万条DNA条形码数据上,闭/开世界分类任务均显著优于基准模型
  • 特别适合不微调直接用于特征提取的基因组分析场景

掩码语言建模(MLM)作为预训练目标被广泛应用于基因组序列建模。尽管预训练模型能有效作为下游任务的编码器,但预训练与推理之间的分布差异严重影响性能:预训练需将[MASK]标记映射为预测,而推理时[MASK]不存在,导致编码器不重视非[MASK]区域的编码,且浪费参数和算力于仅服务于MLM的任务。本文提出一种基于掩码自编码器框架的改进型编码器-解码器架构,针对BERT-based Transformer解决该效率问题。我们实证表明,这种错配在无需微调的基因组分析流程中尤为严重。在包含超过200万条唯一DNA条形码的BIOSCAN-5M数据集上评估,相较于因果模型和双向架构的MLM预训练模型,本方法在闭世界与开世界分类任务中均取得显著性能提升。

原文摘要 · Abstract (English)

Masked language modelling (MLM) as a pretraining objective has been widely adopted in genomic sequence modelling. While pretrained models can successfully serve as encoders for various downstream tasks, the distribution shift between pretraining and inference detrimentally impacts performance, as the pretraining task is to map [MASK] tokens to predictions, yet the [MASK] is absent during downstream applications. This means the encoder does not prioritize its encodings of non-[MASK] tokens, and expends parameters and compute on work only relevant to the MLM task, despite this being irrelevant at deployment time. In this work, we propose a modified encoder-decoder architecture based on the masked autoencoder framework, designed to address this inefficiency within a BERT-based transformer. We empirically show that the resulting mismatch is particularly detrimental in genomic pipelines where models are often used for feature extraction without fine-tuning. We evaluate our approach on the BIOSCAN-5M dataset, comprising over 2 million unique DNA barcodes. We achieve substantial performance gains in both closed-world and open-world classification tasks when compared against causal models and bidirectional architectures pretrained with MLM tasks.

基因组建模掩码自编码特征提取Transformer

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