Mask-Guided Multi-Channel SwinUNETR Framework for Robust MRI Classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Joshi, Smriti, Garrucho, Lidia, Osuala, Richard, Diaz, Oliver, Lekadir, Karim
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912557744259072
author Joshi, Smriti
Garrucho, Lidia
Osuala, Richard
Diaz, Oliver
Lekadir, Karim
author_facet Joshi, Smriti
Garrucho, Lidia
Osuala, Richard
Diaz, Oliver
Lekadir, Karim
contents Breast cancer is one of the leading causes of cancer-related mortality in women, and early detection is essential for improving outcomes. Magnetic resonance imaging (MRI) is a highly sensitive tool for breast cancer detection, particularly in women at high risk or with dense breast tissue, where mammography is less effective. The ODELIA consortium organized a multi-center challenge to foster AI-based solutions for breast cancer diagnosis and classification. The dataset included 511 studies from six European centers, acquired on scanners from multiple vendors at both 1.5 T and 3 T. Each study was labeled for the left and right breast as no lesion, benign lesion, or malignant lesion. We developed a SwinUNETR-based deep learning framework that incorporates breast region masking, extensive data augmentation, and ensemble learning to improve robustness and generalizability. Our method achieved second place on the challenge leaderboard, highlighting its potential to support clinical breast MRI interpretation. We publicly share our codebase at https://github.com/smriti-joshi/bcnaim-odelia-challenge.git.
format Preprint
id arxiv_https___arxiv_org_abs_2508_20621
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Mask-Guided Multi-Channel SwinUNETR Framework for Robust MRI Classification
Joshi, Smriti
Garrucho, Lidia
Osuala, Richard
Diaz, Oliver
Lekadir, Karim
Computer Vision and Pattern Recognition
Breast cancer is one of the leading causes of cancer-related mortality in women, and early detection is essential for improving outcomes. Magnetic resonance imaging (MRI) is a highly sensitive tool for breast cancer detection, particularly in women at high risk or with dense breast tissue, where mammography is less effective. The ODELIA consortium organized a multi-center challenge to foster AI-based solutions for breast cancer diagnosis and classification. The dataset included 511 studies from six European centers, acquired on scanners from multiple vendors at both 1.5 T and 3 T. Each study was labeled for the left and right breast as no lesion, benign lesion, or malignant lesion. We developed a SwinUNETR-based deep learning framework that incorporates breast region masking, extensive data augmentation, and ensemble learning to improve robustness and generalizability. Our method achieved second place on the challenge leaderboard, highlighting its potential to support clinical breast MRI interpretation. We publicly share our codebase at https://github.com/smriti-joshi/bcnaim-odelia-challenge.git.
title Mask-Guided Multi-Channel SwinUNETR Framework for Robust MRI Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2508.20621