Deep graph regularized learning for binary classification

Research output: Contribution to conferencePaper

56 Downloads (Pure)

Abstract

With growing interest in data-driven classification, deep learning is now prevalent thanks to its ability to learn feature mapping functions solely from data. For very small training sets, however, deep learning, even with traditional regularization techniques, often overfits, resulting in sub-par classification performance. In this paper, we propose a novel binary classifier deep learning method, based on an iterative quadratic programming (QP) formulation with a graph Laplacian regularizer (GLR), combining the merits of model-based and data-driven approaches. Specifically, the proposed network employs a convolutional neural network (CNN) to learn deep features, which are used to define edge weights for a graph to pose a convex QP problem. Further, we design a novel loss function to penalize samples at the class boundary during semi-supervised learning. Results demonstrate that given a small-size training dataset, our network outperforms several state-of-the-art classifiers, including CNN, model-based GLR, and dynamic graph CNN classifiers.
Original languageEnglish
Pages3537-3541
Number of pages5
DOIs
Publication statusPublished - 12 May 2019
Event2019 International Conference on Acoustics, Speech, and Signal Processing - Brighton, United Kingdom
Duration: 12 May 201917 May 2019

Conference

Conference2019 International Conference on Acoustics, Speech, and Signal Processing
Abbreviated titleICASSP 2019
CountryUnited Kingdom
CityBrighton
Period12/05/1917/05/19

    Fingerprint

Keywords

  • graph Laplacian regularization
  • binary classification
  • semi-supervised learning,
  • deep learning

Cite this

Ye, M., Stankovic, V., Stankovic, L., & Cheung, G. (2019). Deep graph regularized learning for binary classification. 3537-3541. Paper presented at 2019 International Conference on Acoustics, Speech, and Signal Processing, Brighton, United Kingdom. https://doi.org/10.1109/ICASSP.2019.8682725