CHDNet: A lightweight weakly supervised segmentation network for lung CT image
Fangfang Lu,Tianxiang Liu,Ting Zhang,Bei Jin,Weiyan Gu
DOI: https://doi.org/10.1016/j.displa.2024.102650
IF: 3.074
2024-04-01
Displays
Abstract:Deep learning methods have ushered in an unprecedented transformation in medical image segmentation by automating the segmentation of computed tomography (CT) slices. However, challenges persist in the application of these deep learning methods, including models with a high number of training parameters which hinder their clinical deployment and practical use. Furthermore, acquiring a large volume of high-quality labels from radiologists for training is prohibitively expensive. In light of these challenges, this paper introduces a weakly supervised method aimed at addressing the scarcity of high-quality labels by annotating only a small subset of pixels within regions of interest. This strategy expedites the labeling process, requiring just a few seconds to complete. Additionally, we present a lightweight segmentation model designed to minimize computational complexity. By incorporating parallel dilated convolutions and depthwise separable convolutions, this model consists of 11.03M parameters and can segment lesions in lung CT slices in just 0.05 s, showcasing high segmentation efficiency. To capture shape and size information from point labels, we propose a Convex-Hull-based Segmentation (CHS) loss function. This loss function empowers our weakly supervised model to approximate the performance of a fully supervised model when evaluated against point labels. We conduct a comprehensive evaluation, comparing our proposed method with other state-of-the-art methods using multiple metrics on public datasets. The experimental results demonstrate a significant improvement in performance compared to current state-of-the-art (SOTA) point label segmentation methods, with an increase of 19.08% in Intersection over Union (IoU), 20.54% in Dice Similarity Coefficient (DSC), 2.60% in sensitivity, and 1.33% in specificity. In comparison to the SOTA fully supervised methods, our approach exhibits a minimal IoU difference of just 2.1%. Further ablation experiments validate the effectiveness of the key methodologies introduced in this work.
engineering, electrical & electronic,instruments & instrumentation,optics,computer science, hardware & architecture