HybriDNA融合变压器与Mamba2,高效处理超长DNA序列并生成功能元件。
HybriDNA: A Hybrid Transformer-Mamba2 Long-Range DNA Language Model
- 采用混合架构,结合注意力机制与状态空间模型处理长序列。
- 在33个基准上达顶尖水平,可生成具有特定功能的调控元件。
- 支持从300M到7B参数扩展,性能随规模提升稳定增长。
自然语言处理与大模型的发展激发了对DNA建模的兴趣,常被称为“生命语言”。但其面临两大挑战:需处理超长序列且保持单核苷酸分辨率,同时兼顾生成与理解任务。为此,我们提出HybriDNA,一种解码器仅的混合Transformer-Mamba2 DNA语言模型,无缝融合注意力机制与选择性状态空间模型。该设计使HybriDNA能以单核苷酸精度处理长达131kb的DNA序列。在来自BEND、GUE和LRB基准的33个DNA理解数据集上,其表现达到当前最优,并展现出生成具有期望特性的合成顺式调控元件(CREs)的卓越能力。此外,我们验证了HybriDNA遵循预期缩放规律,性能随模型规模从300M、3B至7B参数持续提升。这些结果凸显其多功能性及推动DNA研究与应用的潜力,为理解与工程化“生命语言”开辟新路径。
原文摘要 · Abstract (English)
Advances in natural language processing and large language models have sparked growing interest in modeling DNA, often referred to as the "language of life". However, DNA modeling poses unique challenges. First, it requires the ability to process ultra-long DNA sequences while preserving single-nucleotide resolution, as individual nucleotides play a critical role in DNA function. Second, success in this domain requires excelling at both generative and understanding tasks: generative tasks hold potential for therapeutic and industrial applications, while understanding tasks provide crucial insights into biological mechanisms and diseases. To address these challenges, we propose HybriDNA, a decoder-only DNA language model that incorporates a hybrid Transformer-Mamba2 architecture, seamlessly integrating the strengths of attention mechanisms with selective state-space models. This hybrid design enables HybriDNA to efficiently process DNA sequences up to 131kb in length with single-nucleotide resolution. HybriDNA achieves state-of-the-art performance across 33 DNA understanding datasets curated from the BEND, GUE, and LRB benchmarks, and demonstrates exceptional capability in generating synthetic cis-regulatory elements (CREs) with desired properties. Furthermore, we show that HybriDNA adheres to expected scaling laws, with performance improving consistently as the model scales from 300M to 3B and 7B parameters. These findings underscore HybriDNA's versatility and its potential to advance DNA research and applications, paving the way for innovations in understanding and engineering the "language of life".
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。