22/11/2021

Stabilized Semi-Supervised Training for COVID Lesion Segmentation

Pranjal Sahu, Saikiran Kumar Vunnava, HONG QIN

Keywords: covid, ct, segmentation, semi-supervised, noisy training, lesion segmentation, label propagation

Abstract: We propose a novel stabilized semi-supervised training method to solve the chal-lenging problem of covid lesion segmentation in CT scans. We first study the limita-tions of current models and based on our findings we introduce a lightweight SU-Net(Small U-Net) architecture. During training we feed the CT scans in sorted order oflesion occupancy and calculate a reliability score at each epoch to determine the stop-ping criteria. We test the proposed method on the largest publicly available COVID CTdataset called MOSMED dataset. By harnessing around 800 un-labelled COVID CTvolumes comprising 25k CT slices, we improve the segmentation accuracy by around2-4 dice percentage points depending upon the availability of labelled training data. We also compare our method with a recently published COVID lesion segmentation methodcalled Semi-InfNet. The proposed method outperforms Semi-InfNet model and achievesstate-of-the-art covid segmentation result on MOSMED dataset.

 0
 0
 0
 0
This is an embedded video. Talk and the respective paper are published at BMVC 2021 virtual conference. If you are one of the authors of the paper and want to manage your upload, see the question "My papertalk has been externally embedded..." in the FAQ section.

Comments

Post Comment
no comments yet
code of conduct: tbd

Similar Papers