Distributed Deep Learning for Multi-Label Chest Radiography Classification

Maram Mahmoud A. Monshi, Maram Mahmoud A. Monshi, Josiah Poon, Vera Chung

2022

Abstract

Chest radiography supports the clinical diagnosis and treatment for a series of thoracic diseases, such as cardiomegaly, pneumonia, and lung lesion. With the revolution of deep learning and the availability of large chest radiography datasets, binary chest radiography classifiers have been widely proposed in the literature. However, these automatic classifiers neglect label co-occurrence and inter-dependency in chest radiography and fail to make full use of accelerators, resulting in inefficient and computationally expensive models. This paper first studies the effect of chest radiography image format, variations of Dense Convolutional Network (DenseNet-121) architecture, and parallel training on chest radiography multi-label classification task. Then, we propose Xclassifier, an efficient multi-label classifier that trains an enhanced DenseNet-121 with a blur pooling framework to classify chest radiography based on fourteen predefined labels. Xclassifier accomplishes an ideal memory utilization and GPU computation and achieves 84.10% AUC on the MIMIC-CXR dataset and 83.89% AUC on the CheXpert dataset. The code used to generate the experiment results mentioned in this paper can be found here: https://github.com/MaramMonshi/Xclassifier.

Paper Citation

in Harvard Style

Monshi M., Poon J. and Chung V. (2022). Distributed Deep Learning for Multi-Label Chest Radiography Classification. In Proceedings of the 17th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022) - Volume 4: VISAPP; ISBN 978-989-758-555-5, SciTePress, pages 949-956. DOI: 10.5220/0010849400003124

in Bibtex Style

@conference{visapp22,
author={Maram Mahmoud A. Monshi and Josiah Poon and Vera Chung},
title={Distributed Deep Learning for Multi-Label Chest Radiography Classification},
booktitle={Proceedings of the 17th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022) - Volume 4: VISAPP},
year={2022},
pages={949-956},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0010849400003124},
isbn={978-989-758-555-5},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 17th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022) - Volume 4: VISAPP
TI - Distributed Deep Learning for Multi-Label Chest Radiography Classification
SN - 978-989-758-555-5
AU - Monshi M.
AU - Poon J.
AU - Chung V.
PY - 2022
SP - 949
EP - 956
DO - 10.5220/0010849400003124
PB - SciTePress