AutoGRN: An Automated Graph Neural Network Framework for Gene Regulatory Network Inference

Gene regulatory network (GRN) inference is essential for understanding gene interactions that control biological functions and disease progression. Its goal is to uncover regulatory relationships among genes using gene expression data, such as single-cell RNA sequencing (scRNA-seq). With the rise of deep learning, numerous approaches—particularly those using Graph Neural Networks (GNNs)—have shown promising progress in GRN inference. However, scRNA-seq datasets exhibit high sparsity, noise, dropout events, and cellular heterogeneity. These characteristics make it difficult for fixed GNN architectures to generalize across scRNA-seq datasets, often leading to performance degradation. To address this challenge, we propose AutoGRN, an automated graph neural network framework for gene regulatory network inference tasks. AutoGRN incorporates a range of components that potentially influence GRN inference performance into its search space. Through a genetic search algorithm constrained by information entropy, AutoGRN identifies optimal GNN architectures by adapting to the characteristics of individual datasets. Experimental results show that AutoGRN outperforms existing methods in prediction accuracy across multiple public datasets.