A plug-and-play framework for curvilinear structure segmentation based on a learned reconnecting regularization
arXiv:2408.12943 · doi:10.1016/j.neucom.2024.128055
Abstract
Curvilinear structures are present in various fields in image processing such as blood vessels in medical imaging or roads in remote sensing. Their detection is crucial for many applications. In this article, we propose an unsupervised plug-and-play framework for the segmentation of curvilinear structures that focuses on the preservation of their connectivity. This framework includes an algorithm for generating realistic pairs of connected/disconnected curvilinear structures and a reconnecting regularization operator that can be learned from a synthetic dataset. Once learned, this regularization operator can be plugged into a variational segmentation scheme and used to segment curvilinear structure images without requiring annotations. We demonstrate the interest of our approach on the segmentation of vascular images both in 2D and 3D and compare its results with classic unsupervised and deep learning-based approach. Comparative evaluations against unsupervised classic and deep learning-based methods highlight the superior performance of our approach, showcasing remarkable improvements in preserving the connectivity of curvilinear structures (approximately 90% in 2D and 70% in 3D). We finally showcase the good generalizability behavior of our approach on two different applications : road cracks and porcine corneal cells segmentations.
References in corpus (9)
- Trainable Nonlinear Reaction Diffusion: A Flexible Framework for Fast and Effective Image Restoration
- Learned Primal-dual Reconstruction
- CNN-Based Projected Gradient Descent for Consistent Image Reconstruction
- A Topological Loss Function for Deep-Learning based Image Segmentation using Persistent Homology
- Learning Proximal Operators: Using Denoising Networks for Regularizing Inverse Imaging Problems
- Topology-Preserving Deep Image Segmentation
- Local Intensity Order Transformation for Robust Curvilinear Object Segmentation
- Curvilinear object segmentation in medical images based on ODoS filter and deep learning network
- Cascaded multitask U-Net using topological loss for vessel segmentation and centerline extraction