Abstract
Existing level set models employ regularization based only on gradient information, 1D curvature or 2D curvature. For 3D image segmentation, however, an appropriate curvature-based regularization should involve a well-defined 3D curvature energy. This is the first paper to introduce a regularization energy that incorporates 3D scalar curvature for 3D image segmentation, inspired by the Einstein-Hilbert functional. To derive its Euler-Lagrange equation, we employ a two-step gradient descent strategy, alternately updating the level set function and its gradient. The paper also establishes the existence and uniqueness of the viscosity solution for the proposed model. Experimental results demonstrate that our proposed model outperforms other state-of-the-art models in 3D image segmentation.
Original language | English |
---|---|
Number of pages | 22 |
Journal | International Journal of Computer Vision |
Early online date | 15 Jun 2025 |
DOIs | |
Publication status | E-pub ahead of print - 15 Jun 2025 |
Funding
This research was supported by National Key R&D Program of China 2021YFA1003003; the National Natural Science Foundation of China T2225012; the program of China Scholarships Council 202307300007; the Engineering and Physical Sciences Research Council (EPSRC grant no. EP/X035883/1).
Keywords
- Riemannian geometry
- Ricci curvature tensor
- image segmentation
- variational mode