First U-Net Layers Contain More Domain Specific Information Than The Last Ones
Not on the map. Why?
Abstract (arXiv)
MRI scans appearance significantly depends on scanning protocols and, consequently, the data-collection institution. These variations between clinical sites result in dramatic drops of CNN segmentation quality on unseen domains. Many of the recently proposed MRI domain adaptation methods operate with the last CNN layers to suppress domain shift. At the same time, the core manifestation of MRI variability is a considerable diversity of image intensities. We hypothesize that these differences can be eliminated by modifying the first layers rather than the last ones. To validate this simple idea, we conducted a set of experiments with brain MRI scans from six domains. Our results demonstrate that 1) domain-shift may deteriorate the quality even for a simple brain extraction segmentation task (surface Dice Score drops from 0.85-0.89 even to 0.09); 2) fine-tuning of the first layers significantly outperforms fine-tuning of the last layers in almost all supervised domain adaptation setups. Moreover, fine-tuning of the first layers is a better strategy than fine-tuning of the whole network, if the amount of annotated data from the new domain is strictly limited.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). First posted on arXiv 17 Aug 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 1; 2023: 2; 2024: 3; 2025: 1.
10 papers.
- 2023MedShapeNet -- A Large-Scale Dataset of 3D Medical Shapes for Computer Visioncited by 31
- 2021Transfer Learning in Magnetic Resonance Brain Imaging: a Systematic Reviewcited by 19
- 2024ConvLoRA and AdaBN based Domain Adaptation via Self-Trainingcited by 10
- 2021Anatomy of Domain Shift Impact on U-Net Layers in MRI Segmentationcited by 9
- 2021Robust White Matter Hyperintensity Segmentation on Unseen Domaincited by 8
- 2023Benchmarking the Influence of Pre-training on Explanation Performance in MR Image Classificationcited by 8
- 2022Feather-Light Fourier Domain Adaptation in Magnetic Resonance Imagingcited by 5
- 2024Rethinking U-net Skip Connections for Biomedical Image Segmentationcited by 3
- 2024Sim-to-real supervised domain adaptation for radioisotope identificationcited by 3
- 2025M3DA: Benchmark for Unsupervised Domain Adaptation in 3D Medical Image Segmentationcited by 1
- 2025M3DA: Benchmark for Unsupervised Domain Adaptation in 3D Medical Image Segmentation24 Feb 2025
- 2024Sim-to-real supervised domain adaptation for radioisotope identification10 Dec 2024
- 2024Rethinking U-net Skip Connections for Biomedical Image Segmentation13 Feb 2024
- 2024ConvLoRA and AdaBN based Domain Adaptation via Self-Training7 Feb 2024
- 2023MedShapeNet -- A Large-Scale Dataset of 3D Medical Shapes for Computer Vision30 Aug 2023
- 2023Benchmarking the Influence of Pre-training on Explanation Performance in MR Image Classification21 Jun 2023
- 2022Feather-Light Fourier Domain Adaptation in Magnetic Resonance Imaging31 Jul 2022
- 2021Anatomy of Domain Shift Impact on U-Net Layers in MRI Segmentation10 Jul 2021
- 2021Robust White Matter Hyperintensity Segmentation on Unseen Domain12 Feb 2021
- 2021Transfer Learning in Magnetic Resonance Brain Imaging: a Systematic Review2 Feb 2021