A Unified Attention U-Net Framework for Cross-Modality Tumor Segmentation in MRI and CT
Not on the map. Why?
Abstract (arXiv)
This study presents a unified Attention U-Net architecture trained jointly on MRI (BraTS 2021) and CT (LIDC-IDRI) datasets to investigate the generalizability of a single model across diverse imaging modalities and anatomical sites. Our proposed pipeline incorporates modality-harmonized preprocessing, attention-gated skip connections, and a modality-aware Focal Tversky loss function. To the best of our knowledge, this study is among the first to evaluate a single Attention U-Net trained simultaneously on separate MRI (BraTS) and CT (LIDC-IDRI) tumor datasets, without relying on modality-specific encoders or domain adaptation. The unified model demonstrates competitive performance in terms of Dice coefficient, IoU, and AUC on both domains, thereby establishing a robust and reproducible baseline for future research in cross-modality tumor segmentation.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 16 of its 23 references that are in the map. First posted on arXiv 7 Jan 2026.
Builds on
16 of 23 references.
References · 16
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
- 2018Albumentations: fast and flexible image augmentationscited by 558
- 2021The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classificationcited by 507
- 2017Tversky loss function for image segmentation using 3D fully convolutional deep networkscited by 228
- 2018Boundary loss for highly unbalanced segmentationcited by 187
- 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 171
- 2019Models Genesis: Generic Autodidactic Models for 3D Medical Image Analysiscited by 159
4 more
- 2020Unsupervised Bidirectional Cross-Modality Adaptation via Deeply Synergistic Image and Feature Alignment for Medical Image Segmentationcited by 134
- 2016HeMIS: Hetero-Modal Image Segmentationcited by 102
- 2018SynSeg-Net: Synthetic Segmentation Without Target Modality Ground Truthcited by 68
- 2018Unsupervised Cross-Modality Domain Adaptation of ConvNets for Biomedical Image Segmentations with Adversarial Losscited by 60
Built on by
Not cited yet.