Mammographic image metadata learning for model pretraining and explainable predictions

Abstract

Purpose: To introduce a novel technique for pretraining deep neural networks on mammographic images, where the network learns to predict multiple metadata attributes and simultaneously to match images from the same patient and study. Further to demonstrate how this network can be used to produce explainable predictions. Methods: We trained a neural network on a dataset of 85,558 raw mammographic images and seven types of metadata, using a combination of supervised and self-supervised learning techniques. We evaluated the performance of our model on a dataset of 4,678 raw mammographic images using classification accuracy and correlation. We also designed an ablation study to demonstrate how the model can produce explainable predictions. Results: The model learned to predict all but one of the seven metadata fields with classification accuracy ranging from 78-99% on the validation dataset. The model was able to predict which images were from the same patient with over 93% accuracy on a balanced dataset. Using a simple X-ray system classifier built on top of the first model, representations learned on the initial X-ray system classification task showed by far the largest effect size on ablation, illustrating a method for producing explainable predictions. Conclusions: It is possible to train a neural network to predict several kinds of mammogram metadata simultaneously. The representations learned by the model for these tasks can be summed to produce an image representation that captures features unique to a patient and study. With such a model, ablation offers a promising method to enhance the explainability of deep learning predictions.