Computer Science ›› 2026, Vol. 53 ›› Issue (7): 280-288.doi: 10.11896/jsjkx.250900078

• Database & Big Data & Data Science • Previous Articles     Next Articles

Graph-structure-aware Single-cell Transcriptomic Embedding Clustering Model

YANG Hang, HUANG Ruizhang, XUE Jingjing, QIN Yongbin, CHEN Yanping, LIN Chuan   

  1. College of Computer Science and Technology,Guizhou University,Guiyang 550025,China
    Text Computing and Cognitive Intelligence Engineering Research Center,Ministry of Education,Guizhou University,Guiyang 550025,China
    State Key Laboratory of Public Big Data,Guizhou University,Guiyang 550025,China
  • Received:2025-09-11 Revised:2026-01-06 Online:2026-07-15 Published:2026-07-10
  • About author:YANG Hang,born in 2000,postgra-duate.His main research interests include natural language processing and bio-clustering.
    HUANG Ruizhang,born in 1979,professor,Ph.D,is a member of CCF(No.52039M).Her main research interests include big data,data mining,information extraction.
  • Supported by:
    National Key Research and Development Program of China(2023YFC3304500),Key Funded Projects of the Science and Technology Foundation Program of Guizhou Province([2024]003) and Funded Projects of the Science-Technology Foundation Program of Guizhou Province([2023] 448).

Abstract: Cell clustering,a core task in single-cell RNA sequencing(scRNA-seq) analysis,plays a crucial role in scRNA-seq data analysis.In recent years,single-cell deep embedding representation models have gained popularity due to their ability to simultaneously learn feature representation and perform clustering.However,these models still face several significant challenges,including massive data,widespread dropout events,and complex noise patterns in the transcriptome.This paper proposes a graph-structure-aware single-cell transcriptomic embedding clustering model(scGAEC).This model innovatively integrates a contrastive learning mechanism with a self-developed graph-aware approach for more in-depth embedding representation.It also designs a decoder based on the zero-inflated negative binomial(ZINB) model to reconstruct gene expression information.By employing a joint mutual supervision strategy,the model optimizes clustering loss,contrastive loss,ZINB loss,and gene expression matrix reconstruction loss in a coordinated manner,thereby enhancing clustering performance and deep learning of latent representation.Experimental results demonstrate that scGAEC achieves average performance improvements of 30.63% in NMI and 52.17% in ARI over six competing models across four single-cell RNA sequencing datasets from different sequencing platforms,significantly outperforming various state-of-the-art methods.

Key words: Clustering, Single-cell RNA clustering, Graph-structure aware, Contrastive learning, Zero-inflated negative binomial model

CLC Number: 

  • Q811
[1]SIMMONS S K,LITHWICK-YANAI G,ADICONIS X,et al.Mostly natural sequencing-by-synthesis for scRNA-seq using Ultima sequencing[J].Nature Biotechnology,2023,41(2):204-211.
[2]PAPALEXI E,SATIJA R.Single-cell RNA sequencing to explore immune cell heterogeneity[J].Nature Reviews Immunology,2018,18(1):35-45.
[3]SVENSSON V,NATARAJAN K N,LY L H,et al.Power ana-lysis of single-cell RNA-sequencing experiments[J].Nature Methods,2017,14(4):381-387.
[4]KISELEV V Y,ANDREWS T S,HEMBERG M.Challenges in unsupervised clustering of single-cell RNA-seq data[J].Nature Reviews Genetics,2019,20(5):273-282.
[5]MACQUEEN J.Some methods for classification and analysis of multivariate observations[C]//Proceedings of the Fifth Berkeley Symposium on Mathematical Statistics and Probability.University of California Press,1967,5:281-298.
[6]JOHNSON S C.Hierarchical clustering schemes[J].Psy-chometrika,1967,32(3):241-254.
[7]GRÜN D,KESTER L,VAN O A.Validation of noise models for single-cell transcriptomics[J].Nature Methods,2014,11(6):637-640.
[8]LOPEZ R,REGIER J,COLE M B,et al.Deep generative mode-ling for single-cell transcriptomics[J].Nature Methods,2018,15(12):1053-1058.
[9]XIE J,GIRSHICK R,FARHADI A.Unsupervised deep embedding for clustering analysis[C]//International Conference on Machine Learning.PMLR,2016:478-487.
[10]TIAN T,WAN J,SONG Q,et al.Clustering single-cell RNA-seq data with a model-based deep learning approach[J].Nature Machine Intelligence,2019,1(4):191-198.
[11]TIAN T,ZHANG J,LIN X,et al.Model-based deep embedding for constrained clustering analysis of single cell RNA-seq data[J].Nature Communications,2021,12(1):1873.
[12]CHEN L,WANG W,ZHAI Y,et al.Deep soft K-means clustering with self-training for single-cell RNA sequence data[J].NAR Genomics and Bioinformatics,2020,2(2):lqaa039.
[13]LIU T,JIA C,BI Y,et al.scDFN:enhancing single-cell RNA-seq clustering with deep fusion networks[J].Briefings in Bioinformatics,2024,25(6):bbae486.
[14]YU Z,LU Y,WANG Y,et al.Zinb-based graph embedding autoencoder for single-cell rna-seq interpretations[C]//Procee-dings of the AAAI Conference on Artificial Intelligence.2022:4671-4679.
[15]CHENG Y,SU Y,YU Z,et al.Unsupervised deep embedded fusion representation of single-cell transcriptomics[C]//Procee-dings of the AAAI Conference on Artificial Intelligence.2023:5036-5044.
[16]JIA C.Kinetic foundation of the zero-inflated negative binomial model for single-cell RNA sequencing data[J].SIAM Journal on Applied Mathematics,2020,80(3):1336-1355.
[17]TU W,ZHOU S,LIU X,et al.Deep fusion clustering network[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2021:9978-9987.
[18]ZENG Y,ZHOU X,RAO J,et al.Accurately clustering single-cell RNA-seq data by capturing structural relations between cells through graph convolutional network[C]//2020 IEEE International Conference on Bioinformatics and Biomedicine(BIBM).IEEE,2020:519-522.
[19]WANG J,MA A,CHANG Y,et al.scGNN is a novel graph neural network framework for single-cell RNA-Seq analyses[J].Nature Communications,2021,12(1):1882.
[20]ERASLAN G,SIMON L M,MIRCEA M,et al.Single-cellRNA-seq denoising using a deep count autoencoder[J].Nature Communications,2019,10(1):390.
[21]CHEN L,WANG W,ZHAI Y,et al.Single-cell transcriptome data clustering via multinomial modeling and adaptive fuzzy k-means algorithm[J].Frontiers in Genetics,2020,11:295.
[22]LI X,WANG K,LYU Y,et al.Deep learning enables accurate clustering with batch effect removal in single-cell RNA-seq ana-lysis[J].Nature Communications,2020,11(1):2338.
[23]HU D,LIANG K,ZHOU S,et al.scDFC:a deep fusion clustering method for single-cell RNA-seq data[J].Briefings in Bioinformatics,2023,24(4):bbad216.
[24]VELIČKOVIĆ P,CUCURULL G,CASANOVA A,et al.Graph attention networks[J].arXiv:1710.10903,2017.
[25]WAN H,CHEN L,DENG M.scNAME:neighborhood contrastive clustering with ancillary mask estimation for scRNA-seq data[J].Bioinformatics,2022,38(6):1575-1583.
[26]WANG J,XIA J,WANG H,et al.scDCCA:deep contrastiveclustering for single-cell RNA-seq data based on auto-encoder network[J].Briefings in Bioinformatics,2023,24(1):bbac625.
[27]SCHAUM N,KARKANIAS J,NEFF N F,et al.Single-celltranscriptomics of 20 mouse organs creates a Tabula Muris:The Tabula Muris Consortium[J].Nature,2018,562(7727):367.
[28]CAMP J G,SEKINE K,GERBER T,et al.Multilineage communication regulates human liver bud development from pluripotency[J].Nature,2017,546(7659):533-538.
[29]YU J,CHEN N,GAO M,et al.Unsupervised gene-cell collective representation learning with optimal transport[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2024:356-364.
[1] ZHU Rong, HU Mengyao, DAI Lingyun, LI Feng. Single-cell Multi-omics Clustering Method Based on Missing Value Imputation and Cross-modalAlignment [J]. Computer Science, 2026, 53(7): 289-297.
[2] DONG Dong, JIN Pengchao. Model-based Trajectory Anomalies Detection Algorithm for Longitudinal Data [J]. Computer Science, 2026, 53(6A): 250800004-8.
[3] ZHANG Zihao, WU Zezhong. Optimization of HAN-based GNN-Transformer Collaborative Contrastive Learning Framework [J]. Computer Science, 2026, 53(6A): 250900103-8.
[4] CHANG Yanan, SUN Yi, CUI Jianqun, YAN Xianglong, ZONG Chenglu. Fuzzy Clustering-based DTN Routing Algorithm for IoT [J]. Computer Science, 2026, 53(6): 358-366.
[5] NI Yongting, QIAN Jin, YAN Shaowei, WU Yueyang. Fuzzy Three-way Clustering Based on Mean Shift [J]. Computer Science, 2026, 53(6): 332-338.
[6] YE Jianhong, WU Yongjin, HUANG Hongkai. Structure-aware Trace Clustering Method Based on Process Edit Distance [J]. Computer Science, 2026, 53(5): 164-173.
[7] TIAN Xin, ZHU Guosheng, XIONG Yuran, WU You. Study on Drug Target Affinity Prediction Based on Multi-view Comparison and Homology Information [J]. Computer Science, 2026, 53(5): 337-345.
[8] LIU Yichen, LIN Yan, ZHOU Zeyu, GUO Shengnan, LIN Youfang, WAN Huaiyu. Efficient Semantic-aware Trajectory Representation Learning Method via State Space Model [J]. Computer Science, 2026, 53(4): 134-142.
[9] CHENG Zimeng, YANG Xinyue, AI Haojun, WANG Zhongyuan. Unsupervised Infrared Image Generation Method Based on Dual Semantic Contrastive Learning [J]. Computer Science, 2026, 53(4): 260-268.
[10] YU Chengcheng, JIANG Yongfa, CHEN Fangshu, WANG Jiahui, MENG Xiankai. Multi-view Exercise Representation and Forgetting Mechanism for Deep KnowledgeTracing [J]. Computer Science, 2026, 53(3): 107-114.
[11] WANG Yiming, JIAO Min, ZHAO Suyun, CHEN Hong, LI Cuiping. Prompt-conditioned Representation Learning with Diffusion Models for Semi-supervised Clustering [J]. Computer Science, 2026, 53(3): 158-165.
[12] LI Wenli, FENG Xiaonian, QIAN Tieyun. Few-shot Continuous Toxicity Detection Based on Large Language Model Augmentation [J]. Computer Science, 2026, 53(3): 321-330.
[13] CHEN Xiaolan, MAO Shun, LI Weisheng, LIN Ronghua, TANG Yong. Robust Knowledge Tracing Model Based on Two-level Contrastive Learning [J]. Computer Science, 2026, 53(2): 31-38.
[14] YANG Ming, HE Chaobo, YANG Jiaqi. Direction-aware Siamese Network for Knowledge Concept Prerequisite Relation Prediction [J]. Computer Science, 2026, 53(2): 39-47.
[15] LI Chunying, TANG Zhikang, ZHUANG Zhiwei, LI Wenbo, GUO Yanxi, ZHANG Xiaowei. DCL-FKT:Personalized Knowledge Tracing via Dual Contrastive Learning and ForgettingMechanism [J]. Computer Science, 2026, 53(2): 99-106.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!