Slicer: Feature Learning for Class Separability with Least-Squares Support Vector Machine Loss and COVID-19 Chest X-Ray Case Study

TitleSlicer: Feature Learning for Class Separability with Least-Squares Support Vector Machine Loss and COVID-19 Chest X-Ray Case Study
Publication TypeConference Paper
Year of Publication2021
AuthorsCharte, David, Sevillano-García Iván, Lucena-González María Jesús, Martín-Rodríguez José Luis, Charte Francisco, and Herrera Francisco
EditorGonzález, Hugo Sanjurjo, López Iker Pastor, Bringas Pablo García, Quintián Héctor, and Corchado Emilio
Conference NameHybrid Artificial Intelligent Systems
Pagination305–315
PublisherSpringer International Publishing
Conference LocationCham
ISBN Number978-3-030-86271-8
Abstract

Datasets from real-world applications usually deal with many variables and present difficulties when modeling them with traditional classifiers. There is a variety of feature selection and extraction tools that may help with the dimensionality problem, but most of them do not focus on the complexity of the classes. In this paper, a new autoencoder-based model for addressing class complexity in data is introduced, aiming to extract features that present classes in a more separable fashion, thus simplifying the classification task. This is possible thanks to a combination of the standard reconstruction error with a least-squares support vector machine loss function. This model is then applied to a practical use case: classification of chest X-rays according to the presence of COVID-19, showing that learning features that increase linear class separability can boost classification performance. For this purpose, a specific convolutional autoencoder architecture has been designed and trained using the recently published COVIDGR dataset. The proposed model is evaluated by means of several traditional classifiers and metrics, in order to establish the improvements caused by the extracted features. The advantages of using a feature learner and traditional classifiers are also discussed.

Notes

PID2020-119478GB-I00