用对比学习分析基因组序列,实现无需组装的高效读段映射与结构识别。
Learning Genomic Structure from $k$-mers
- 通过对比学习训练编码器,使同一基因组区域的k-mer在嵌入空间中聚集。
- 在E. coli和古DNA数据上达到BWA-aln水平的映射精度与速度。
- 支持自监督或有监督训练,适合宏基因组物种鉴定等下游任务。
测序产生的大量短核苷酸序列(reads)需重新拼接以重建完整基因组。本文提出一种基于对比学习的方法,训练编码器生成嵌入,使同一基因组区域的序列在嵌入空间中聚类,保留其序列连续性形成的轨迹。模型仅基于基因组结构训练,可生成通用的k-mer表示,适用于多种读段数据分析任务。我们在E. coli基因组上验证该框架,应用于模拟古DNA读段映射与结构变异检测,并展示其在宏基因组物种识别中的潜力。通过引入领域特定噪声模型增强嵌入鲁棒性,且在有线性参考基因组时采用距离阈值参数Γ的监督对比学习。模型也可完全自监督训练,无需依赖专用算法构建全基因组组装。基于预训练嵌入的小型预测头在短基因组上表现媲美BWA-aln,在准确率和运行时间上相当。由于对基因组总大小具有良好的扩展性,该方法在宏基因组及人类基因组级别基因组映射中极具前景。
原文摘要 · Abstract (English)
Sequencing a genome to determine an individual's DNA produces an enormous number of short nucleotide subsequences known as reads, which must be reassembled to reconstruct the full genome. We present a method for analyzing this type of data using contrastive learning, in which an encoder model is trained to produce embeddings that cluster together sequences from the same genomic region. The sequential nature of genomic regions is preserved in the form of trajectories through this embedding space. Trained solely to reflect the structure of the genome, the resulting model provides a general representation of $k$-mer sequences, suitable for a range of downstream tasks involving read data. We apply our framework to learn the structure of the $E.\ coli$ genome, and demonstrate its use in simulated ancient DNA (aDNA) read mapping and identification of structural variations. Furthermore, we illustrate the potential of using this type of model for metagenomic species identification. We show how incorporating a domain-specific noise model can enhance embedding robustness, and how a supervised contrastive learning setting can be adopted when a linear reference genome is available, by introducing a distance thresholding parameter $Γ$. The model can also be trained fully self-supervised on read data, enabling analysis without the need to construct a full genome assembly using specialized algorithms. Small prediction heads based on a pre-trained embedding are shown to perform on par with BWA-aln, the current gold standard approach for aDNA mapping, in terms of accuracy and runtime for short genomes. Given the method's favorable scaling properties with respect to total genome size, inference using our approach is highly promising for metagenomic applications and for mapping to genomes comparable in size to the human genome.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。