Minutia reconstruction in sonar images with diffusion probabilistic models
Boyu Zhao,Qian Zhou,Lijun Huang,Qiang Zhang,Yalong Zhu,Junhu Ma
DOI: https://doi.org/10.1016/j.engappai.2024.108850
IF: 8
2024-06-21
Engineering Applications of Artificial Intelligence
Abstract:Underwater sonar images tend to carry specific types of noise due to the unique imaging and collection approach. Existing image denoising algorithms are mostly designed for Gaussian white noise in optical scenes, and will degrade when migrated to underwater scenes. In addition, this class of algorithms requires numerous pairs of sonar images that meet their requirements, which is virtually impossible. In an innovative first, we have designed a dedicated combination for submarine sonar scenarios, namely Rapid Diffusion Probabilistic Model (RDPM). Specifically, the primary framework of RDPM is a self-encoder based on diffusion model, whose training relies exclusively on sonar images so as to reduce the complexity of combination. Compared to current algorithms, RDPM incorporates two additional limiters (i.e., global limiter and local limiter) to constrain the denoising process, which reduces its uncontrollability. Furthermore, these two limiters are trained in a separate manner, which not only dilutes the fine-tuning cost and model size, making it easier to deploy, but also facilitates the migration to different sea areas. Adequate experiments demonstrate that our proposed method outperforms unpaired denoising models on several publicly available sonar image datasets, and excels in controllability and migratability. Our code is available at https://github.com/zbyhnu/LimitDiffusionModel.git .
automation & control systems,computer science, artificial intelligence,engineering, electrical & electronic, multidisciplinary