Deep Learning Approaches for Bone Marrow Edema Detection and Interpretation in Dual-Energy CT

Chunsu Park,Jeong-Woon Kang,Dong-Eon Lee,Wookon Son,Sang Min Lee,Chankue Park,MinWoo Kim
DOI: https://doi.org/10.2139/ssrn.4189440
2022-01-01
SSRN Electronic Journal
Abstract:Dual-energy computed tomography can serve as an excellent substitute for magnetic resonance imaging to identify bone marrow edema, but it has been rarely used in practice owing to issues with low contrast. To overcome these problems, we constructed a framework based on deep learning techniques to screen the disease from axial bone images and identify the local positions of bone lesions. To cope with the scarcity of labeled samples, we developed a generative adversarial networks (GAN). We combined the ideas of data augmentation optimized for GAN and Wasserstein GAN to stably generate synthetic images. A classification model was trained on the real and synthetic samples. As a result, its F1-score was improved by around 2.5% due to the GAN technique in diagnostic tests. The best F1-score achieved was 79.4%, where the recall and precision were 81.2% and 77.7%. In addition, the framework was designed to suggest attention maps using explainable AI techniques to visually explain the disease predictions of the network.
What problem does this paper attempt to address?