Chest X-Rays Image Classification from Variational Autoencoders Latent Features

L Crespi, D Loiacono, A Chiti - 2021 IEEE Symposium Series …, 2021 - ieeexplore.ieee.org
2021 IEEE Symposium Series on Computational Intelligence (SSCI), 2021ieeexplore.ieee.org
Chest X-Ray (CXR) is one of the most common diagnostic imaging used in everyday clinical
practice. In this work, we present a Deep Learning (DL) approach to extract from CXR
images a set of features that would capture as much information as possible. In particular,
our aim is to extract highly general features that could be successfully used for a large
variety of real-world classification tasks. Accordingly, we trained several β-, Variational
Autoencoder (β- VAE) models on CheXpert, a popular dataset consisting in a very broad …
Chest X-Ray (CXR) is one of the most common diagnostic imaging used in everyday clinical practice. In this work, we present a Deep Learning (DL) approach to extract from CXR images a set of features that would capture as much information as possible. In particular, our aim is to extract highly general features that could be successfully used for a large variety of real-world classification tasks. Accordingly, we trained several , Variational Autoencoder VAE) models on CheXpert, a popular dataset consisting in a very broad and publicly available collection of labeled CXR images; through these models, high level features have been extracted and used to train Machine Learning (ML) classifier (Random Forest, K-Nearest Neighbours, Extremely Randomised Trees, Gradient Boosting), to classify CXR images thanks to the information extracted from the - VAEs; finally, trained classifiers have been combined in ensembles to improve the performances without the need of further training or models engineering. Despite, as expected, our approach does not achieve the same performance of the state-of-the-art models specifically devised for this classification task, our results are promising and show the viability of using the high level features extracted by the Autoencoders for classification tasks.
ieeexplore.ieee.org