arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.01780cs.LGq-bio.GN
收藏

D3LM:一种用于双向DNA理解和生成的离散DNA扩散语言模型

D3LM: A Discrete DNA Diffusion Language Model for Bidirectional DNA Understanding and Generation

Gaoling School of Artificial Intelligence(人工智能学院)

Zhao Yang, Hengchang Liu, Chuan Cao, Bing Su

AI总结:

D3LM通过掩码扩散统一双向DNA理解和生成,实现优于现有模型的性能。

摘要:

早期的DNA基础模型采用了BERT式训练, 在DNA理解任务上取得了良好的性能,但缺乏生成能力。最近的自回归模型能够实现DNA生成,但采用的是从左到右的因果建模,这对于具有内在双向调控关系的DNA来说效果不佳。我们提出了D3LM(Discrete DNA Diffusion Language Model),通过掩码扩散统一了双向表示学习和DNA生成。D3LM直接采用了核苷酸变换器(NT)v2架构,但将训练目标重新表述为离散DNA空间中的掩码扩散,使模型在单个模型中同时具备双向理解和生成能力。与相同规模的NT v2相比,D3LM在理解任务上取得了更好的性能。值得注意的是,在调控元件生成方面,D3LM实现了SFID为10.92,接近真实DNA序列(7.85),并大幅超越了先前自回归模型的最佳结果29.16。我们的工作表明扩散语言模型是统一DNA基础模型的一种有前景的范式。我们进一步对DNA领域中的掩码扩散模型进行了首次系统研究,探讨了诸如分词方案和采样策略等实用设计选择,从而提供了实证见解和坚实的基础,为未来研究奠定了基础。D3LM已发布在https://huggingface.co/collections/Hengchang-Liu/d3lm。

英文摘要:

Early DNA foundation models adopted BERT-style training, achieving good performance on DNA understanding tasks but lacking generative capabilities. Recent autoregressive models enable DNA generation, but employ left-to-right causal modeling that is suboptimal for DNA where regulatory relationships are inherently bidirectional. We present D3LM (\textbf{D}iscrete \textbf{D}NA \textbf{D}iffusion \textbf{L}anguage \textbf{M}odel), which unifies bidirectional representation learning and DNA generation through masked diffusion. D3LM directly adopts the Nucleotide Transformer (NT) v2 architecture but reformulates the training objective as masked diffusion in discrete DNA space, enabling both bidirectional understanding and generation capabilities within a single model. Compared to NT v2 of the same size, D3LM achieves improved performance on understanding tasks. Notably, on regulatory element generation, D3LM achieves an SFID of 10.92, closely approaching real DNA sequences (7.85) and substantially outperforming the previous best result of 29.16 from autoregressive models. Our work suggests diffusion language models as a promising paradigm for unified DNA foundation models. We further present the first systematic study of masked diffusion models in the DNA domain, investigating practical design choices such as tokenization schemes and sampling strategies, thereby providing empirical insights and a solid foundation for future research. D3LM has been released at https://huggingface.co/collections/Hengchang-Liu/d3lm.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑