Saved in:
Bibliographic Details
Main Authors: Zhang, Yushan, Wandt, Bastian, Magnusson, Maria, Felsberg, Michael
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2403.05327
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914964450574336
author Zhang, Yushan
Wandt, Bastian
Magnusson, Maria
Felsberg, Michael
author_facet Zhang, Yushan
Wandt, Bastian
Magnusson, Maria
Felsberg, Michael
contents Scene flow estimation is an essential ingredient for a variety of real-world applications, especially for autonomous agents, such as self-driving cars and robots. While recent scene flow estimation approaches achieve a reasonable accuracy, their applicability to real-world systems additionally benefits from a reliability measure. Aiming at improving accuracy while additionally providing an estimate for uncertainty, we propose DiffSF that combines transformer-based scene flow estimation with denoising diffusion models. In the diffusion process, the ground truth scene flow vector field is gradually perturbed by adding Gaussian noise. In the reverse process, starting from randomly sampled Gaussian noise, the scene flow vector field prediction is recovered by conditioning on a source and a target point cloud. We show that the diffusion process greatly increases the robustness of predictions compared to prior approaches resulting in state-of-the-art performance on standard scene flow estimation benchmarks. Moreover, by sampling multiple times with different initial states, the denoising process predicts multiple hypotheses, which enables measuring the output uncertainty, allowing our approach to detect a majority of the inaccurate predictions. The code is available at https://github.com/ZhangYushan3/DiffSF.
format Preprint
id arxiv_https___arxiv_org_abs_2403_05327
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DiffSF: Diffusion Models for Scene Flow Estimation
Zhang, Yushan
Wandt, Bastian
Magnusson, Maria
Felsberg, Michael
Computer Vision and Pattern Recognition
Scene flow estimation is an essential ingredient for a variety of real-world applications, especially for autonomous agents, such as self-driving cars and robots. While recent scene flow estimation approaches achieve a reasonable accuracy, their applicability to real-world systems additionally benefits from a reliability measure. Aiming at improving accuracy while additionally providing an estimate for uncertainty, we propose DiffSF that combines transformer-based scene flow estimation with denoising diffusion models. In the diffusion process, the ground truth scene flow vector field is gradually perturbed by adding Gaussian noise. In the reverse process, starting from randomly sampled Gaussian noise, the scene flow vector field prediction is recovered by conditioning on a source and a target point cloud. We show that the diffusion process greatly increases the robustness of predictions compared to prior approaches resulting in state-of-the-art performance on standard scene flow estimation benchmarks. Moreover, by sampling multiple times with different initial states, the denoising process predicts multiple hypotheses, which enables measuring the output uncertainty, allowing our approach to detect a majority of the inaccurate predictions. The code is available at https://github.com/ZhangYushan3/DiffSF.
title DiffSF: Diffusion Models for Scene Flow Estimation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.05327