Saved in:
Bibliographic Details
Main Authors: He, Diya, Liu, Qingchen, Zhang, Cong, Qin, Jiahu
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2602.05257
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918323962249216
author He, Diya
Liu, Qingchen
Zhang, Cong
Qin, Jiahu
author_facet He, Diya
Liu, Qingchen
Zhang, Cong
Qin, Jiahu
contents Object pose estimation is a fundamental problem in computer vision and plays a critical role in virtual reality and embodied intelligence, where agents must understand and interact with objects in 3D space. Recently, score based generative models have to some extent solved the rotational symmetry ambiguity problem in category level pose estimation, but their efficiency remains limited by the high sampling cost of score-based diffusion. In this work, we propose a new framework, RFM-Pose, that accelerates category-level 6D object pose generation while actively evaluating sampled hypotheses. To improve sampling efficiency, we adopt a flow-matching generative model and generate pose candidates along an optimal transport path from a simple prior to the pose distribution. To further refine these candidates, we cast the flow-matching sampling process as a Markov decision process and apply proximal policy optimization to fine-tune the sampling policy. In particular, we interpret the flow field as a learnable policy and map an estimator to a value network, enabling joint optimization of pose generation and hypothesis scoring within a reinforcement learning framework. Experiments on the REAL275 benchmark demonstrate that RFM-Pose achieves favorable performance while significantly reducing computational cost. Moreover, similar to prior work, our approach can be readily adapted to object pose tracking and attains competitive results in this setting.
format Preprint
id arxiv_https___arxiv_org_abs_2602_05257
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle RFM-Pose:Reinforcement-Guided Flow Matching for Fast Category-Level 6D Pose Estimation
He, Diya
Liu, Qingchen
Zhang, Cong
Qin, Jiahu
Computer Vision and Pattern Recognition
Robotics
Object pose estimation is a fundamental problem in computer vision and plays a critical role in virtual reality and embodied intelligence, where agents must understand and interact with objects in 3D space. Recently, score based generative models have to some extent solved the rotational symmetry ambiguity problem in category level pose estimation, but their efficiency remains limited by the high sampling cost of score-based diffusion. In this work, we propose a new framework, RFM-Pose, that accelerates category-level 6D object pose generation while actively evaluating sampled hypotheses. To improve sampling efficiency, we adopt a flow-matching generative model and generate pose candidates along an optimal transport path from a simple prior to the pose distribution. To further refine these candidates, we cast the flow-matching sampling process as a Markov decision process and apply proximal policy optimization to fine-tune the sampling policy. In particular, we interpret the flow field as a learnable policy and map an estimator to a value network, enabling joint optimization of pose generation and hypothesis scoring within a reinforcement learning framework. Experiments on the REAL275 benchmark demonstrate that RFM-Pose achieves favorable performance while significantly reducing computational cost. Moreover, similar to prior work, our approach can be readily adapted to object pose tracking and attains competitive results in this setting.
title RFM-Pose:Reinforcement-Guided Flow Matching for Fast Category-Level 6D Pose Estimation
topic Computer Vision and Pattern Recognition
Robotics
url https://arxiv.org/abs/2602.05257