Self-Supervised Reinforcement Learning for Active Object Detection
- 21 July 2022
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Robotics and Automation Letters
- Vol. 7 (4) , 10224-10231
Abstract
Active object detection (AOD) offers significant advantage in expanding the perceptual capacity of a robotics system. AOD is formulated as a sequential action decision process to determine optimal viewpoints to identify objects of interest in a visual scene. While reinforcement learning (RL) has been successfully used to solve many AOD problems, conventional RL methods suffer from (i) sample inefficiency, and (ii) unstable outcome due to inter-dependencies of action type (direction of view change) and action range (step size of view change). To address these issues, we propose a novel self-supervised RL method, which employs self-supervised representations of viewpoints to initialize the policy network, and a self-supervised loss on action range to enhance the network parameter optimization. The output and target pairs of self-supervised learning loss are automatically generated from the policy network online prediction and a range shrinkage algorithm (RSA), respectively. The proposed method is evaluated and benchmarked on two public datasets (T-LESS and AVD) using on-policy and off-policy RL algorithms. The results show that our method enhances detection accuracy and achieves faster convergence on both datasets. By evaluating on a more complex environment with a larger state space (where viewpoints are more densely sampled), our method achieves more robust and stable performance. Our experiment on real robot application scenario to disambiguate similar objects in a cluttered scene has also demonstrated the effectiveness of the proposed method.Keywords
Funding Information
- Agency for Science, Technology and Research (A18A2b0046)
This publication has 23 references indexed in Scilit:
- Extreme Trust Region Policy Optimization for Active Object RecognitionIEEE Transactions on Neural Networks and Learning Systems, 2018
- Multi-view 6D Object Pose Estimation and Camera Motion Planning Using RGBD Images2017
- Split-Brain Autoencoders: Unsupervised Learning by Cross-Channel Prediction2017
- A dataset for developing and benchmarking active vision2017
- T-LESS: An RGB-D Dataset for 6D Pose Estimation of Texture-Less Objects2017
- Deep active object recognition by joint label and action predictionComputer Vision and Image Understanding, 2017
- Faster R-CNN: Towards Real-Time Object Detection with Region Proposal NetworksIEEE Transactions on Pattern Analysis and Machine Intelligence, 2016
- Deep Residual Learning for Image Recognition2016
- Unsupervised Visual Representation Learning by Context Prediction2015
- Information theoretic sensor data selection for active object recognition and state estimationIEEE Transactions on Pattern Analysis and Machine Intelligence, 2002