Edge-Aware Vessel Segmentation Using Scribble Supervision

Zhanqiang Guo,Jianjiang Feng,Jie Zhou
DOI: https://doi.org/10.1109/isbi56570.2024.10635896
2024-01-01
Abstract:Accurate segmentation of blood vessels is critical for diagnosing various diseases. However, the complexity of manually labeling vessels impedes the practical adoption of fully supervised methods. To alleviate this challenge, we propose a weakly supervised vessel segmentation framework. Our approach leverages scribble annotation to train the Unet and identifies reliable foreground and background regions. Addressing the issue of insufficient boundary information inherent in scribble annotation, we incorporate a conventional approach specifically designed to leverage the innate structural attributes of vessels for edge detection, subsequently ensuring effective edge supervision. In addition, a bilateral filtering module is introduced to improve edge awareness of network. Furthermore, to augment the quantity of annotated pixels, we employ an image mixing strategy for data augmentation, thereby enhancing the network’s segmentation capability. The experimental results on three datasets show that our framework outperforms the existing scribble-based methods.
What problem does this paper attempt to address?