Feature Space Guidance for Breast Cancer Classification in DCE-MRI
Authors: Benjamin Hamm, Yannick Kirchhoff, Maximilian Rokuss, Moritz Langenberg, Constantin Ulrich, Tassilo Wald, Jeremias Traub, Karol Gotkowski, +1 more
Organizations: German Cancer Research Center (DKFZ) Heidelberg, Division of Medical Image Computing, Germany · Medical Faculty, Heidelberg University, Germany · Faculty of Mathematics and Computer Science, Heidelberg University, Germany · HIDSS4Health – Helmholtz Information and Data Science School for Health, Karlsruhe/Heidelberg, Germany · Helmholtz Imaging, German Cancer Research Center (DKFZ), Heidelberg, Germany · Pattern Analysis and Learning Group, Department of Radiation Oncology, Heidelberg University Hospital, Germany
Dynamic contrast enhanced breast MRI (DCE-MRI) is a powerful clinical tool for breast cancer detection, providing high resolution anatomical detail together with rich temporal contrast information. However, high dimensional 4D inputs, small lesions, and heterogeneous acquisition protocols across clinical sites hinder robust automated classification of healthy, benign, and malignant cases. To address these challenges, we propose a framework that dynamically analyzes latent representations to adapt to protocol-specific characteristics. Spatial variability is mitigated by reducing confounding background uptake and compensating for misalignment caused by deformable soft tissue. Additionally, relationships in the latent space across phases are leveraged to select the most informative temporal features, improving robustness to protocol-specific temporal variability. Finally, task specific discriminative features are promoted through large scale supervised lesion segmentation pretraining, which substantially enhances downstream finetuning. Evaluated under leave-one-center-out validation on the ODELIA dataset and the held-out AMBL cohort, the proposed framework substantially outperforms finetuned radiology foundation models and prior methods, improving mean AUROC by nearly 8 points and balanced accuracy by 4 points over the strongest baseline. Additionally, our method achieved first place in the MICCAI ODELIA Breast MRI Challenge 2025, further demonstrating its effectiveness for robust breast cancer classification. We publicly release our codebase under https://github.com/MIC-DKFZ/CURIAtor.
Figures & tables
Figure 1: Overview of the proposed foundation model-driven framework for breast cancer classification from DCE-MRI. The framework includes A) the BreastDivider model for ROI detection and breast separation, B) adaptive temporal signal analysis with the CURIAtor module selects the most informative features, C) large-scale anatomically informed pretraining, and D) transfer learning for final lesion classification.
Figure 2: Pairwise distance matrix D computed by the CURIAtor module for UMCU 0178187682 Right (ODELIA). Each entry encodes the volume-level similarity between two temporal acquisitions within the anatomically constrained ROI; lower values indicate higher similarity. Post4 is farthest from Post1 and from every other post-contrast phase, and is selected (purple) alongside Pre and Post1. Overlay: ∣Vn−VPost1∣ within the ROI on one axial slice.
Computer Sciences Department, University of Oviedo, Asturias, Spain · Biomedical Engineering Center (BME), University of Oviedo, Asturias, Spain · Electrical Engineering Department, University of Oviedo, Asturias, Spain +6
Nankai University, No. 94 Weijin Road, Nankai District, Tianjin, 300071, China · Zhongguancun Academy, Beijing, China · Beijing University of Posts and Telecommunications, Beijing, China +5
German Cancer Research Center (DKFZ) Heidelberg, Division of Medical Image Computing, Germany · Medical Faculty, Heidelberg University, Germany · Faculty of Mathematics and Computer Science, Heidelberg University, Germany +3