A two-stage dual-task learning strategy for early prediction of pathological complete response to neoadjuvant chemotherapy for breast cancer using dynamic contrast-enhanced magnetic resonance images.
Authors
Affiliations (2)
Affiliations (2)
- Radiation Oncology, The University of Texas Southwestern Medical Center, 2280 Inwood Rd, Dallas, Texas, 75390-9096, UNITED STATES.
- Department of Radiation Oncology, The University of Texas Southwestern Medical Center, 2280 Inwood Rd, Dallas, Texas, 75390-9096, UNITED STATES.
Abstract
Early prediction of treatment response can facilitate personalized treatment for breast cancer patients. Studies on the I-SPY 2 clinical trial demonstrate that multi-time point dynamic contrast-enhanced magnetic resonance (DCEMR) imaging improves the accuracy of predicting pathological complete response (pCR) to chemotherapy. However, previous image-based prediction models usually rely on mid- or post-treatment images to ensure the accuracy of prediction, which may outweigh the benefit of response-based adaptive treatment strategy. Accurately predicting the pCR at the early time point is desired yet remains challenging. To improve prediction accuracy at the early time point of treatment, we proposed a two-stage dual-task learning strategy to train a deep neural network for early prediction using only early-treatment data. We developed and evaluated our proposed method using the I-SPY 2 dataset, which included DCEMR images acquired at three time points: pretreatment (T0), after 3 weeks (T1) and 12 weeks of treatment (T2). At the first stage, we trained a convolutional long short-term memory (LSTM) model using all the data to predict pCR and extract the latent space image representation at T2. At the second stage, we trained a dual-task model to simultaneously predict pCR and the image representation at T2 using images from T0 and T1. This allowed us to predict pCR earlier without using images from T2. By using the conventional single-stage single-task strategy, the area under the receiver operating characteristic curve (AUROC) was 0.799. By using the proposed two-stage dual-task learning strategy, the AUROC was improved to 0.820. Our proposed two-stage dual-task learning strategy can improve model performance significantly (p=0.0025) for predicting pCR at the early time point (3rd week) of neoadjuvant chemotherapy for high-risk breast cancer patients. The early prediction model can potentially help physicians to intervene early and develop personalized plans at the early stage of chemotherapy.