Computer Science ›› 2026, Vol. 53 ›› Issue (4): 101-111.doi: 10.11896/jsjkx.250500097

• Interdisciplinary Integration of Artificial Intelligence and Theoretical Computer Science • Previous Articles     Next Articles

KGMamba:Gene Regulatory Network Prediction Model Based on Kolmogorov-Arnold Network Optimizing Graph Convolutional Network and Mamba

GAO Tai, REN Yanzhang, WANG Huiqing, LI Ying, WANG Bin   

  1. College of Computer Science and Technology(College of Data Science), Taiyuan University of Technology, Jinzhong, Shanxi 030600, China
  • Received:2025-05-22 Revised:2025-09-03 Online:2026-04-15 Published:2026-04-08
  • About author:GAO Tai,born in 2000,postgraduate,is a member of CCF(No.Z6724G).His main research interests include deep learning and bioinformatics.
    WANG Bin,born in 1986,Ph.D,professor,Ph.D supervisor.His main research interests include deep learning,brain imaging research,medical images and bioinformatics.
  • Supported by:
    National Natural Science Foundation of China(62176177) and Science and Technology Cooperation and Exchange Special Projects of Shanxi(202304041101034).

Abstract: Gene regulatory network(GRN) inference is pivotal for deciphering cell development mechanisms and propelling precisionmedicine research.However,existing deep learning approaches confront challenges of high computational complexity and inadequate global feature capture.To tackle this,a novel efficient prediction model integrating the Kolmogorov-Arnold network(KAN) driven graph convolutional network(KGCN) and Mamba module is proposed.Firstly,the multi-layer perceptron(MLP) in traditional graph convolution is replaced by KAN’s learnable spline functions,which retain local feature extraction while reducing redundancy through restructured computation,significantly improving efficiency.Secondly,the Mamba module is innovatively incorporated to prioritize attention to gene nodes critical for global regulation via its selective mechanism.Together,these components enable a unified optimization of both local and global feature modeling.Experimental comparisons with six other deep-learning models on public datasets are performed.Results demonstrate that this model outperforms others in AUC and AUPR performance metrics,while also showcasing remarkable robustness and computational efficiency,further demonstrating the superiority of the model.

Key words: Gene regulatory network, Deep learning, Kolmogorov-Arnold network, Graph convolutional network, Mamba

CLC Number: 

  • TP183
[1]NGUYEN H,TRAN D,TRAN B,et al.A comprehensive survey of regulatory network inference methods using single cell RNA sequencing data[J].Briefings in Bioinformatics,2021,22(3):bbaa190.
[2]MARBACH D,COSTELLO J C,KÜFFNER R,et al.Wisdomof crowds for robust gene network inference[J].Nature Methods,2012,9(8):796-804.
[3]RICE J J,TU Y,STOLOVITZKY G.Reconstructing biological networks using conditional correlation analysis[J].Bioinformatics,2005,21(6):765-773.
[4]FAITH J J,HAYETE B,THADEN J T,et al.Large-scale mapping and validation of Escherichia coli transcriptional regulation from a compendium of expression profiles[J].PLoS Biology,2007,5(1):54-66.
[5]FRIEDMAN N.Inferring cellular networks using probabilisticgraphical models[J].Science,2004,303(5659):799-805.
[6]JI R,GENG Y,QUAN X.Inferring gene regulatory networks with graph convolutional network based on causal feature reconstruction[J].Scientific Reports,2024,14(1):21342.
[7]WEI P J,GUO Z,GAO Z,et al.Inference of gene regulatory networks based on directed graph convolutional networks[J].Briefings in Bioinformatics,2024,25(4):bbae309.
[8]KOMMU S,WANG Y,WANG Y,et al.Gene Regulatory Network Inference with Joint Representation from Graph Neural Network and Single-Cell Foundation Model[J/OL].https://www.biorxiv.org/content/10.1101/2024.12.16.628715v1.full.pdf.
[9]LIU Z,WANG Y,VAIDYA S,et al.Kan:Kolmogorov-Arnold networks[J].arXiv:2404.19756,2024.
[10]GU A,DAO T.Mamba:Linear-time sequence modeling with selective state spaces[J].arXiv:2312.00752,2023.
[11]YANG B,XU Y,MAXWELL A,et al.MICRAT:a novel algorithm for inferring gene regulatory networks using time series gene expression data[J].BMC Systems Biology,2018,12:19-29.
[12]LI L,SUN L,CHEN G,et al.LogBTF:gene regulatory network inference using Boolean threshold network model from single-cell gene expression data[J].Bioinformatics,2023,39(5):btad256.
[13]HUYNH-THU V A,GEURTS P.dynGENIE3:dynamical GENIE3 for the inference of gene networks from time series expression data[EB/OL].Scientific Reports,2018,8:3384.https://www.nature.com/articles/s41598-018-21715-0.
[14]MARGOLIN A A,NEMENMAN I,BASSO K,et al.ARAC-NE:an algorithm for the reconstruction of gene regulatory networks in a mammalian cellular context[C]//BMC Bioinformatics.2006:1-15.
[15]ZENG Y,HE Y,ZHENG R,et al.Inferring single-cell gene regulatory network by non-redundant mutual information[J].Briefings in Bioinformatics,2023,24(5):bbad326.
[16]ZHANG Y,WANG M,WANG Z,et al.MetaSEM:gene regulatory network inference from single-cell RNA data by meta-learning[J].International Journal of Molecular Sciences,2023,24(3):2595.
[17]MA M Y,SUN J X,HU C L.Modeling Gene Regulatory Networks with Global Coupling Parameters[J].Computer Science,2023,50(S2):138-144.
[18]HUYNH-THU V A,IRRTHUM A,WEHENKEL L,et al.Inferring regulatory networks from expression data using tree-based methods[J].PloS One,2010,5(9):e12776.
[19]LI Z J,LIAO S,LIU A F,et al.Gene Association Analysis Algorithm for GeneRegulatory Network[J].Computer Engineering and Applications,2025,61(3):155-165.
[20]ZHAO M,HE W,TANG J,et al.A hybrid deep learning framework for gene regulatory network inference from single-cell transcriptomic data[J].Briefings in Bioinformatics,2022,23(2):bbab568.
[21]YUAN Y,BAR-JOSEPH Z.Deep learning for inferring gene relationships from single-cell expression data[J].Proceedings of the National Academy of Sciences,2019,116(52):27151-27158.
[22]CHEN J,CHEONG C W,LAN L,et al.DeepDRIM:a deep neural network to reconstruct cell-type-specific gene regulatory network using single-cell RNA-seq data[J].Briefings in Bioinformatics,2021,22(6):bbab325.
[23]SHU H,DING F,ZHOU J,et al.Boosting single-cell gene regulatory network reconstruction via bulk-cell transcriptomic data[J].Briefings in Bioinformatics,2022,23(5):bbac389.
[24]XU J,ZHANG A,LIU F,et al.STGRNS:an interpretabletransformer-based method for inferring gene regulatory networks from single-cell transcriptomic data[J].Bioinformatics,2023,39(4):btad165.
[25]HAMILTON W,YING Z,LESKOVEC J.Inductive representation learning on large graphs[C]//Proceedings of the 31st International Conference on Neural Information Processing Systems.2017:1025-1035.
[26]VELIČKOVIĆ P,CUCURULL G,CASANOVA A,et al.Graph attention networks[J].arXiv:1710.10903,2017.
[27]BRODY S,ALON U,YAHAV E.How attentive are graph attention networks?[J].arXiv:2105.14491,2021.
[28]SOMVANSHI S,JAVED S A,ISLAM M M,et al.A survey on Kolmogorov-Arnold network[J].arXiv:2411.06078,2024.
[29]SU Z,CHEN M,AI J,et al.Research on Recommendation Algorithm Based on Kolmogorov-Arnold Network-driven Scoring and Review Fusion[J].Journal of Chinese Computer Systems.2025,46(11):2600-2609.
[30]CHEON M.Kolmogorov-Arnold network for satellite imageclassification in remote sensing[J].arXiv:2406.00600,2024.
[31]JIANG W,CHEN T,GAO X,et al.Epidemiology-informed network for robust rumor detection[C]//Proceedings of the ACM on Web Conference 2025.2025:3618-3627.
[32]XU K,CHEN L,WANG S.Kolmogorov-Arnold Networks for Time Series:Bridging Predictive Power and Interpretability[J].arXiv:2406.02496,2024.
[33]GU A,GOEL K,RÉ C.Efficiently modeling long sequences with structured state spaces[J].arXiv:2111.00396,2021.
[34]SMITH J T H,WARRINGTON A,LINDERMAN S W.Simplified state space layers for sequence modeling[J].arXiv:2208.04933,2022.
[35]MEHTA H,GUPTA A,CUTKOSKY A,et al.Long range language modeling via gated state spaces[J].arXiv:2206.13947,2022.
[36]QU H,NING L,AN R,et al.A survey of Mamba[J].arXiv:2408.01129,2024.
[37]PRATAPA A,JALIHAL A P,LAW J N,et al.Benchmarking algorithms for gene regulatory network inference from single-cell transcriptomic data[J].Nature Methods,2020,17(2):147-154.
[38]MOORE J E,PURCARO M J,PRATT H E,et al.Expanded encyclopaedias of DNA elements in the human and mouse genomes[J].Nature,2020,583(7818):699-710.
[39]GARCIA-ALONSO L,HOLLAND C H,IBRAHIM M M,et al.Benchmark and integration of resources for the estimation of human transcription factor activities[J].Genome Research,2019,29(8):1363-1375.
[40]SZKLARCZYK D,GABLE A L,LYON D,et al.STRING v11:protein-protein association networks with increased coverage,supporting functional discovery in genome-wide experimental datasets[J].Nucleic Acids Research,2019,47(D1):D607-D613.
[41]XU H,BAROUKH C,DANNENFELSER R,et al.ESCAPE:database for integrating high-content published data collected from human and mouse embryonic stem cells[J].Database,2013,2013:bat045.
[42]ZHU X,JING X Y,ZHANG F,et al.Distance learning by mining hard and easy negative samples for person re-identification[J].Pattern Recognition,2019,95:211-222.
[43]MAO G,PANG Z,ZUO K,et al.Predicting gene regulatorylinks from single-cell RNA-seq data using graph neural networks[J].Briefings in Bioinformatics,2023,24(6):bbad414.
[44]KC K,LI R,CUI F,et al.GNE:a deep learning framework for gene network inference by aggregating biological information[J].BMC Systems Biology,2019,13:1-14.
[45]ARISDAKESSIAN C,POIRION O,YUNITS B,et al.DeepImpute:an accurate,fast,and scalable deep neural network method to impute single-cell RNA-seq data[J].Genome Biology,2019,20:1-14.
[46]HAN H,CHO J W,LEE S,et al.TRRUST v2:an expanded ref-erence database of human and mouse transcriptional regulatory interactions[J].Nucleic Acids Research,2018,46(D1):D380-D386.
[1] ZHANG Xueqin, WANG Zhineng, LI Jinsheng, LU Yisong, LUO Fei. Key Node Identification in Temporal Social Networks Based on Deep Learning and Multi-feature Fusion [J]. Computer Science, 2026, 53(4): 143-154.
[2] GU Bokai, LIU Dun, SUN Yang. STWD-DLFRD:Multi-granularity Fake Review Detection via Sequential Three-way Decisions and Deep Learning [J]. Computer Science, 2026, 53(4): 188-196.
[3] WANG Jinghong, LI Pengchao, MI Jusheng, WANG Wei. Multi-channel Graph Kolmogorov-Arnold Network Based on WL Graph Core [J]. Computer Science, 2026, 53(4): 224-234.
[4] PENG Juhong, ZHANG Zhengyue, DING Zixu, FAN Xinyu, HU Changyu, ZHAO Mingjun. Multi-view Local Language Feature and Global Feature Fusion for Conversational Aspect-based Sentiment Quadruple Analysis [J]. Computer Science, 2026, 53(4): 384-392.
[5] ZHENG Cheng, BAN Qingqing. Knowledge-assisted and Reinforced Syntax-driven for Aspect-based Sentiment Analysis [J]. Computer Science, 2026, 53(4): 406-414.
[6] YIN Chuang, LIU Jianyi, ZHANG Ru. Cross-modal Fusion Few-sample Ransomware Classifier:Multimodal Encoding Based on Pre-trained Models [J]. Computer Science, 2026, 53(4): 435-444.
[7] WANG Jinghong, LI Pengchao, WANG Xizhao, ZHANG Zili. Dual-channel Graph Neural Network Based on KAN [J]. Computer Science, 2026, 53(3): 188-196.
[8] FU Yukai, LI Qingzhen, DONG Zhixue, SHI Dongli, ZHAO Peng. Pedestrian Re-identification Methods Based on Limited Target Data and Deep Learning [J]. Computer Science, 2026, 53(3): 287-294.
[9] YU Ding, LI Zhangwei. Prediction Method of RNA Secondary Structure Based on Transformer Architecture [J]. Computer Science, 2026, 53(3): 375-382.
[10] DU Jiantong, GUAN Zeli, XUE Zhe. Multi-task Learning-based Ophthalmic Video Feature Fusion and Multi-dimensional Profiling [J]. Computer Science, 2026, 53(3): 383-391.
[11] SU Ruitao, REN Jiongjiong, CHEN Shaozhen. Deep Learning-based Neural Differential Distinguishers for GIFT-128 and ASCON [J]. Computer Science, 2026, 53(3): 453-458.
[12] LI Zequn, DING Fei. Fatigue Driving Detection Based on Dual-branch Fusion and Segmented Domain AdaptationTransfer Learning [J]. Computer Science, 2026, 53(3): 78-87.
[13] CHANG Xuanwei, DUAN Liguo, CHEN Jiahao, CUI Juanjuan, LI Aiping. Method for Span-level Sentiment Triplet Extraction by Deeply Integrating Syntactic and Semantic
Features
[J]. Computer Science, 2026, 53(2): 322-330.
[14] XI Penghui, WU Xiazhen, JIANG Wencong, FANG Liangda, HE Chaobo, GUAN Quanlong. Review of Personalized Educational Resource Recommendations [J]. Computer Science, 2026, 53(2): 1-15.
[15] ZHAI Jie, LI Yanhao, CHEN Lexuan, GUO Weibin. Dynamic Recommendation of Personalized Hands-on Learning Materials Based on LightweightEducational LLMs [J]. Computer Science, 2026, 53(2): 48-56.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!