arXiv:2410.11934cs.CV2024-10NeurIPS被引 4

无需标签数据,用新损失函数实现更准的流体运动估计

Dual-frame Fluid Motion Estimation with Test-time Optimization and Zero-divergence Loss

  • 自监督学习+零散度损失,不依赖标注数据
  • 仅用1%样本即超越有监督方法性能
  • 支持测试时优化,跨域适应性强,适合真实流体分析

三维粒子跟踪测速(3D PTV)是分析湍流的关键技术,其核心为双帧流体运动估计算法。近年来深度学习方法虽取得高精度,但严重依赖大量标注数据。本文提出一种完全自监督的新方法,仅需先前方法1%的无标签训练样本,即可显著超越全监督模型。方法引入专用于湍流领域的零散度损失,并基于splat操作高效实现。自监督特性自然支持测试时优化,由此开发出动态测速增强模块(DVE)。在未见的留一合成域及真实物理/生物域上均表现出强跨域鲁棒性。代码、数据与模型详见https://github.com/Forrest-110/FluidMotionNet。

原文摘要 · Abstract (English)

3D particle tracking velocimetry (PTV) is a key technique for analyzing turbulent flow, one of the most challenging computational problems of our century. At the core of 3D PTV is the dual-frame fluid motion estimation algorithm, which tracks particles across two consecutive frames. Recently, deep learning-based methods have achieved impressive accuracy in dual-frame fluid motion estimation; however, they heavily depend on large volumes of labeled data. In this paper, we introduce a new method that is completely self-supervised and notably outperforms its fully-supervised counterparts while requiring only 1% of the training samples (without labels) used by previous methods. Our method features a novel zero-divergence loss that is specific to the domain of turbulent flow. Inspired by the success of splat operation in high-dimensional filtering and random fields, we propose a splat-based implementation for this loss which is both efficient and effective. The self-supervised nature of our method naturally supports test-time optimization, leading to the development of a tailored Dynamic Velocimetry Enhancer (DVE) module. We demonstrate that strong cross-domain robustness is achieved through test-time optimization on unseen leave-one-out synthetic domains and real physical/biological domains. Code, data and models are available at https://github.com/Forrest-110/FluidMotionNet.

流体分析自监督测试时优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。