Self2Seg: Single-Image Self-Supervised Joint Segmentation and Denoising (2309.10511v2)
Abstract: We develop Self2Seg, a self-supervised method for the joint segmentation and denoising of a single image. To this end, we combine the advantages of variational segmentation with self-supervised deep learning. One major benefit of our method lies in the fact, that in contrast to data-driven methods, where huge amounts of labeled samples are necessary, Self2Seg segments an image into meaningful regions without any training database. Moreover, we demonstrate that self-supervised denoising itself is significantly improved through the region-specific learning of Self2Seg. Therefore, we introduce a novel self-supervised energy functional in which denoising and segmentation are coupled in a way that both tasks benefit from each other. We propose a unified optimisation strategy and numerically show that for noisy microscopy images our proposed joint approach outperforms its sequential counterpart as well as alternative methods focused purely on denoising or segmentation.
- Convex variational methods on graphs for multiclass segmentation of high-dimensional data and point clouds. Journal of Mathematical Imaging and Vision, 58:468–493, 2017.
- A narrow band method for the convex formulation of discrete multilabel problems. Multiscale Modeling & Simulation, 8(5):2048–2078, 2010.
- Noise2Self: Blind denoising by self-supervision. In International Conference on Machine Learning, pages 524–533. PMLR, 2019.
- Self-similarity-based image denoising. Communications of the ACM, 54(5):109–117, 2011.
- DenoiSeg: joint denoising and segmentation. In Computer Vision–ECCV 2020 Workshops: Glasgow, UK, August 23–28, 2020, Proceedings, Part I, pages 324–337. Springer, 2021.
- Xiaohao Cai. Variational image segmentation model coupled with image restoration achievements. Pattern Recognition, 48(6):2029–2042, 2015.
- A two-stage image segmentation method using a convex variant of the Mumford–Shah model and thresholding. SIAM Journal on Imaging Sciences, 6(1):368–390, 2013.
- A first-order primal-dual algorithm for convex problems with applications to imaging. Journal of mathematical imaging and vision, 40:120–145, 2011.
- An active contour model without edges. In Scale-Space Theories in Computer Vision: Second International Conference, Scale-Space’99 Corfu, Greece, September 26–27, 1999 Proceedings 2, pages 141–151. Springer, 1999.
- Algorithms for finding global minimizers of image segmentation and denoising models. SIAM journal on applied mathematics, 66(5):1632–1648, 2006.
- Enhancing joint reconstruction and segmentation with non-convex Bregman iteration. Inverse Problems, 35(5):055001, 2019.
- Imre Csiszár. Information geometry and alternating minimization procedures. Statistics and Decisions, Dedewicz, 1:205–237, 1984.
- Super-resolution from a single image. In 2009 IEEE 12th international conference on computer vision, pages 349–356. IEEE, 2009.
- Lifting-based variational multiclass segmentation algorithm: Design, convergence analysis, and implementation with applications in medical imaging. Journal of Computational and Applied Mathematics, 439:115583, 2024.
- Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
- A review of image denoising and segmentation methods based on medical images. International Journal of Machine Learning and Computing, 9(3):288–295, 2019.
- Noise2Void-learning denoising from single noisy images. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 2129–2137, 2019.
- Noise2Noise: Learning image restoration without clean data. arXiv preprint arXiv:1803.04189, 2018.
- A fast blind zero-shot denoiser. Nature Machine Intelligence, 4(11):953–963, 2022.
- Implicit active contours driven by local binary fitting energy. In 2007 IEEE conference on computer vision and pattern Recognition, pages 1–7. IEEE, 2007.
- A three-stage variational image segmentation framework incorporating intensity inhomogeneity information. SIAM Journal on Imaging Sciences, 13(3):1692–1715, 2020.
- A survey on deep learning in medical image analysis. Medical image analysis, 42:60–88, 2017.
- When image denoising meets high-level vision tasks: A deep learning approach. arXiv preprint arXiv:1706.04284, 2017.
- Signal dependent noise removal from a single image. In 2014 IEEE International Conference on Image Processing (ICIP), pages 2679–2683. IEEE, 2014.
- Variational multi-phase segmentation using high-dimensional local features. In 2016 IEEE Winter Conference on Applications of Computer Vision (WACV), pages 1–9. IEEE, 2016.
- Optimal approximations by piecewise smooth functions and associated variational problems. Communications on pure and applied mathematics, 1989.
- Robust noise region-based active contour model via local similarity factor for image segmentation. Pattern Recognition, 61:104–119, 2017.
- Leveraging self-supervised denoising for image segmentation. In 2020 IEEE 17th international symposium on biomedical imaging (ISBI), pages 428–432. IEEE, 2020.
- A Mumford-Shah level-set approach for the inversion and segmentation of X-ray tomography data. Journal of Computational Physics, 221(2):539–557, 2007.
- Variational methods in imaging, volume 167. Springer, 2009.
- Paul Tseng. Convergence of a block coordinate descent method for nondifferentiable minimization. Journal of optimization theory and applications, 109(3):475, 2001.
- Deep image prior. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 9446–9454, 2018.
- scikit-image: image processing in Python. PeerJ, 2:e453, 6 2014.
- Active contours without level sets. In 2012 19th IEEE International Conference on Image Processing, pages 2549–2552. IEEE, 2012.
- An improved variational level set method for MR image segmentation and bias field correction. Magnetic Resonance Imaging, 31(3):439–447, 2013.
- Active contours with selective local or global segmentation: a new formulation and level set method. Image and Vision computing, 28(4):668–676, 2010.
- Internal statistics of a single natural image. In CVPR 2011, pages 977–984. IEEE, 2011.