Longitudinal deep neural networks for assessing metastatic brain cancer on a large open benchmark

Affiliations
  • 1Department of Neurosurgery, NYU Langone Health, New York, NY, USA.
  • 2NVIDIA, Santa Clara, CA, USA.
  • 3Electrical and Computer Engineering, NYU Tandon School of Engineering, New York, NY, USA.
  • 4Department of Radiology, NYU Langone Health, New York, NY, USA.
  • 5Center for Data Science, New York University, New York, NY, USA.
  • 6Harvard Medical School, Boston, MA, USA.
  • 7Department of Radiation Oncology, NYU Langone Health, New York, NY, USA.
  • 8Eikon Therapeutics, New York, NY, USA.
  • 9Columbia University Vagelos College of Surgeons and Physicians, New York, NY, USA.
  • 10Department of Neurosurgery, NYU Langone Health, New York, NY, USA. eric.oermann@nyulangone.org.
  • 11Department of Radiology, NYU Langone Health, New York, NY, USA. eric.oermann@nyulangone.org.
  • 12Center for Data Science, New York University, New York, NY, USA. eric.oermann@nyulangone.org.

|

Abstract

The detection and tracking of metastatic cancer over the lifetime of a patient remains a major challenge in clinical trials and real-world care. Advances in deep learning combined with massive datasets may enable the development of tools that can address this challenge. We present NYUMets-Brain, the world’s largest, longitudinal, real-world dataset of cancer consisting of the imaging, clinical follow-up, and medical management of 1,429 patients. Using this dataset we developed Segmentation-Through-Time, a deep neural network which explicitly utilizes the longitudinal structure of the data and obtained state-of-the-art results at small (<10 mm) metastases detection and segmentation. We also demonstrate that the monthly rate of change of brain metastases over time are strongly predictive of overall survival (HR 1.27, 95%CI 1.18-1.38). We are releasing the dataset, codebase, and model weights for other cancer researchers to build upon these results and to serve as a public benchmark.