Multi-task Graph Convolutional Neural Network for Calcification Morphology and Distribution Analysis in Mammograms

Abstract

The morphology and distribution of microcalcifications in a cluster are the most important characteristics for radiologists to diagnose breast cancer. However, it is time-consuming and difficult for radiologists to identify these characteristics, and there also lacks of effective solutions for automatic characterization. In this study, we proposed a multi-task deep graph convolutional network (GCN) method for the automatic characterization of morphology and distribution of microcalcifications in mammograms. Our proposed method transforms morphology and distribution characterization into node and graph classification problem and learns the representations concurrently. Through extensive experiments, we demonstrate significant improvements with the proposed multi-task GCN comparing to the baselines. Moreover, the achieved improvements can be related to and enhance clinical understandings. We explore, for the first time, the application of GCNs in microcalcification characterization that suggests the potential of graph learning for more robust understanding of medical images.

References in corpus (7)

• Distilling the Knowledge in a Neural Network
• MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
• Semi-Supervised Classification with Graph Convolutional Networks
• DeeperGCN: All You Need to Train Deeper GCNs
• Improving Multi-Task Deep Neural Networks via Knowledge Distillation for Natural Language Understanding
• DeepGCNs: Can GCNs Go as Deep as CNNs?
• RGCNN: Regularized Graph CNN for Point Cloud Segmentation