A versatile informative diffusion model for single-cell ATAC-seq data generation and analysis

Lei Huang,Lei Xiong,Na Sun,Zunpeng Liu,Ka-Chun Wong,Manolis Kellis

2024-08-27

Abstract:The rapid advancement of single-cell ATAC sequencing (scATAC-seq) technologies holds great promise for investigating the heterogeneity of epigenetic landscapes at the cellular level. The amplification process in scATAC-seq experiments often introduces noise due to dropout events, which results in extreme sparsity that hinders accurate analysis. Consequently, there is a significant demand for the generation of high-quality scATAC-seq data in silico. Furthermore, current methodologies are typically task-specific, lacking a versatile framework capable of handling multiple tasks within a single model. In this work, we propose ATAC-Diff, a versatile framework, which is based on a latent diffusion model conditioned on the latent auxiliary variables to adapt for various tasks. ATAC-Diff is the first diffusion model for the scATAC-seq data generation and analysis, composed of auxiliary modules encoding the latent high-level variables to enable the model to learn the semantic information to sample high-quality data. Gaussian Mixture Model (GMM) as the latent prior and auxiliary decoder, the yield variables reserve the refined genomic information beneficial for downstream analyses. Another innovation is the incorporation of mutual information between observed and hidden variables as a regularization term to prevent the model from decoupling from latent variables. Through extensive experiments, we demonstrate that ATAC-Diff achieves high performance in both generation and analysis tasks, outperforming state-of-the-art models.

Genomics,Biomolecules

What problem does this paper attempt to address?

The problems that this paper attempts to solve mainly focus on several key challenges in single - cell ATAC - seq (scATAC - seq) data analysis:

1. **Data Sparsity and Noise Problems**:
- The amplification process in scATAC - seq experiments often introduces noise due to dropout events, which makes the data extremely sparse and thus hinders accurate analysis.
- Such sparsity and noise problems seriously affect the study of epigenetic landscape heterogeneity, especially at the cell level.

2. **Requirement for High - Quality scATAC - seq Data Generation**:
- Currently, there is a lack of effective tools to generate high - quality scATAC - seq data, which is crucial for subsequent bioinformatics analysis.

3. **Task - Specificity of Existing Methods**:
- Existing methods are usually designed for specific tasks and lack a general framework to handle multiple tasks. This limits the flexibility and efficiency of these methods in various application scenarios.

To solve these problems, the paper proposes a new model named **ATAC - Diff**. This model is based on the latent diffusion model and can adapt to various tasks by introducing an auxiliary module to encode potential high - level variables. Specifically, the main contributions of ATAC - Diff include:

- **Applying the Diffusion Model to scATAC - seq Data Analysis for the First Time**: Utilize the powerful generation ability of the diffusion model to deal with the sparsity and noise problems of scATAC - seq data.
- **Introducing a High - Information - Content Latent Space**: Use the Gaussian Mixture Model (GMM) as a latent prior and combine mutual information maximization to ensure that the model learns meaningful biological representations.
- **Multi - task Processing Ability**: Achieve multiple tasks, such as data generation, denoising, filling in missing values, and subgroup clustering, within a unified framework.

Through extensive experimental verification, ATAC - Diff performs excellently in multiple tasks, exceeding or reaching the performance levels of existing models.