Image Restoration via Integration of Optimal Control Techniques and the Hamilton-Jacobi-Bellman Equation
Dragos-Patru Covei
TL;DR
The paper addresses image restoration under noise by formulating it as a stochastic optimal-control problem driven by the Hamilton–Jacobi–Bellman (HJB) framework. By assuming radial symmetry, the HJB is reduced to a nonlinear ODE in a single radius variable, solved via a shooting method to obtain a unique, decreasing value function $V(y)$ and the corresponding optimal feedback $p^*(y)$. The authors implement this approach with Python, extracting the optimal control from $\nabla V$ and simulating restorative diffusion, reporting significant improvements in PSNR and SSIM on degraded images. The work provides a principled, adaptive denoising mechanism that balances noise suppression with detail preservation, and it points to promising extensions such as multi-scale schemes and hybrid integrations with data-driven methods for real-time applicability.
Abstract
In this paper, we propose a novel image restoration framework that integrates optimal control techniques with the Hamilton-Jacobi-Bellman (HJB) equation. Motivated by models from production planning, our method restores degraded images by balancing an intervention cost against a state-dependent penalty that quantifies the loss of critical image information. Under the assumption of radial symmetry, the HJB equation is reduced to an ordinary differential equation and solved via a shooting method, from which the optimal feedback control is derived. Numerical experiments, supported by extensive parameter tuning and quality metrics such as PSNR and SSIM, demonstrate that the proposed framework achieves significant improvement in image quality. The results not only validate the theoretical model but also suggest promising directions for future research in adaptive and hybrid image restoration techniques.
