DeepGene: An Efficient Foundation Model for Genomics based on Pan-genome Graph Transformer

Xiang Zhang,Mingjie Yang,Xunhang Yin,Yining Qian,Fei Sun
DOI: https://doi.org/10.1101/2024.04.24.590879
2024-05-14
Abstract:Decoding the language of DNA sequences is a fundamental problem in genome research. Mainstream pre-trained models like DNABERT-2 and Nucleotide Transformer have demonstrated remarkable achievements across a spectrum of DNA analysis tasks. Yet, these models still face the pivotal challenge of (1) genetic language diversity, or the capability to capture genetic variations across individuals or populations in the foundation models; (2) model efficiency, specifically how to enhance performance at scalable costs for large-scale genetic foundational models; (3) length extrapolation, or the ability to accurately interpret sequences ranging from short to long within a unified model framework. In response, we introduce DeepGene, a model leveraging Pan-genome and Minigraph representations to encompass the broad diversity of genetic language. DeepGene employs the rotary position embedding to improve the length extrapolation in various genetic analysis tasks. On the 28 tasks in Genome Understanding Evaluation, DeepGene reaches the top position in 9 tasks, second in 5, and achieves the overall best score. DeepGene outperforms other cutting-edge models for its compact model size and superior efficiency in processing sequences of varying lengths. The datasets and source code of DeepGene are available at GitHub (https://github.com/wds-seu/DeepGene).
Bioinformatics
What problem does this paper attempt to address?
The paper addresses three issues: 1) genetic linguistic diversity by using a whole-genome graph transformer to capture genetic variation between individuals or populations; 2) model efficiency by balancing performance and scalable costs in handling large-scale genetic models; and 3) length extrapolation, which accurately interprets sequences from short to long within a unified model framework. To achieve this, the paper proposes the DeepGene model, which utilizes whole-genome and Minigraph representations to enhance length extrapolation capability and demonstrates excellent performance on various genetic analysis tasks.