arXiv:2505.20836cs.LGq-bio.GN2025-05被引 3

用混合架构蒸馏让小模型超越大教师,在基因组建模中表现更优。

HAD: Hybrid Architecture Distillation Outperforms Teacher in Genomic Sequence Modeling

  • 结合蒸馏与重建任务,用分组掩码策略提升预训练效率。
  • 参数量相近时性能领先,部分任务甚至超过500倍大的教师模型。
  • 适合追求高效基因组建模的科研与工业应用者。

受自然语言领域掩码语言建模(MLM)成功的启发,自监督预训练与微调范式在DNA序列建模中也取得了显著进展。然而,以往方法常依赖海量预训练数据或参数量巨大的基础模型,带来巨大计算负担。为解决此问题,许多研究尝试使用更紧凑的模型以达成类似效果,但仍存在明显差距。本文提出一种混合架构蒸馏(HAD)方法,融合蒸馏与重建任务,实现更高效、更有效的预训练。具体地,我们以NTv2-500M作为教师模型,并设计分组掩码策略,在MLM预训练中同时对可见标记对齐特征嵌入并重构不可见标记。在核苷酸转换器基准(Nucleotide Transformer Benchmark)和基因组基准上的全面实验表明,相较于参数量相近的模型,我们的模型表现优异;更令人惊讶的是,其在某些子任务上甚至超越了参数量超过500倍的教师模型。最后,通过t-SNE可视化,显示本模型能深入理解基因组序列的内在表征模式。

原文摘要 · Abstract (English)

Inspired by the great success of Masked Language Modeling (MLM) in the natural language domain, the paradigm of self-supervised pre-training and fine-tuning has also achieved remarkable progress in the field of DNA sequence modeling. However, previous methods often relied on massive pre-training data or large-scale base models with huge parameters, imposing a significant computational burden. To address this, many works attempted to use more compact models to achieve similar outcomes but still fell short by a considerable margin. In this work, we propose a Hybrid Architecture Distillation (HAD) approach, leveraging both distillation and reconstruction tasks for more efficient and effective pre-training. Specifically, we employ the NTv2-500M as the teacher model and devise a grouping masking strategy to align the feature embeddings of visible tokens while concurrently reconstructing the invisible tokens during MLM pre-training. To validate the effectiveness of our proposed method, we conducted comprehensive experiments on the Nucleotide Transformer Benchmark and Genomic Benchmark. Compared to models with similar parameters, our model achieved excellent performance. More surprisingly, it even surpassed the distillation ceiling-teacher model on some sub-tasks, which is more than 500 $\times$ larger. Lastly, we utilize t-SNE for more intuitive visualization, which shows that our model can gain a sophisticated understanding of the intrinsic representation pattern in genomic sequences.

基因组建模蒸馏自监督学习

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