# Mumford-Shah Loss Functional for Image Segmentation with Deep Learning

Source: [Mumford-Shah Loss Functional for Image Segmentation with Deep Learning](<https://www.sota2.com/research/paper/68fa66e11e209131cea12c59>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2025-10-25T22:00:21.133Z

Authors: Boah Kim, Jong Chul Ye

Published: 2019-04-05T05:17:18.000Z

## Abstract

Recent state-of-the-art image segmentation algorithms are mostly based on deep neural networks, thanks to their high performance and fast computation time. However, these methods are usually trained in a supervised manner, which requires large number of high quality ground-truth segmentation masks. On the other hand, classical image segmentation approaches such as level-set methods are formulated in a self-supervised manner by minimizing energy functions such as Mumford-Shah functional, so they are still useful to help generation of segmentation masks without labels. Unfortunately, these algorithms are usually computationally expensive and often have limitation in semantic segmentation. In this paper, we propose a novel loss function based on Mumford-Shah functional that can be used in deep-learning based image segmentation without or with small labeled data. This loss function is based on the observation that the softmax layer of deep neural networks has striking similarity to the characteristic function in the Mumford-Shah functional. We show that the new loss function enables semi-supervised and unsupervised segmentation. In addition, our loss function can be also used as a regularized function to enhance supervised semantic segmentation algorithms. Experimental results on multiple datasets demonstrate the effectiveness of the proposed method.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Medical Image Segmentation on ACDC (5-fold cross-validation)](<https://www.sota2.com/research/sota/medical-image-segmentation-on-acdc-5-fold-cross-validation.md>): Medical Image Segmentation; ACDC (5-fold cross-validation); Mean DSC: 0.839.

- [Medical Image Segmentation on ACDC (test)](<https://www.sota2.com/research/sota/medical-image-segmentation-on-acdc-test.md>): Medical Image Segmentation; ACDC (test); Avg DSC: 83.9.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/1904.02872>)

[PDF](<https://arxiv.org/pdf/1904.02872.pdf>)
