A Data-Centric Approach to Deep Learning for Brain Metastasis Analysis at MRI.

Background With the increasing incidence of brain metastases (BMs), artificial intelligence models have shown promise in assisting with the detection and volumetric analysis of lesions at MRI. However, current models are limited in identifying small lesions and lack generalizability. Purpose To develop a generalizable deep learning system for detecting, segmenting, and longitudinally tracking BMs of any size at MRI. Materials and Methods In this retrospective study, a data-centric approach to deep learning model development was used. A multicenter dataset was collected, comprising pre- and/or posttreatment MRI scans from patients with BMs and MRI scans from patients with cancer without BMs (December 2015 to August 2023). Iterative data annotation by radiologists with systematic quality control increased the consistency of reference segmentations. A modified nnU-Net framework, with robust data preprocessing and augmentation, was used. Lesion-wise detection metrics and segmentation performance, Dice similarity coefficient, and normalized surface distance were evaluated. Results In total, 1985 scans from 1623 patients (mean age, 62.0 years ± 12.2 [SD]; 743 female patients, 157 patients of unknown sex), with 5552 BMs, were included. BMs were present in 64.8% of the scans (1286 of 1985), 36.0% (463 of 1286) of which were posttreatment scans. The model was trained on 1451 scans acquired on 30 different scanners. In internal testing (