Classification and visualization of chemotherapy-induced cognitive impairment in volumetric convolutional neural networks

Abstract

Breast cancer is the most common female cancer worldwide, and breast cancer accounts for 30% of female cancers. Of all the treatment modalities, breast cancer survivors who have undergone chemotherapy might complain about cognitive impairment during and after cancer treatment. This phenomenon, chemo-brain, is used to describe the alterations in cognitive functions after receiving systemic chemotherapy. Few reports detect the chemotherapy-induced cognitive impairment (CICI) by performing functional MRI (fMRI) and a deep learning analysis. In this study, we recruited 55 postchemotherapy breast cancer survivors (C+ group) and 65 healthy controls (HC group) and extracted mean fractional amplitudes of low-frequency fluctuations (mfALFF) from resting-state fMRI as our input feature. Two state-of-the-art deep learning architectures, ResNet-50 and DenseNet-121, were transformed to 3D, embedded with squeeze and excitation (SE) blocks and then trained to differentiate cerebral alterations based on the effect of chemotherapy. An integrated gradient was applied to visualize the pattern that was recognized by our model. The average performance of SE-ResNet-50 models was an accuracy of 80%, precision of 78% and recall of 70%; on the other hand, the SE-DenseNet-121 model reached identical results with an average of 80% accuracy, 86% precision and 80% recall. The regions with the greatest contributions highlighted by the integrated gradients algorithm for differentiating chemo-brain were the frontal, temporal, parietal and occipital lobe. These regions were consistent with other studies and strongly associated with the default mode and dorsal attention networks. We constructed two volumetric state-of-the-art models and visualized the patterns that are critical for identifying chemo-brains from normal brains. We hope that these results will be helpful in clinically tracking chemo-brain in the future.