IJMLC 2018 Vol.8(4): 298-310 ISSN: 2010-3700
DOI: 10.18178/ijmlc.2018.8.4.703

An Improvement of Data Classification Using Random Multimodel Deep Learning (RMDL)

Mojtaba Heidarysafa, Kamran Kowsari, Donald E. Brown, Kiana Jafari Meimandi, and Laura E. Barnes

Abstract—The exponential growth in the number of complex datasets every year requires more enhancement in machine learning methods to provide robust and accurate data classification. Lately, deep learning approaches have achieved surpassing results in comparison to previous machine learning algorithms. However, finding the suitable structure for these models has been a challenge for researchers. This paper introduces Random Multimodel Deep Learning (RMDL): a new ensemble, deep learning approach for classification. RMDL solves the problem of finding the best deep learning structure and architecture while simultaneously improving robustness and accuracy through ensembles of deep learning architectures. In short, RMDL trains multiple randomly generated models of Deep Neural Network (DNN), Convolutional Neural Network (CNN) and Recurrent Neural Network (RNN) in parallel and combines their results to produce better result of any of those models individually. In this paper, we describe RMDL model and compare the results for image and text classification as well as face recognition. We used MNIST and CIFAR-10 datasets as ground truth datasets for image classification and WOS, Reuters, IMDB, and 20newsgroup datasets for text classification. Lastly, we used ORL dataset to compare the model performance on face recognition task.

Index Terms—Deep neural networks, document classification, hierarchical learning, multimodel deep learning.

The authors are with the Department of System and Information Engineering, University of Virginia, Charlottesville, VA 22911 USA (e-mail: {kk7nc, mh4pk, deb, kj6vd, lb3dp}@virginia.edu).

[PDF]

Cite: Mojtaba Heidarysafa, Kamran Kowsari, Donald E. Brown, Kiana Jafari Meimandi, and Laura E. Barnes, "An Improvement of Data Classification Using Random Multimodel Deep Learning (RMDL)," International Journal of Machine Learning and Computing vol. 8, no. 4, pp. 298-310, 2018.

General Information

  • ISSN: 2010-3700 (Online)
  • Abbreviated Title: Int. J. Mach. Learn. Comput.
  • Frequency: Bimonthly
  • DOI: 10.18178/IJMLC
  • Editor-in-Chief: Dr. Lin Huang
  • Executive Editor:  Ms. Cherry L. Chen
  • Abstracing/Indexing: Scopus (since 2017), EI (INSPEC, IET), Google Scholar, Crossref, ProQuest, Electronic Journals Library.
  • E-mail: ijmlc@ejournal.net