Abstract:Abstract Motivation With the development of single-cell RNA sequencing (scRNA-seq) techniques, increasingly more large-scale gene expression datasets become available. However, to analyze datasets produced by different experiments, batch effects among different datasets must be considered. Although several methods have been recently published to remove batch effects in scRNA-seq data, two problems remain to be challenging and not completely solved: (i) how to reduce the distribution differences of different batches more accurately; and (ii) how to align samples from different batches to recover the cell type clusters. Results We proposed a novel deep-learning approach, which is a hierarchical distribution-matching framework assisted with contrastive learning to address these two problems. Firstly, we design a hierarchical framework for distribution matching based on a deep autoencoder. This framework employs an adversarial training strategy to match the global distribution of different batches. This provides an improved foundation to further match the local distributions with a maximum mean discrepancy-based loss. For local matching, we divide cells in each batch into clusters and develop a contrastive learning mechanism to simultaneously align similar cluster pairs and keep noisy pairs apart from each other. This allows to obtain clusters with all cells of the same type (true positives), and avoid clusters with cells of different type (false positives). We demonstrate the effectiveness of our method on both simulated and real datasets. Results show that our new method significantly outperforms the state-of-the-art methods and has the ability to prevent overcorrection. Availability and implementation The python code to generate results and figures in this article is available at https://github.com/zhanglabNKU/HDMC, the data underlying this article is also available at this github repository. Supplementary information Supplementary data are available at Bioinformatics online.

Deep learning enables accurate clustering and batch effect removal in single-cell RNA-seq analysis

Deep learning enables accurate clustering with batch effect removal in single-cell RNA-seq analysis

Deep Learning for clustering single-cell RNA-seq Data

Clustering single-cell RNA-seq data with a model-based deep learning approach

Batch alignment of single-cell transcriptomics data using deep metric learning

Self-supervised deep clustering of single-cell RNA-seq data to hierarchically detect rare cell populations

A joint deep learning model enables simultaneous batch effect correction, denoising, and clustering in single-cell transcriptomics

HDMC: a novel deep learning-based framework for removing batch effects in single-cell RNA-seq data

A critical assessment of clustering algorithms to improve cell clustering and identification in single-cell transcriptome study

scDFN: enhancing single-cell RNA-seq clustering with deep fusion networks

Integrating Deep Supervised, Self-Supervised and Unsupervised Learning for Single-Cell RNA-seq Clustering and Annotation

Deep Batch Integration and Denoise of Single‐Cell RNA‐Seq Data

Deep learning-based clustering method for single-cell RNA data

Deep soft K-means clustering with self-training for single-cell RNA sequence data

Clustering single-cell RNA sequencing data via iterative smoothing and self-supervised discriminative embedding

A Fusion Learning Model Based on Deep Learning for Single-Cell RNA Sequencing Data Clustering

Deep structural clustering for single-cell RNA-seq data jointly through autoencoder and graph neural network

A Hybrid Deep Clustering Approach for Robust Cell Type Profiling Using Single-cell RNA-seq Data

scASDC: Attention Enhanced Structural Deep Clustering for Single-cell RNA-seq Data

Single-cell Curriculum Learning-based Deep Graph Embedding Clustering

Single-Cell Transcriptome Data Clustering via Multinomial Modeling and Adaptive Fuzzy K-Means Algorithm