mlsanityData to 14 Oct 2026

BMDS-Net:Deployment-aware multi-modal brain tumor segmentation with adaptive fusion,decoder regularization,and Bayesian calibration

arXiv 2601.17504 ↗First posted 24 Jan 2026cs.CV q-bio.QMcited by 0

Not on the map. Why?

Abstract (arXiv)

Multi-modal MRI enables detailed brain tumor sub-region segmentation, but clinical deployment remains affected by missing sequences,boundary errors, and overconfident predictions. We present BMDS-Net, a two-stage Swin UNETR framework that combines adaptivemodality fusion, boundary-aware decoder regularization, and efficient Bayesian calibration. The deterministic stage adds zero-initializedmultimodal contextual fusion (MMCF) and residual-gated deep decoder supervision (DDS) to modulate modality contributions andregularize hierarchical boundary reconstruction. The second stage replaces only the final convolutional layer with a Bayesian layer,enabling Monte Carlo uncertainty estimates with limited additional training cost. On BraTS 2021 (1,251 cases), BMDS-Net achieved Dice scores of 0.9293, 0.9098, and 0.8675 for WT, TC, and ET, with HD95 values of 2.27, 2.22, and 3.27 mm. Configuration comparisons showed that auxiliary supervision favored complete-input segmentation, whereas the complete MMCF+gated-DDS configuration was most robust under single-modality removal. Bayesian BMDS-Net achieved an expected calibration error of 0.0037, close to a three-model ensemble (0.0035) at approximately 60% lower training cost. On BraTS 2020, the same protocol preserved the leading internal ranking of the complete configuration among five Swin UNETR-controlled variants. The implementation code is available at https://github.com/RyanZhou168/BMDS-Net.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 29 references that are in the map. First posted on arXiv 24 Jan 2026.

Builds on

15 of 29 references.

References · 15

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  2. 2017On Calibration of Modern Neural Networkscited by 4,960
  3. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  4. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
  5. 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
  6. 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
  7. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
  8. 2021The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classificationcited by 518
  9. 2014Deeply-Supervised Netscited by 367
  10. 20183D MRI brain tumor segmentation using autoencoder regularizationcited by 328
  11. 2021TransBTS: Multimodal Brain Tumor Segmentation Using Transformercited by 265
  12. 2021nnFormer: Interleaved Transformer for Volumetric Segmentationcited by 180
3 more
  1. 2023MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentationcited by 162
  2. 2016HeMIS: Hetero-Modal Image Segmentationcited by 103
  3. 2022MyoPS-Net: Myocardial Pathology Segmentation with Flexible Combination of Multi-Sequence CMR Imagescited by 22

Built on by

Not cited yet.