Kernel Support Vector Machines and Convolutional Neural Networks

Shihao Jiang, Richard Hartley, Basura Fernando

    Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

    8 Citations (Scopus)

    Abstract

    Convolutional Neural Networks (CNN) have achieved great success in various computer vision tasks due to their strong ability in feature extraction. The trend of development of CNN architectures is to increase their depth so as to increase their feature extraction ability. Kernel Support Vector Machines (SVM), on the other hand, are known to give optimal separating surfaces by their ability to automatically select support vectors and perform classification in higher dimensional spaces. We investigate the idea of combining the two such that best of both worlds can be achieved and a more compact model can perform as well as deeper CNNs. In the past, attempts have been made to use CNNs to extract features from images and then classify with a kernel SVM, but this process was performed in two separate steps. In this paper, we propose one single model where a CNN and a kernel SVM are integrated together and can be trained end-to-end. In particular, we propose a fully-differentiable Radial Basis Function (RBF) layer, where it can be seamless adapted to a CNN environment and forms a better classifier compared to the normal linear classifier. Due to end-to-end training, our approach allows the initial layers of the CNN to extract features more adapted to the kernel SVM classifier. Our experiments demonstrate that the hybrid CNN-kSVM model gives superior results to a plain CNN model, and also performs better than the method where feature extraction and classification are performed in separate stages, by a CNN and a kernel SVM respectively.

    Original languageEnglish
    Title of host publication2018 International Conference on Digital Image Computing
    Subtitle of host publicationTechniques and Applications, DICTA 2018
    EditorsMark Pickering, Lihong Zheng, Shaodi You, Ashfaqur Rahman, Manzur Murshed, Md Asikuzzaman, Ambarish Natu, Antonio Robles-Kelly, Manoranjan Paul
    PublisherInstitute of Electrical and Electronics Engineers Inc.
    ISBN (Electronic)9781538666029
    DOIs
    Publication statusPublished - 16 Jan 2019
    Event2018 International Conference on Digital Image Computing: Techniques and Applications, DICTA 2018 - Canberra, Australia
    Duration: 10 Dec 201813 Dec 2018

    Publication series

    Name2018 International Conference on Digital Image Computing: Techniques and Applications, DICTA 2018

    Conference

    Conference2018 International Conference on Digital Image Computing: Techniques and Applications, DICTA 2018
    Country/TerritoryAustralia
    CityCanberra
    Period10/12/1813/12/18

    Fingerprint

    Dive into the research topics of 'Kernel Support Vector Machines and Convolutional Neural Networks'. Together they form a unique fingerprint.

    Cite this