Automatic Segmentation of Pneumothorax in Chest Radiographs Based on Dual-Task Interactive Learning Method

Zhen-tai Lu,Yaping Wang
DOI: https://doi.org/10.1145/3591156.3591163
2023-03-24
Abstract:Pneumothorax is one of the common diseases of the lungs and can be fatal in severe cases. With the development of deep learning, the convolutional neural network-based methods have been well used in the field of medical images. However, the chest X-ray images themselves are very complex, it is difficult for current algorithms to achieve pneumothorax segmentation well. Therefore, to address the problem that current deep learning-based segmentation methods cannot meet the demand of existing pneumothorax image segmentation tasks, this paper presents a novel spatially adaptive weighted fusion Signed Distance Map (SAWFSDM) network segmentation model. The proposed model utilizes the rich boundary information in the Signed Distance Map (SDM) to improve the segmentation results by learning the binary mask and SDM symbolic distance map in parallel through a multi-task learning strategy, which provides more supervised information to the training of the model through multiple labels. In addition, the boundary information learned from the SDM branch is fused into the binary mask branch during the decoding process, which strengthens the ability of model feature extraction and makes the model learn more accurate contour shapes. The segmentation model proposed in this paper has been verified by a large number of experiments on the SIIM-ACR pneumothorax segmentation dataset, and the results show that the SAWFSDM model can effectively sense the boundaries of the pneumothorax and improve the accuracy of pneumothorax segmentation, which has significant advantages compared with other advanced pneumothorax segmentation methods.
Medicine,Engineering,Computer Science
What problem does this paper attempt to address?