.png){kind=icon}
返回
Learning Robust Shape Regularization for Generalizable Medical Image Segmentation
.png){kind=icon}
PRE
AI
DOI：10.1109/TMI.2024.3371987.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Generalizable medical image segmentation enables models to generalize to unseen target domains under domain shift issues. Recent progress demonstrates that the shape of the segmentation objective, with its high consistency and robustness across domains, can serve as a reliable regularization to aid the model for better cross-domain performance, where existing methods typically seek a shared framework to render segmentation maps and shape prior concurrently. However, due to the inherent texture and style preference of modern deep neural networks, the edge or silhouette of the extracted shape will inevitably be undermined by those domain-specific texture and style interferences of medical images under domain shifts. To address this limitation, we devise a novel framework with a separation between the shape regularization and the segmentation map. Specifically, we first customize a novel whitening transform-based probabilistic shape regularization extractor namely WT-PSE to suppress undesirable domain-specific texture and style interferences, leading to more robust and high-quality shape representations. Second, we deliver a Wasserstein distance-guided knowledge distillation scheme to help the WT-PSE to achieve more flexible shape extraction during the inference phase. Finally, by incorporating domain knowledge of medical images, we propose a novel instance-domain whitening transform method to facilitate a more stable training process with improved performance. Experiments demonstrate the performance of our proposed method on both multi-domain and single-domain generalization.
Keyword:
Shape
Image segmentation
Biomedical imaging
Training
Image edge detection
Feature extraction
Artificial neural networks
Medical image segmentation
whitening transform
shape regularization
knowledge distillation
期刊
IF:
9.8
论文数:
6.3K
被引数:
3.7W
机构
引用论文
.png){kind=icon}
Generalizing Deep Learning for Medical Image Segmentation to Unseen Domains via Deep Stacked Transformation通过深度堆叠变换将医学图像分割的深度学习推广到看不见的领域
OAAI
OAAI分享收藏分享收藏
查看更多内容