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

  1. David Charte 1
  2. Iván Sevillano-García 1
  3. María Jesús Lucena-González 2
  4. José Luis Martín-Rodríıguez 2
  5. Francisco Charte 3
  6. Francisco Herrera 1
  1. 1 Universidad de Granada
    info

    Universidad de Granada

    Granada, España

    ROR https://ror.org/04njjy449

  2. 2 Hospital Universitario San Cecilio
    info

    Hospital Universitario San Cecilio

    Granada, España

  3. 3 Universidad de Jaén
    info

    Universidad de Jaén

    Jaén, España

    ROR https://ror.org/0122p5f64

Libro:
Hybrid Artificial Intelligent Systems: 16th International Conference, HAIS 2021. Bilbao, Spain. September 22–24, 2021. Proceedings
  1. Hugo Sanjurjo González (coord.)
  2. Iker Pastor López (coord.)
  3. Pablo García Bringas (coord.)
  4. Héctor Quintián (coord.)
  5. Emilio Corchado (coord.)

Editorial: Springer International Publishing AG

ISBN: 978-3-030-86271-8 978-3-030-86270-1

Año de publicación: 2021

Páginas: 305-315

Congreso: Hybrid Artificial Intelligent Systems (HAIS) (16. 2021. Bilbao)

Tipo: Aportación congreso

Resumen

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.