Home > Archive > 2018 > Volume 8 Number 3 (Jun. 2018) >
IJMLC 2018 Vol.8(3): 241-251 ISSN: 2010-3700
DOI: 10.18178/ijmlc.2018.8.3.694

Deep Unfolded IRLS-ADMM Network for Classification and Sparse Feature Selection

Xian Yang and Yike Guo

Abstract—Two main branches of machine learning methods are model-based methods and deep neural networks. Model-based methods can explicitly include prior knowledge into the model at expense of difficulties in inference, while neural networks are featured in their strong predictive power and straightforward inference approach with the lack of model interpretability. To construct models which are entitled with the advantages of both methods and overcome their problems, the deep unfolding strategy has been developed recently. This paper adopts the idea of deep unfolding to construct a classification and feature selection method. The proposed method is based on the sparse classification; and the iterative inference process of the sparse classification is unfolded into a layer-wise structure analogous to a neural network. Thus, the architecture of our network is fully motivated by the sparse classification method. Different from other neural networks which are essentially black-box methods, our deep unfolded network acts as white-box that features selected in the predictive model can be explicitly returned. Experimental results show the both predictive power and feature selection ability of our methods.

Index Terms—Deep unfolded neural network, IRLS-ADMM net, sparse classification, feature selection, white-box method.

The authors are with the Data Science Institute, Imperial College London, SW7 2AZ, UK (e-mail: xian.yang08@impeial.ac.uk, y.guo@impeial.ac.uk).


Cite: Xian Yang and Yike Guo, "Deep Unfolded IRLS-ADMM Network for Classification and Sparse Feature Selection," International Journal of Machine Learning and Computing vol. 8, no. 3, pp. 241-251, 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: Inspec (IET), Google Scholar, Crossref, ProQuest, Electronic Journals LibraryCNKI.
  • E-mail: ijmlc@ejournal.net

Article Metrics in Dimensions