mlsanityData to 7 Oct 2026

A Unified Attention U-Net Framework for Cross-Modality Tumor Segmentation in MRI and CT

arXiv 2601.06187 ↗First posted 7 Jan 2026cs.CVcited by 0

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

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  4. 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
  5. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
  6. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
  7. 2018Albumentations: fast and flexible image augmentationscited by 558
  8. 2021The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classificationcited by 507
  9. 2017Tversky loss function for image segmentation using 3D fully convolutional deep networkscited by 228
  10. 2018Boundary loss for highly unbalanced segmentationcited by 187
  11. 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 171
  12. 2019Models Genesis: Generic Autodidactic Models for 3D Medical Image Analysiscited by 159
4 more
  1. 2020Unsupervised Bidirectional Cross-Modality Adaptation via Deeply Synergistic Image and Feature Alignment for Medical Image Segmentationcited by 134
  2. 2016HeMIS: Hetero-Modal Image Segmentationcited by 102
  3. 2018SynSeg-Net: Synthetic Segmentation Without Target Modality Ground Truthcited by 68
  4. 2018Unsupervised Cross-Modality Domain Adaptation of ConvNets for Biomedical Image Segmentations with Adversarial Losscited by 60

Built on by

Not cited yet.