Pre-training with Simulated Ultrasound Images for Breast Mass Segmentation and Classification

Abstract

We investigate the usefulness of formula-driven supervised learning (FDSL) for breast ultrasound (US) image analysis. Medical data are usually too scarce to develop a better performing deep learning model from scratch. Transfer learning with networks pre-trained on ImageNet is commonly applied to address this problem. FDSL techniques have been recently investigated as an alternative solution to ImageNet based approaches. In the FDSL setting, networks for transfer learning applications are developed using large amounts of synthetic images generated with mathematical formulas, possibly taking into account the characteristics of the target data. In this work, we use Field II to develop a large synthetic dataset of 100 000 US images presenting different contour objects, as shape features play an important role in breast mass characterization in US. Synthetic data are utilized to pre-train the ResNet50 classification model and various variants of the U-Net segmentation network. Next, the pre-trained models are fine-tuned on breast mass US images. Our results demonstrate that the proposed FDSL approach can provide good performance with respect to breast mass classification and segmentation.