Slicerfeature Learning for Class Separability with Least-Squares Support Vector Machine Loss and COVID-19 Chest X-Ray Case Study
- David Charte 1
- Iván Sevillano-García 1
- María Jesús Lucena-González 2
- José Luis Martín-Rodríıguez 2
- Francisco Charte 3
- Francisco Herrera 1
-
1
Universidad de Granada
info
-
2
Hospital Universitario San Cecilio
info
Hospital Universitario San Cecilio
Granada, España
-
3
Universidad de Jaén
info
- Hugo Sanjurjo González (coord.)
- Iker Pastor López (coord.)
- Pablo García Bringas (coord.)
- Héctor Quintián (coord.)
- Emilio Corchado (coord.)
Publisher: Springer International Publishing AG
ISBN: 978-3-030-86271-8, 978-3-030-86270-1
Year of publication: 2021
Pages: 305-315
Congress: Hybrid Artificial Intelligent Systems (HAIS) (16. 2021. Bilbao)
Type: Conference paper
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.