Recurrent graph optimal transport for learning 3D flow motion in particle tracking
Recurrent graph optimal transport for learning 3D flow motion in particle tracking
复制标题
粒子跟踪中学习三维流动运动的递归图最优传输
DOI:
10.1038/s42256-023-00648-y
复制
发表时间:
2023-05
影响因子:
23.8
通讯作者:
Jiaming Liang;Chao Xu;Shengze Cai
中科院分区:
文献类型:
--
作者:
Jiaming Liang;Chao Xu;Shengze Cai
Flow visualization technologies such as particle tracking velocimetry are broadly used for studying three-dimensional turbulent flow in natural and industrial processes. Despite the advances in three-dimensional acquisition techniques, it is challenging to develop motion estimation algorithms in particle tracking due to large particle displacements, dense particle distributions and high computational cost. We present an end-to-end solution called graph optimal transport (GotFlow3D) to learn the three-dimensional fluid flow motion from consecutive particle images. The proposed model uses a graph neural network to extract geometric features and to further enrich the particle representations. The extracted deep features are subsequently used to correspond particles between consecutive frames, and the flow motion is then iteratively updated with a recurrent neural network approach. Experimental evaluations—including assessments on numerical experiments and validations on real-world experiments—demonstrate that GotFlow3D achieves state-of-the-art performance compared with recently developed scene flow learners and particle tracking algorithms. We believe that the high accuracy, robustness and generalization ability of our method can provide deeper insight into the complex dynamics of many physical and biological systems.