DeepAI AI Chat
Log In Sign Up

A Unified Mammogram Analysis Method via Hybrid Deep Supervision

by   Rongzhao Zhang, et al.

Automatic mammogram classification and mass segmentation play a critical role in a computer-aided mammogram screening system. In this work, we present a unified mammogram analysis framework for both whole-mammogram classification and segmentation. Our model is designed based on a deep U-Net with residual connections, and equipped with the novel hybrid deep supervision (HDS) scheme for end-to-end multi-task learning. As an extension of deep supervision (DS), HDS not only can force the model to learn more discriminative features like DS, but also seamlessly integrates segmentation and classification tasks into one model, thus the model can benefit from both pixel-wise and image-wise supervisions. We extensively validate the proposed method on the widely-used INbreast dataset. Ablation study corroborates that pixel-wise and image-wise supervisions are mutually beneficial, evidencing the efficacy of HDS. The results of 5-fold cross validation indicate that our unified model matches state-of-the-art performance on both mammogram segmentation and classification tasks, which achieves an average segmentation Dice similarity coefficient (DSC) of 0.85 and a classification accuracy of 0.89. The code is available at


Regression and Learning with Pixel-wise Attention for Retinal Fundus Glaucoma Segmentation and Detection

Observing retinal fundus images by an ophthalmologist is a major diagnos...

FU-net: Multi-class Image Segmentation Using Feedback Weighted U-net

In this paper, we present a generic deep convolutional neural network (D...

2D and 3D Segmentation of uncertain local collagen fiber orientations in SHG microscopy

Collagen fiber orientations in bones, visible with Second Harmonic Gener...

Guided Collaborative Training for Pixel-wise Semi-Supervised Learning

We investigate the generalization of semi-supervised learning (SSL) to d...

DeepMIM: Deep Supervision for Masked Image Modeling

Deep supervision, which involves extra supervisions to the intermediate ...

Spatial Aggregation of Holistically-Nested Networks for Automated Pancreas Segmentation

Accurate automatic organ segmentation is an important yet challenging pr...

CompositeTasking: Understanding Images by Spatial Composition of Tasks

We define the concept of CompositeTasking as the fusion of multiple, spa...