HPS-Net: Multi-Task Network for Medical Image Segmentation with Predictable Performance

被引:1
|
作者
Wei, Xin [1 ]
Wan, Huan [2 ]
Ye, Fanghua [1 ]
Min, Weidong [1 ]
机构
[1] Nanchang Univ, Sch Software, 235 East Nanjing Rd, Nanchang 330047, Jiangxi, Peoples R China
[2] Jiangxi Normal Univ, Sch Comp Informat Engn, 99 Ziyang Ave, Nanchang 330022, Jiangxi, Peoples R China
来源
SYMMETRY-BASEL | 2021年 / 13卷 / 11期
基金
中国国家自然科学基金;
关键词
symmetrical structure; medical image; image segmentation; deep learning; CNNs; loss function; TRIANGLE INEQUALITY;
D O I
10.3390/sym13112107
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
In recent years, medical image segmentation (MIS) has made a huge breakthrough due to the success of deep learning. However, the existing MIS algorithms still suffer from two types of uncertainties: (1) the uncertainty of the plausible segmentation hypotheses and (2) the uncertainty of segmentation performance. These two types of uncertainties affect the effectiveness of the MIS algorithm and then affect the reliability of medical diagnosis. Many studies have been done on the former but ignore the latter. Therefore, we proposed the hierarchical predictable segmentation network (HPS-Net), which consists of a new network structure, a new loss function, and a cooperative training mode. According to our knowledge, HPS-Net is the first network in the MIS area that can generate both the diverse segmentation hypotheses to avoid the uncertainty of the plausible segmentation hypotheses and the measure predictions about these hypotheses to avoid the uncertainty of segmentation performance. Extensive experiments were conducted on the LIDC-IDRI dataset and the ISIC2018 dataset. The results show that HPS-Net has the highest Dice score compared with the benchmark methods, which means it has the best segmentation performance. The results also confirmed that the proposed HPS-Net can effectively predict TNR and TPR.
引用
收藏
页数:12
相关论文
共 50 条
  • [41] Multi-Task Learning U-Net for Functional Shoulder Sub-Task Segmentation
    Chu, En-Ping
    Liu, Kai-Chun
    Hsieh, Chia-Yeh
    Chang, Chih-Ya
    Tsao, Yu
    Chan, Chia-Tai
    2023 45TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY, EMBC, 2023,
  • [42] MMSeg: A novel multi-task learning framework for class imbalance and label scarcity in medical image segmentation
    Yang, Feiyang
    Li, Xiongfei
    Wang, Bo
    Zhang, Tianyang
    Yu, Xinyi
    Yi, Xingcheng
    Zhu, Rui
    KNOWLEDGE-BASED SYSTEMS, 2025, 309
  • [43] MULTI-TASK ATTENTIONAL U-NET FOR HYPERSPECTRAL IMAGE DENOISING
    Xiong, Fengchao
    Gu, Zhongyi
    Zheng, Wenbin
    Li, Tianhan
    Zhou, Jun
    IGARSS 2023 - 2023 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2023, : 7336 - 7339
  • [44] LESION ATTRIBUTES SEGMENTATION FOR MELANOMA DETECTION WITH MULTI-TASK U-NET
    Chen, Eric Z.
    Dong, Xu
    Li, Xiaoxiao
    Jiang, Hongda
    Rong, Ruichen
    Wu, Junyan
    2019 IEEE 16TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI 2019), 2019, : 485 - 488
  • [45] Lesion attributes segmentation for melanoma detection with multi-task u-net
    Chen, Eric Z.
    Dong, Xu
    Li, Xiaoxiao
    Jiang, Hongda
    Rong, Ruichen
    Wu, Junyan
    Proceedings - International Symposium on Biomedical Imaging, 2019, 2019-April : 485 - 488
  • [46] Multi-task fully convolutional networks for building segmentation on SAR image
    Zhang, Zenghui
    Guo, Weiwei
    Yu, Wenhao
    Yu, Wenxian
    JOURNAL OF ENGINEERING-JOE, 2019, 2019 (20): : 7074 - 7077
  • [47] Multi-task contrastive learning for semi-supervised medical image segmentation with multi-scale uncertainty estimation
    Xing, Chengcheng
    Dong, Haoji
    Xi, Heran
    Ma, Jiquan
    Zhu, Jinghua
    PHYSICS IN MEDICINE AND BIOLOGY, 2023, 68 (18):
  • [48] X-ray Image Segmentation using Multi-task Learning
    Park, Sejin
    Jeong, Woojin
    Moon, Young Shik
    KSII TRANSACTIONS ON INTERNET AND INFORMATION SYSTEMS, 2020, 14 (03) : 1104 - 1120
  • [49] Multi-task Low-rank Affinity Pursuit for Image Segmentation
    Cheng, Bin
    Liu, Guangcan
    Wang, Jingdong
    Huang, Zhongyang
    Yan, Shuicheng
    2011 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2011, : 2439 - 2446
  • [50] Semi-supervised Medical Image Segmentation Based on Multi-scale Knowledge Discovery and Multi-task Ensemble
    Tu, Yudie
    Li, Xiangru
    Zhong, Yunpeng
    Mei, Huanyu
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT XIII, 2024, 14437 : 209 - 222