arXiv:2511.14806q-bio.GNcs.AI2025-11AAAI被引 5

动态分词+上下文感知,让基因组模型更懂复杂序列

MergeDNA: Context-aware Genome Modeling with Dynamic Tokenization through Token Merging

  • 用可学习的动态分词器自动合并相邻碱基为词语
  • 在三个基因组基准上超越现有模型,零样本表现也领先
  • 适合做基因组分析、多组学研究的AI工程师和生物学家

基因组建模面临两大难题:不同区域信息密度差异大,且缺乏明确的最小词汇单元。现有方法依赖四种碱基或固定分词器,采用简单掩码语言建模预训练,难以适应基因组序列的复杂变化。本文提出MergeDNA,通过引入可微分的分词合并技术,构建层次化架构,联合优化动态基因组分词器与上下文感知的潜空间变换器。分词模块通过多层可微分合并块,在局部窗口约束下自动将相邻碱基合并为词语;潜空间编码器通过全注意力机制捕捉这些合并词的全局上下文。对称地使用潜空间解码器和局部解码器,模型以两项预训练任务进行学习:合并词重建同时训练分词器并自适应筛选关键词,自适应掩码词预测则学习预测筛选后的词以捕获重要信息。大量实验表明,该模型在三个主流基因组基准及多个多组学任务中,经微调或零样本评估均表现优异,优于典型分词方法和大规模基因组基础模型。

原文摘要 · Abstract (English)

Modeling genomic sequences faces two unsolved challenges: the information density varies widely across different regions, while there is no clearly defined minimum vocabulary unit. Relying on either four primitive bases or independently designed DNA tokenizers, existing approaches with naive masked language modeling pre-training often fail to adapt to the varying complexities of genomic sequences. Leveraging Token Merging techniques, this paper introduces a hierarchical architecture that jointly optimizes a dynamic genomic tokenizer and latent Transformers with context-aware pre-training tasks. As for network structures, the tokenization module automatically chunks adjacent bases into words by stacking multiple layers of the differentiable token merging blocks with local-window constraints, then a Latent Encoder captures the global context of these merged words by full-attention blocks. Symmetrically employing a Latent Decoder and a Local Decoder, MergeDNA learns with two pre-training tasks: Merged Token Reconstruction simultaneously trains the dynamic tokenization module and adaptively filters important tokens, while Adaptive Masked Token Modeling learns to predict these filtered tokens to capture informative contents. Extensive experiments show that MergeDNA achieves superior performance on three popular DNA benchmarks and several multi-omics tasks with fine-tuning or zero-shot evaluation, outperforming typical tokenization methods and large-scale DNA foundation models.

基因组建模动态分词多组学Transformer

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