PIV-FlowDiffuser: Transfer-Learning-Based Denoising Diffusion Models for Particle Image Velocimetry.
Qianyu Zhu, Junjie Wang, Jeremiah Hu, Jia Ai, Yong Lee
Abstract
Open AccessDeep learning algorithms have significantly reduced the computational time and improved the spatial resolution of particle image velocimetry (PIV). However, the models trained on synthetic datasets might have degraded performances on practical particle images due to domain gaps. As a result, special residual patterns are often observed for the vector fields of deep learning-based estimators. To reduce the special noise step by step, we employ a denoising diffusion model (FlowDiffuser) for PIV analysis. And a data-hungry iterative denoising diffusion model is trained via a transfer learning strategy, resulting in our PIV-FlowDiffuser method. Specifically, we carry out the following: (1) pre-training a FlowDiffuser model with multiple optical flow datasets of the computer vision community, such as Sintel and KITTI; (2) fine-tuning the pre-trained model on synthetic PIV datasets. Note that the PIV images are upsampled by a factor of two to resolve small-scale turbulent flow structures. The visualized results indicate that our PIV-FlowDiffuser effectively suppresses the noise patterns. Therefore, the denoising diffusion model reduces the average endpoint error (AEE) by 59.4% over the RAFT256-PIV baseline on the classic Cai's dataset. In addition, PIV-FlowDiffuser exhibits enhanced generalization performance on unseen particle images due to transfer learning. Overall, this study highlights transfer-learning-based denoising diffusion models for PIV.