高级检索
当前位置: 首页 > 详情页

DMSPS: Dynamically mixed soft pseudo-label supervision for scribble-supervised medical image segmentation

文献详情

资源类型:
WOS体系:
Pubmed体系:

收录情况: ◇ SCIE

机构: [1]School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, Chengdu, China [2]Department of Radiation Oncology, Sichuan Cancer Hospital and Institute, Sichuan Cancer Center, Chengdu, China [3]Shanghai Artificial Intelligence Laboratory, Shanghai, China [4]School of Medicine, University of Electronic Science and Technology of China, Chengdu, China [5]SenseTime Research, Shanghai, China
出处:
ISSN:

关键词: Weakly-supervised learning Scribble annotation Soft pseudo-label Uncertainty Annotation expansion

摘要:
High performance of deep learning on medical image segmentation rely on large-scale pixel-level dense annotations, which poses a substantial burden on medical experts due to the laborious and time-consuming annotation process, particularly for 3D images. To reduce the labeling cost as well as maintain relatively satisfactory segmentation performance, weakly-supervised learning with sparse labels has attained increasing attentions. In this work, we present a scribble-based framework for medical image segmentation, called Dynamically Mixed Soft Pseudo-label Supervision (DMSPS). Concretely, we extend a backbone with an auxiliary decoder to form a dual-branch network to enhance the feature capture capability of the shared encoder. Considering that most pixels do not have labels and hard pseudo-labels tend to be over-confident to result in poor segmentation, we propose to use soft pseudo-labels generated by dynamically mixing the decoders' predictions as auxiliary supervision. To further enhance the model's performance, we adopt a two-stage approach where the sparse scribbles are expanded based on predictions with low uncertainties from the first-stage model, leading to more annotated pixels to train the second-stage model. Experiments on ACDC dataset for cardiac structure segmentation, WORD dataset for 3D abdominal organ segmentation and BraTS2020 dataset for 3D brain tumor segmentation showed that: (1) compared with the baseline, our method improved the average DSC from 50.46% to 89.51%, from 75.46% to 87.56% and from 52.61% to 76.53% on the three datasets, respectively; (2) DMSPS achieved better performance than five state-of-the-art scribble-supervised segmentation methods, and is generalizable to different segmentation backbones. The code is available online at: https://github.com/HiLab-git/DMSPS.Copyright © 2024 Elsevier B.V. All rights reserved.

基金:
语种:
WOS:
PubmedID:
中科院(CAS)分区:
出版当年[2023]版:
大类 | 1 区 医学
小类 | 1 区 计算机:人工智能 1 区 计算机:跨学科应用 1 区 工程:生物医学 1 区 核医学
最新[2023]版:
大类 | 1 区 医学
小类 | 1 区 计算机:人工智能 1 区 计算机:跨学科应用 1 区 工程:生物医学 1 区 核医学
JCR分区:
出版当年[2023]版:
Q1 COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE Q1 COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONS Q1 ENGINEERING, BIOMEDICAL Q1 RADIOLOGY, NUCLEAR MEDICINE & MEDICAL IMAGING
最新[2023]版:
Q1 COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE Q1 COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONS Q1 ENGINEERING, BIOMEDICAL Q1 RADIOLOGY, NUCLEAR MEDICINE & MEDICAL IMAGING

影响因子: 最新[2023版] 最新五年平均 出版当年[2023版] 出版当年五年平均 出版前一年[2023版]

第一作者:
第一作者机构: [1]School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, Chengdu, China
通讯作者:
通讯机构: [1]School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, Chengdu, China [3]Shanghai Artificial Intelligence Laboratory, Shanghai, China
推荐引用方式(GB/T 7714):
APA:
MLA:

资源点击量:43370 今日访问量:0 总访问量:3120 更新日期:2024-09-01 建议使用谷歌、火狐浏览器 常见问题

版权所有©2020 四川省肿瘤医院 技术支持:重庆聚合科技有限公司 地址:成都市人民南路四段55号