.png){kind=icon}
返回
Context Label Learning: Improving Background Class Representations in Semantic Segmentation
.png){kind=icon}
OA
AI
DOI：10.1109/TMI.2023.3242838.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Background samples provide key contextual information for segmenting regions of interest (ROIs). However, they always cover a diverse set of structures, causing difficulties for the segmentation model to learn good decision boundaries with high sensitivity and precision. The issue concerns the highly heterogeneous nature of the background class, resulting in multi-modal distributions. Empirically, we find that neural networks trained with heterogeneous background struggle to map the corresponding contextual samples to compact clusters in feature space. As a result, the distribution over background logit activations may shift across the decision boundary, leading to systematic over-segmentation across different datasets and tasks. In this study, we propose context label learning (CoLab) to improve the context representations by decomposing the background class into several subclasses. Specifically, we train an auxiliary network as a task generator, along with the primary segmentation model, to automatically generate context labels that positively affect the ROI segmentation accuracy. Extensive experiments are conducted on several challenging segmentation tasks and datasets. The results demonstrate that CoLab can guide the segmentation model to map the logits of background samples away from the decision boundary, resulting in significantly improved segmentation accuracy. Code is available at https://github.com/ZerojumpLine/CoLab.
Keyword:
Image segmentation
Tumors
Task analysis
Liver
Kidney
Context modeling
Training
Underfitting
multi-task learning
self-supervised learning
image segmentation
期刊
IF:
9.8
论文数:
6.3K
被引数:
3.7W
机构
引用论文
分享收藏Robust whole-brain segmentation: Application to traumatic brain injury.png)
MEDICAL IMAGE ANALYSIS
OAAI
MEDICAL IMAGE ANALYSISIF11.8
OAAI分享收藏CT radiomics to predict high-risk intraductal papillary mucinous neoplasms of the pancreasMEDICAL PHYSICSOAAI
MEDICAL PHYSICSIF3.2
OAAI分享收藏
查看更多内容