Chest X-Rays Image Classification from beta-Variational Autoencoders Latent Features
Not on the map. Why?
Abstract (arXiv)
Chest X-Ray (CXR) is one of the most common diagnostic techniques used in everyday clinical practice all around the world. We hereby present a work which intends to investigate and analyse the use of Deep Learning (DL) techniques to extract information from such images and allow to classify them, trying to keep our methodology as general as possible and possibly also usable in a real world scenario without much effort, in the future. To move in this direction, we trained several beta-Variational Autoencoder (beta-VAE) models on the CheXpert dataset, one of the largest publicly available collection of labeled CXR images; from these models, latent features have been extracted and used to train other Machine Learning models, able to classify the original images from the features extracted by the beta-VAE. Lastly, tree-based models have been combined together in ensemblings to improve the results without the necessity of further training or models engineering. Expecting some drop in pure performance with the respect to state of the art classification specific models, we obtained encouraging results, which show the viability of our approach and the usability of the high level features extracted by the autoencoders for classification tasks.
Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 16 of its 34 references that are in the map. First posted on arXiv 29 Sep 2021.
Builds on
16 of 34 references.
References · 16
- 2014Adam: A Method for Stochastic Optimizationcited by 37,093
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,321
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,713
- 2016Densely Connected Convolutional Networkscited by 6,707
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,457
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,326
- 2016Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learningcited by 1,512
- 2019CheXpert: A Large Chest Radiograph Dataset with Uncertainty Labels and Expert Comparisoncited by 1,419
- 2019MIMIC-CXR-JPG, a large publicly available database of labeled chest radiographscited by 593
- 2017CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learningcited by 460
- 2020Interpreting Chest X-rays via CNNs that Exploit Hierarchical Disease Dependencies and Uncertainty Labelscited by 56
- 2020Weakly Supervised Lesion Localization With Probabilistic-CAM Poolingcited by 36
4 more
- 2018Large Scale Automated Reading of Frontal and Lateral Chest X-Rays using Dual Convolutional Neural Networkscited by 20
- 2017Boosted Cascaded Convnets for Multilabel Classification of Thoracic Diseases in Chest Radiographscited by 18
- 2019Interpreting chest X-rays via CNNs that exploit hierarchical disease dependencies and uncertainty labelscited by 8
- 2021Image Embedding and Model Ensembling for Automated Chest X-Ray Interpretationcited by 2
Built on by
Not cited yet.