Automated Detection of Brain Metastases on T1-Weighted MRI Using a Convolutional Neural Network: Impact of Volume

Gabriel Chartrand1, Ramón D Emiliani1, Sophie A Pawlowski1

  • 1AFX Medical Inc., Montréal, Canada.

Abstract

Insights

Machine learning, specifically convolutional neural networks (CNNs), can improve brain metastasis (BM) detection. Combining volume-aware loss and sampling strategies significantly enhanced sensitivity for detecting small BM on MRI scans.

Area of Science:

  • Radiology
  • Medical Imaging
  • Machine Learning

Background:

  • Brain metastases (BM) detection and segmentation are crucial for treatment planning.
  • Convolutional neural networks (CNNs) show promise but struggle with detecting small lesions due to sensitivity-precision trade-offs.

Purpose of the Study:

  • To optimize BM detection and segmentation using machine learning.
  • To improve the sensitivity of CNNs for detecting small brain metastases.

Main Methods:

  • A U-Net inspired CNN model was trained on 530 radiation oncology patients' contrast-enhanced T1-weighted MRI scans.
  • Compared Dice, Dice+boundary, and volume-aware (VA) loss functions with label and VA sampling strategies.
  • Evaluated performance using Dice scores, volumetric error, sensitivity, and precision.

Main Results:

  • The combination of VA loss and VA sampling achieved the highest sensitivity (91%) and precision (81%).
  • VA loss reduced false negatives by 58% for small BM (2.5-6 mm diameter).
  • VA sampling further reduced false negatives by 30% for small BM.

Conclusions:

  • Incorporating BM size into CNN loss and sampling functions enhances detection sensitivity for small lesions.
  • The proposed pipeline, using a single MRI sequence, achieved 91% detection sensitivity with minimal false positives (0.66 per scan).

Related Concept Videos