Medical image segmentation is crucial for accurate disease detection and diagnosis, but current deep learning methods often struggle with generalization across diverse imaging systems. This paper introduces \(\text {BMS}^3\) (Bayesian Modeling Based SwinUNet Segmentation on a Self-distillation Architecture), a novel approach for addressing the challenge of domain invariance in medical image segmentation. By integrating Bayesian modeling for feature extraction with the efficient Swin Transformer-based U-Net architecture, \(\text {BMS}^3\) decomposes images into domain-invariant shape components and domain-specific appearance attributes. This approach enhances generalization to unseen data and improves performance on cross-domain tasks. Additionally, we incorporated an optional self-distillation mechanism to further boost performance on imbalanced datasets. Extensive experiments on multiple medical imaging datasets demonstrate that \(\text {BMS}^3\) outperforms state-of-the-art methods, including ResNet, TransUNet, and BayeSeg, in both segmentation accuracy and computational efficiency. Our method showed particular promise in maintaining high performance across diverse medical imaging systems, addressing a critical need in clinical applications where data heterogeneity is common. \(\text {BMS}^3\) represents a significant advancement in creating a more robust and adaptable medical image segmentation systems.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

\(\text {BMS}^3\) : Bayesian Modeling Based SwinUNet Segmentation on Self-distillation Architecture

  • Jiecheng Liao,
  • Ruijie Hu,
  • Junhao Lu,
  • Weifeng Su,
  • Shi He,
  • Yixuan Ji,
  • Liangfu Chen

摘要

Medical image segmentation is crucial for accurate disease detection and diagnosis, but current deep learning methods often struggle with generalization across diverse imaging systems. This paper introduces \(\text {BMS}^3\) (Bayesian Modeling Based SwinUNet Segmentation on a Self-distillation Architecture), a novel approach for addressing the challenge of domain invariance in medical image segmentation. By integrating Bayesian modeling for feature extraction with the efficient Swin Transformer-based U-Net architecture, \(\text {BMS}^3\) decomposes images into domain-invariant shape components and domain-specific appearance attributes. This approach enhances generalization to unseen data and improves performance on cross-domain tasks. Additionally, we incorporated an optional self-distillation mechanism to further boost performance on imbalanced datasets. Extensive experiments on multiple medical imaging datasets demonstrate that \(\text {BMS}^3\) outperforms state-of-the-art methods, including ResNet, TransUNet, and BayeSeg, in both segmentation accuracy and computational efficiency. Our method showed particular promise in maintaining high performance across diverse medical imaging systems, addressing a critical need in clinical applications where data heterogeneity is common. \(\text {BMS}^3\) represents a significant advancement in creating a more robust and adaptable medical image segmentation systems.