Volume 3, Issue 4
Unsupervised Deep Learning Meets Chan-Vese Model

Dihan Zheng, Chenglong Bao, Zuoqiang Shi, Haibin Ling & Kaisheng Ma

CSIAM Trans. Appl. Math., 3 (2022), pp. 662-691.

Published online: 2022-11

Export citation
  • Abstract

The Chan-Vese (CV) model is a classic region-based method in image segmentation. However, its piecewise constant assumption does not always hold for practical applications. Many improvements have been proposed but the issue is still far from well solved. In this work, we propose an unsupervised image segmentation approach that integrates the CV model with deep neural networks, which significantly improves the original CV model’s segmentation accuracy. Our basic idea is to apply a deep neural network that maps the image into a latent space to alleviate the violation of the piecewise constant assumption in image space. We formulate this idea under the classic Bayesian framework by approximating the likelihood with an evidence lower bound (ELBO) term while keeping the prior term in the CV model. Thus, our model only needs the input image itself and does not require pre-training from external datasets. Moreover, we extend the idea to multi-phase case and dataset based unsupervised image segmentation. Extensive experiments validate the effectiveness of our model and show that the proposed method is noticeably better than other unsupervised segmentation approaches.

  • AMS Subject Headings

68U10, 68T45

  • Copyright

COPYRIGHT: © Global Science Press

  • Email address
  • BibTex
  • RIS
  • TXT
@Article{CSIAM-AM-3-662, author = {Zheng , DihanBao , ChenglongShi , ZuoqiangLing , Haibin and Ma , Kaisheng}, title = {Unsupervised Deep Learning Meets Chan-Vese Model}, journal = {CSIAM Transactions on Applied Mathematics}, year = {2022}, volume = {3}, number = {4}, pages = {662--691}, abstract = {

The Chan-Vese (CV) model is a classic region-based method in image segmentation. However, its piecewise constant assumption does not always hold for practical applications. Many improvements have been proposed but the issue is still far from well solved. In this work, we propose an unsupervised image segmentation approach that integrates the CV model with deep neural networks, which significantly improves the original CV model’s segmentation accuracy. Our basic idea is to apply a deep neural network that maps the image into a latent space to alleviate the violation of the piecewise constant assumption in image space. We formulate this idea under the classic Bayesian framework by approximating the likelihood with an evidence lower bound (ELBO) term while keeping the prior term in the CV model. Thus, our model only needs the input image itself and does not require pre-training from external datasets. Moreover, we extend the idea to multi-phase case and dataset based unsupervised image segmentation. Extensive experiments validate the effectiveness of our model and show that the proposed method is noticeably better than other unsupervised segmentation approaches.

}, issn = {2708-0579}, doi = {https://doi.org/10.4208/csiam-am.SO-2021-0049}, url = {http://global-sci.org/intro/article_detail/csiam-am/21152.html} }
TY - JOUR T1 - Unsupervised Deep Learning Meets Chan-Vese Model AU - Zheng , Dihan AU - Bao , Chenglong AU - Shi , Zuoqiang AU - Ling , Haibin AU - Ma , Kaisheng JO - CSIAM Transactions on Applied Mathematics VL - 4 SP - 662 EP - 691 PY - 2022 DA - 2022/11 SN - 3 DO - http://doi.org/10.4208/csiam-am.SO-2021-0049 UR - https://global-sci.org/intro/article_detail/csiam-am/21152.html KW - Image segmentation, Chan-Vese model, variational inference, unsupervised learning. AB -

The Chan-Vese (CV) model is a classic region-based method in image segmentation. However, its piecewise constant assumption does not always hold for practical applications. Many improvements have been proposed but the issue is still far from well solved. In this work, we propose an unsupervised image segmentation approach that integrates the CV model with deep neural networks, which significantly improves the original CV model’s segmentation accuracy. Our basic idea is to apply a deep neural network that maps the image into a latent space to alleviate the violation of the piecewise constant assumption in image space. We formulate this idea under the classic Bayesian framework by approximating the likelihood with an evidence lower bound (ELBO) term while keeping the prior term in the CV model. Thus, our model only needs the input image itself and does not require pre-training from external datasets. Moreover, we extend the idea to multi-phase case and dataset based unsupervised image segmentation. Extensive experiments validate the effectiveness of our model and show that the proposed method is noticeably better than other unsupervised segmentation approaches.

Dihan Zheng, Chenglong Bao, Zuoqiang Shi, Haibin Ling & Kaisheng Ma. (2022). Unsupervised Deep Learning Meets Chan-Vese Model. CSIAM Transactions on Applied Mathematics. 3 (4). 662-691. doi:10.4208/csiam-am.SO-2021-0049
Copy to clipboard
The citation has been copied to your clipboard