Making Multi-Axis Gaussian Graphical Models Scalable to Millions of Cells

TL;DR

This paper introduces a scalable multi-axis Gaussian graphical model method capable of analyzing datasets with millions of cells, enabling new biological insights and outperforming existing methods in speed and accuracy.

Contribution

We develop a novel multi-axis Gaussian graphical model approach that scales to millions of cells, overcoming previous computational limitations and facilitating large-scale biological data analysis.

Findings

Processed million-cell datasets within minutes

Identified novel long non-coding RNAs involved in neuronal development

Outperformed existing hdWGCNA methodology in biological insights

Abstract

Motivation: Networks underlie the generation and interpretation of many biological datasets: gene networks shed light on the regulatory structure of the genome, and cell networks can capture structure of the tumor micro-environment. However, most methods that learn such networks make the faulty 'independence assumption'; to learn the gene network, they assume that no cell network exists. 'Multi-axis' methods, which do not make this assumption, fail to scale beyond a few thousand cells or genes. This limits their applicability to only the smallest datasets. Results: We develop a multi-axis method capable of processing million-cell datasets within minutes. This was previously impossible, and unlocks the use of such methods on modern scRNA-seq datasets, as well as more complex datasets. We show that our method yields novel biological insights from real single-cell data, and compares favorably to the existing hdWGCNA methodology. In particular, it identifies long non-coding RNA genes that potentially have a regulatory or functional role in neuronal development. Availability and implementation: Our methodology is available as a Python package GmGM on PyPI (https://pypi.org/project/GmGM/0.5.3/). The code for all experiments performed in this paper is available on GitHub (https://github.com/BaileyAndrew/GmGM-Bioinformatics). Contact: sceba@leeds.ac.uk Supplementary information: Our proofs, and some additional experiments, are available in the supplementary material. Keywords: gaussian graphical models, multi-axis models, transcriptomics, multi-omics, scalability