Conv-MCD: A Plug-and-Play Multi-task Module for Medical Image Segmentation

by   Balamurali Murugesan, et al.

For the task of medical image segmentation, fully convolutional network (FCN) based architectures have been extensively used with various modifications. A rising trend in these architectures is to employ joint-learning of the target region with an auxiliary task, a method commonly known as multi-task learning. These approaches help impose smoothness and shape priors, which vanilla FCN approaches do not necessarily incorporate. In this paper, we propose a novel plug-and-play module, which we term as Conv-MCD, which exploits structural information in two ways - i) using the contour map and ii) using the distance map, both of which can be obtained from ground truth segmentation maps with no additional annotation costs. The key benefit of our module is the ease of its addition to any state-of-the-art architecture, resulting in a significant improvement in performance with a minimal increase in parameters. To substantiate the above claim, we conduct extensive experiments using 4 state-of-the-art architectures across various evaluation metrics, and report a significant increase in performance in relation to the base networks. In addition to the aforementioned experiments, we also perform ablative studies and visualization of feature maps to further elucidate our approach.



There are no comments yet.


page 6

page 7


Joint shape learning and segmentation for medical images using a minimalistic deep network

Recently, state-of-the-art results have been achieved in semantic segmen...

Self-Adaptive 2D-3D Ensemble of Fully Convolutional Networks for Medical Image Segmentation

Segmentation is a critical step in medical image analysis. Fully Convolu...

Psi-Net: Shape and boundary aware joint multi-task deep network for medical image segmentation

Medical image segmentation is a primary task in many applications, and t...

Histogram of Oriented Gradients Meet Deep Learning: A Novel Multi-task Deep Network for Medical Image Semantic Segmentation

We present our novel deep multi-task learning method for medical image s...

Critical Assessment of Transfer Learning for Medical Image Segmentation with Fully Convolutional Neural Networks

Transfer learning is widely used for training machine learning models. H...

Residual Moment Loss for Medical Image Segmentation

Location information is proven to benefit the deep learning models on ca...

Polygonal Building Segmentation by Frame Field Learning

While state of the art image segmentation models typically output segmen...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.