Segmentation of Intravascular Ultrasound by Deep Learning on Partially Labelled Datasets

YU Wei,WU Peng,TU Shengxian
2023-01-01
Abstract:Typically,neural networks are trained on separate datasets in medical image segmentation tasks due to the difference between each individual task.However,training with shared information can improve the performance of each task and save inference time using one network.This paper aims to propose a partially supervised semantic segmentation method to segment artery structures on intravascular ultrasound(IVUS)using a union of two datasets.Multi-label semantic segmentation was used on the final partially labelled dataset to tackle the annotation inconsistency,and an asymmetric dual-branch network was proposed with class-rebalancing strategy to improve the segmentation performance.The method achieves better performance than the fully supervised methods.Strong correlation and good agreement of lumen area(r=0.99;P<0.001),media-adventitia area(r=0.99;P<0.001),and plaque burden(r=0.95;P<0.001)between the ground truth and the prediction are observed.The proposed method effectively utilizes a partially labelled IVUS dataset with annotation inconsistency and alleviates the problem of"data hunger"in IVUS segmentation.
What problem does this paper attempt to address?