SNU-Net: a Self-Supervised Deep Learning Method for Pneumothorax Segmentation on Chest CT

Bo Xue,Zhiqin Liu,Qingfeng Wang,Qin Tang,Jun Huang,Ying Zhou
DOI: https://doi.org/10.1109/iscas48785.2022.9937654
2022-01-01
Abstract:Pneumothorax(PTX) is an emergency condition with lung collapse, causing breathing difficulties and life-threatening. It is necessary to segment PTX to help clinicians make decisions and guide subsequent treatment. General deep learning often requires a large dataset to improve the generalization ability of the model. However, it is arduous to obtain Computer Tomography (CT) PTX data with annotations. Automatic segmentation of CT-PTX has the problem of small samples, which cause unsatisfactory effect of existing methods. To against those problems, we proposed a self-supervised segmentation model for CT-PTX named SNU-Net (self-supervised nnU-net), which divided the segmentation task into pretext task and downstream task. In our proposed PNE dataset, segmentation accuracy was as high as 99.94%, and the dice coefficient reached 89.02%. Compared with the original nnU-net, our network only needs a few annotated data to achieve a better result.
What problem does this paper attempt to address?