DiffusionST: A Deep Generative Diffusion Model-Based Framework for Enhancing Spatial Transcriptomics Data Quality and Identifying Spatial Domains

Recent advancements in spatial transcriptomics (ST) technology have generated substantial volumes of spatial transcriptome data. However, the quality of this data is often compromised due to the limitations of current sequencing technologies. To address this issue, DiffusionST proposes a method for imputing ST data and clustering the imputed data. The method employs a graph convolutional network model combined with a newly designed loss function, denoising data using the zero-inflated negative binomial distribution, and data enhancement through a diffusion model to improve clustering accuracy. DiffusionST demonstrates superior clustering accuracy compared to eight of the most popular ST clustering algorithms. DiffusionST also excels in data imputation when compared to five single-cell RNA sequencing imputation algorithms. Additionally, DiffusionST's robustness against noise is quantitatively validated by manually introducing random dropout noise into the dataset, where our model significantly enhances the quality of ST data. Moreover, DiffusionST is well-suited for high-resolution ST data and has been demonstrated, through survival analysis and cell-cell communication studies, to dissect spatial domains within breast cancer tissues. These findings provide strong evidence of DiffusionST's efficacy in handling ST data especially with strong noise, making it a valuable tool in this field.