mlsanityData to 14 Oct 2026

Tumor-aware augmentation with task-guided attention analysis improves rectal cancer segmentation from magnetic resonance images

arXiv 2605.05522 ↗First posted 6 May 2026eess.IV cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Although self-supervised pretraining is expected to learn broadly transferable representations, its effectiveness across imaging modalities substantially different from the pretraining domain, and on complex tumor-segmentation tasks, remains understudied. Evaluating CT-pretrained transformers on MRI rectal cancer segmentation, we identified two interacting failure modes in CT-to-MRI transfer: (a) inefficient token usage caused by zero-padding to match pretrained input dimensions, and (b) ineffective feature adaptation. We investigated these vulnerabilities using two primary CT-pretrained hierarchical shifted-window transformer backbones, SMIT and Swin UNETR, together with VoCo as a large-scale-pretrained supporting benchmark; these models differ in pretraining objectives and datasets. Mechanistic analysis leveraged an attention dilution index (ADI), an entropy-based metric quantifying attention diverted toward uninformative padding tokens, and centered kernel alignment (CKA) to measure feature reuse during MRI adaptation. ADI increased with zero-padding, while high feature reuse did not necessarily translate to improved downstream accuracy. To mitigate these issues, we introduced two interventions: a tumor-aware augmentation strategy to expand tumor appearance heterogeneity coverage, and an anisotropic cropping strategy to restore token efficiency. Fine-tuning with these strategies on identical rectal MRI datasets yielded detection rates of 91.1% (225/247) and 88.7% (219/247) for the primary SMIT and Swin UNETR backbones, with the supporting VoCo benchmark reaching 90.3% (223/247), demonstrating significantly improved robustness under CT-to-MRI transfer. This study is among the first to examine when pretrained transformers fail to transfer across imaging modalities and demonstrates how targeted mitigation strategies can systematically overcome cross-modality transfer limitations.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 47 references that are in the map. First posted on arXiv 6 May 2026.

Builds on

18 of 47 references.

References · 18

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  4. 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
  5. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  6. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  7. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
  8. 2020Quantifying Attention Flow in Transformerscited by 775
  9. 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 675
  10. 2019Transfusion: Understanding Transfer Learning for Medical Imagingcited by 368
  11. 2019Med3D: Transfer Learning for 3D Medical Image Analysiscited by 288
  12. 2020Do Wide and Deep Networks Learn the Same Things? Uncovering How Neural Network Representations Vary with Width and Depthcited by 245
6 more
  1. 2022What Makes Transfer Learning Work For Medical Images: Feature Reuse & Other Factorscited by 61
  2. 2021Is it Time to Replace CNNs with Transformers for Medical Images?cited by 60
  3. 2022Towards a Guideline for Evaluation Metrics in Medical Image Segmentationcited by 60
  4. 2024Large-Scale 3D Medical Image Pre-training with Geometric Context Priorscited by 36
  5. 2022Self-supervised 3D anatomy segmentation using self-distilled masked image transformer (SMIT)cited by 35
  6. 2025Triad: Vision Foundation Model for 3D Magnetic Resonance Imagingcited by 24

Built on by

Not cited yet.