Chest X-Ray Analysis of Tuberculosis by Deep Learning with Segmentation and Augmentation
This work addresses computer-aided diagnosis of tuberculosis for medical imaging, but it is incremental as it focuses on preprocessing techniques for small datasets.
The study tackled tuberculosis detection from chest X-rays using deep learning with segmentation and augmentation on a small, imbalanced dataset, finding that lung segmentation enabled training despite overfitting and lossless augmentation achieved the lowest validation loss with similar accuracy.
The results of chest X-ray (CXR) analysis of 2D images to get the statistically reliable predictions (availability of tuberculosis) by computer-aided diagnosis (CADx) on the basis of deep learning are presented. They demonstrate the efficiency of lung segmentation, lossless and lossy data augmentation for CADx of tuberculosis by deep convolutional neural network (CNN) applied to the small and not well-balanced dataset even. CNN demonstrates ability to train (despite overfitting) on the pre-processed dataset obtained after lung segmentation in contrast to the original not-segmented dataset. Lossless data augmentation of the segmented dataset leads to the lowest validation loss (without overfitting) and nearly the same accuracy (within the limits of standard deviation) in comparison to the original and other pre-processed datasets after lossy data augmentation. The additional limited lossy data augmentation results in the lower validation loss, but with a decrease of the validation accuracy. In conclusion, besides the more complex deep CNNs and bigger datasets, the better progress of CADx for the small and not well-balanced datasets even could be obtained by better segmentation, data augmentation, dataset stratification, and exclusion of non-evident outliers.