Multitask Classification and Segmentation for Cancer Diagnosis in Mammography

Abstract

Annotation cost is a bottleneck for collecting massive data in mammography, especially for training deep neural networks. In this paper, we study the use of heterogeneous levels of annotation granularity to improve predictive performances. More precisely, we introduce a multi-task learning scheme for training convolutional neural network (ConvNets), which combines segmentation and classification, using image-level and pixel-level annotations. In this way, different objectives can be used to regularize training by sharing intermediate deep representations. Successful experiments are carried out on the Digital Database of Screening Mammography (DDSM) to validate the relevance of the proposed approach.

arXiv commentInternational Conference on Medical Imaging with Deep Learning 2019. MIDL 2019 [arXiv:1907.08612]

References in corpus (2)

• High-Resolution Breast Cancer Screening with Multi-View Deep Convolutional Neural Networks
• Breast Mass Classification from Mammograms using Deep Convolutional Neural Networks

Cited by in corpus (6)

• SolarNet: A Deep Learning Framework to Map Solar Power Plants In China From Satellite Imagery
• Learning joint segmentation of tissues and brain lesions from task-specific hetero-modal domain-shifted datasets
• Generalizable multi-task, multi-domain deep segmentation of sparse pediatric imaging datasets via multi-scale contrastive regularization and multi-joint anatomical priors
• Evaluation of Complexity Measures for Deep Learning Generalization in Medical Image Analysis
• Joint Liver Lesion Segmentation and Classification via Transfer Learning
• U-GAT: Multimodal Graph Attention Network for COVID-19 Outcome Prediction