Deep Learning Based Multi-Label Chest X-Ray Classification with Entropy Weighting Loss

Shaocong Mo,Ming Cai
DOI: https://doi.org/10.1109/iscid.2019.10111
2019-01-01
Abstract:With large chest X-ray image datasets open-accessed, it is feasible to build diagnosis system based on deep learning methods. In this work, we formulated chest Xray diagnosis as multi-label classification problem. However, multi-labels are treated independently in traditional methods and existing loss can easily lose important information for class with fewer cases. We proposed an entropy weighting loss to observe inter-label dependencies and make full use of class whose cases are fewer than others. From the global perspective of all pathological classes, digging relations between all classes, the proposed loss could improve model performance. We experimented with three basic deep learning models (VGG16, ResNet50, DenseNet121) at first, and decided to use DenseNet121 as a baseline model. We evaluated our approach and achieved better results, whose AUC score is 0.8430 on average, under the Chest X-ray14 and its patient-wise split. We demonstrate that our proposed loss can dig relations between different disease labels just as illness complications in reality, and we do not consider every label is independent.
What problem does this paper attempt to address?