Diffusion-Shock Filtering on the Space of Positions and Orientations
Finn M. Sherry, Kristina Schaefer, Remco Duits
TL;DR
This work extends Regularised Diffusion-Shock Filtering (RDS) from the Euclidean plane to the space of positions and orientations $\mathbb{M}_2 = \mathbb{R}^2 \times S^1$, enabling crossing-preserving denoising and inpainting via an orientation-score lifting with cake wavelets. It introduces two PDE-based variants on $\mathbb{M}_2$—left-invariant RDS and gauge-frame RDS—and analyzes the associated generalized Laplacians: in the left-invariant setting the diffusion is governed by the Lie-Cartan/Laplace-Beltrami operators, while the gauge-frame case yields a data-driven Lie-Cartan Laplacian. Theoretical results show equivalence of the Lie-Cartan and LB operators in the unimodular setting but reveal a discrepancy in the gauge-frame construction, and experiments demonstrate superior PSNR performance and crossing restoration on $\mathbb{M}_2$ compared to TR-TV, BM3D, and NLM. The work provides publicly available code and paves the way for integrating crossing-aware diffusion-shock processing with geometric PDEs and PDE-based deep learning for multi-orientation image analysis.
Abstract
We extend Regularised Diffusion-Shock (RDS) filtering from Euclidean space $\mathbb{R}^2$ to the space of positions and orientations $\mathbb{M}_2 := \mathbb{R}^2 \times S^1$. This has numerous advantages, e.g. making it possible to enhance and inpaint crossing structures, since they become disentangled when lifted to $\mathbb{M}_2$. We create a version of the algorithm using gauge frames to mitigate issues caused by lifting to a finite number of orientations. This leads us to study generalisations of diffusion, since the gauge frame diffusion is not generated by the Laplace-Beltrami operator. RDS filtering compares favourably to existing techniques such as Total Roto-Translational Variation (TR-TV) flow, NLM, and BM3D when denoising images with crossing structures, particularly if they are segmented. Additionally, we see that $\mathbb{M}_2$ RDS inpainting is indeed able to restore crossing structures, unlike $\mathbb{R}^2$ RDS inpainting.
