Sparse evolutionary deep learning with over one million artificial neurons on commodity hardware

Shiwei Liu*, Decebal Constantin Mocanu, Amarsagar Reddy Ramapuram Matavalam, Yulong Pei, Mykola Pechenizkiy

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

13 Citations (Scopus)
60 Downloads (Pure)

Abstract

Artificial neural networks (ANNs) have emerged as hot topics in the research community. Despite the success of ANNs, it is challenging to train and deploy modern ANNs on commodity hardware due to the ever-increasing model size and the unprecedented growth in the data volumes. Particularly for microarray data, the very high dimensionality and the small number of samples make it difficult for machine learning techniques to handle. Furthermore, specialized hardware such as graphics processing unit (GPU) is expensive. Sparse neural networks are the leading approaches to address these challenges. However, off-the-shelf sparsity-inducing techniques either operate from a pretrained model or enforce the sparse structure via binary masks. The training efficiency of sparse neural networks cannot be obtained practically. In this paper, we introduce a technique allowing us to train truly sparse neural networks with fixed parameter count throughout training. Our experimental results demonstrate that our method can be applied directly to handle high-dimensional data, while achieving higher accuracy than the traditional two-phase approaches. Moreover, we have been able to create truly sparse multilayer perceptron models with over one million neurons and to train them on a typical laptop without GPU (https://github.com/dcmocanu/sparse-evolutionary-artificial-neural-networks/tree/master/SET-MLP-Sparse-Python-Data-Structures), this being way beyond what is possible with any state-of-the-art technique.

Original languageEnglish
Pages (from-to)2589-2604
Number of pages16
JournalNeural Computing and Applications
Volume33
Issue number7
Early online date6 Jul 2020
DOIs
Publication statusPublished - 1 Apr 2021

Keywords

  • Adaptive sparse connectivity
  • Microarray gene expression
  • Sparse evolutionary training (SET)
  • Truly sparse neural networks

Fingerprint

Dive into the research topics of 'Sparse evolutionary deep learning with over one million artificial neurons on commodity hardware'. Together they form a unique fingerprint.

Cite this