NMFClustering: Accessible NMF-based clustering utilizing GPU acceleration

Abstract

AbstractSummaryNon-negative Matrix Factorization (NMF) is an algorithm that can reduce high dimensional datasets of tens of thousands of genes to a handful of metagenes which are biologically easier to interpret. Application of NMF on gene expression data has been limited by its computationally intensive nature, which hinders its use on large datasets such as single-cell RNA sequencing (scRNA-seq) count matrices. We have implemented NMF based clustering to run on high performance GPU compute nodes using CuPy, a GPU backed python library, and the Message Passing Interface (MPI). This reduces the computation time by up to three orders of magnitude and makes the NMF Clustering analysis of large RNA-Seq and scRNA-seq datasets practical. We have made the method freely available through the GenePattern gateway, which provides free public access to hundreds of tools for the analysis and visualization of multiple ‘omic data types. Its web-based interface gives easy access to these tools and allows the creation of multi-step analysis pipelines on high performance computing (HPC) clusters that enable reproduciblein silicoresearch for non-programmers.Availability and ImplementationNMFClustering is freely available on the public GenePattern server athttps://genepattern.ucsd.edu. Code for the NMFClustering is available under a BSD style license on github athttps://github.com/genepattern/nmf-gpu.ContactTed Liefeld,jliefeld@cloud.ucsd.eduSupplementary InformationSupplementary data are available at Bioinformatics online and athttps://datasets.genepattern.org/?prefix=data/test_data/NMFClustering/.