Proceedings

Manifold-Aware Self-Training for Unsupervised Domain Adaptation on Regressing 6D Object Pose

Download BibTeX Cite

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. Main Track. Pages 1740-1748.

https://doi.org/10.24963/ijcai.2023/193

Abstract

Domain gap between synthetic and real data in visual regression (e.g., 6D pose estimation) is bridged in this paper via global feature alignment and local refinement on the coarse classification of discretized anchor classes in target space, which imposes a piece-wise target manifold regularization into domain-invariant representation learning. Specifically, our method incorporates an explicit self-supervised manifold regularization, revealing consistent cumulative target dependency across domains, to a self-training scheme (e.g., the popular Self-Paced Self-Training) to encourage more discriminative transferable representations of regression tasks. Moreover, learning unified implicit neural functions to estimate relative direction and distance of targets to their nearest class bins aims to refine target classification predictions, which can gain robust performance against inconsistent feature scaling sensitive to UDA regressors. Experiment results on three public benchmarks of the challenging 6D pose estimation task can verify the effectiveness of our method, consistently achieving superior performance to the state-of-the-art for UDA on 6D pose estimation. Codes and pre-trained models are available https://github.com/Gorilla-Lab-SCUT/MAST.

  1. [Bao et al., 2022] Yiwei Bao, Yunfei Liu, Haofei Wang, and Feng Lu. Generalizing gaze estimation with rotation con- sistency. In CVPR, 2022.
  2. [Brachmann et al., 2014] Eric Brachmann, Alexander Krull, Frank Michel, Stefan Gumhold, Jamie Shotton, and Carsten Rother. Learning 6d object pose estimation using 3d object coordinates. In ECCV, 2014.
  3. [Chen et al., 2011] Minmin Chen, Kilian Q Weinberger, and John Blitzer. Co-training for domain adaptation. NeurIPS, 2011.
  4. [Chen et al., 2013] Ke Chen, Shaogang Gong, Tao Xiang, and Chen Change Loy. Cumulative attribute space for age and crowd density estimation. In CVPR, 2013.
  5. [Chen et al., 2017] Xiaozhi Chen, Huimin Ma, Ji Wan, Bo Li, and Tian Xia. Multi-view 3d object detection net- work for autonomous driving. In CVPR, 2017.
  6. [Chen et al., 2021] Xinyang Chen, Sinan Wang, Jianmin Wang, and Mingsheng Long. Representation subspace dis- tance for domain adaptation regression. In ICML, 2021.
  7. [Chen et al., 2022a] Hansheng Chen, Pichao Wang, Fan Wang, Wei Tian, Lu Xiong, and Hao Li. Epro-pnp: Gen- eralized end-to-end probabilistic perspective-n-points for monocular object pose estimation. In CVPR, 2022.
  8. [Chen et al., 2022b] Yongwei Chen, Zihao Wang, Longkun Zou, Ke Chen, and Kui Jia. Quasi-balanced self-training on noise-aware synthesis of object point clouds for closing domain gap. In ECCV, 2022.
  9. [Collet et al., 2011] Alvaro Collet, Manuel Martinez, and Siddhartha S Srinivasa. The moped framework: Object recognition and pose estimation for manipulation. The in- ternational journal of robotics research, 2011.
  10. [Fischler and Bolles, 1981] Martin A Fischler and Robert C Bolles. Random sample consensus: a paradigm for model fitting with applications to image analysis and automated cartography. Communications of the ACM, 1981.
  11. [Gao et al., 2018] Ge Gao, Mikko Lauri, Jianwei Zhang, and Simone Frintrop. Occlusion resistant object rotation re- gression from point cloud segments. In ECCV Workshops, 2018.
  12. [Geiger et al., 2012] Andreas Geiger, Philip Lenz, and Raquel Urtasun. Are we ready for autonomous driving? the kitti vision benchmark suite. In CVPR, 2012.
  13. [Gopalan et al., 2011] Raghuraman Gopalan, Ruonan Li, and Rama Chellappa. Domain adaptation for object recog- nition: An unsupervised approach. In ICCV, 2011.
  14. [He et al., 2017] Kaiming He, Georgia Gkioxari, Piotr Dollár, and Ross Girshick. Mask r-cnn. In ICCV, 2017.
  15. [He et al., 2020] Yisheng He, Wei Sun, Haibin Huang, Jian- ran Liu, Haoqiang Fan, and Jian Sun. Pvn3d: A deep point-wise 3d keypoints voting network for 6dof pose es- timation. In CVPR, 2020.
  16. [He et al., 2021] Yisheng He, Haibin Huang, Haoqiang Fan, Qifeng Chen, and Jian Sun. Ffb6d: A full flow bidirec- tional fusion network for 6d pose estimation. In CVPR, 2021.
  17. [Hinterstoisser et al., 2012] Stefan Hinterstoisser, Vincent Lepetit, Slobodan Ilic, Stefan Holzer, Gary Bradski, Kurt Konolige, and Nassir Navab. Model based training, de- tection and pose estimation of texture-less 3d objects in heavily cluttered scenes. In ACCV, 2012.
  18. [Hodaň et al., 2016] Tomáš Hodaň, Jiří Matas, and Štěpán Obdržálek. On evaluation of 6d object pose estimation. In ECCV, 2016.
  19. [Hodaň et al., 2019] Tomáš Hodaň, Vibhav Vineet, Ran Gal, Emanuel Shalev, Jon Hanzelka, Treb Connell, Pedro Urbina, Sudipta N Sinha, and Brian Guenter. Photorealis- tic image synthesis for object instance detection. In ICIP, 2019.
  20. [Hodaň et al., 2020] Tomáš Hodaň, Martin Sundermeyer, Bertram Drost, Yann Labbé, Eric Brachmann, Frank Michel, Carsten Rother, and Jiří Matas. BOP challenge 2020 on 6D object localization. ECCV Workshops, 2020.
  21. [Kaskman et al., 2019] Roman Kaskman, Sergey Zakharov, Ivan Shugurov, and Slobodan Ilic. Homebreweddb: Rgb- d dataset for 6d pose estimation of 3d objects. In ICCV Workshops, 2019.
  22. [Kehl et al., 2017] Wadim Kehl, Fabian Manhardt, Federico Tombari, Slobodan Ilic, and Nassir Navab. Ssd-6d: Mak- ing rgb-based 3d detection and 6d pose estimation great again. In ICCV, 2017.
  23. [Kingma and Ba, 2014] Diederik P Kingma and Jimmy Ba. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
  24. [Labbé et al., 2020] Yann Labbé, Justin Carpentier, Mathieu Aubry, and Josef Sivic. Cosypose: Consistent multi-view multi-object 6d pose estimation. In ECCV, 2020.
  25. [Lee and others, 2013] Dong-Hyun Lee et al. Pseudo-label: The simple and efficient semi-supervised learning method for deep neural networks. In Workshop on challenges in representation learning, ICML, 2013.
  26. [Lee et al., 2022] Taeyeop Lee, Byeong-Uk Lee, Inkyu Shin, Jaesung Choe, Ukcheol Shin, In So Kweon, and Kuk-Jin Yoon. Uda-cope: Unsupervised domain adaptation for category-level object pose estimation. In CVPR, 2022.
  27. [Li et al., 2018a] Chi Li, Jin Bai, and Gregory D Hager. A unified framework for multi-view multi-class object pose estimation. In ECCV, 2018.
  28. [Li et al., 2018b] Yi Li, Gu Wang, Xiangyang Ji, Yu Xiang, and Dieter Fox. Deepim: Deep iterative matching for 6d pose estimation. In ECCV, 2018.
  29. [Li et al., 2019] Zhigang Li, Gu Wang, and Xiangyang Ji. Cdpn: Coordinates-based disentangled pose network for real-time rgb-based 6-dof object pose estimation. In ICCV, 2019.
  30. [Li et al., 2021] Zhigang Li, Yinlin Hu, Mathieu Salzmann, and Xiangyang Ji. Sd-pose: Semantic decomposition for cross-domain 6d object pose estimation. In AAAI, 2021.
  31. [Lin et al., 2022] Jiehong Lin, Zewei Wei, Changxing Ding, and Kui Jia. Category-level 6d object pose and size esti- mation using self-supervised deep prior deformation net- works. In ECCV, 2022.
  32. [Manhardt et al., 2018] Fabian Manhardt, Wadim Kehl, Nas- sir Navab, and Federico Tombari. Deep model-based 6d pose refinement in rgb. In ECCV, 2018.
  33. [Manhardt et al., 2019] Fabian Manhardt, Diego Martin Ar- royo, Christian Rupprecht, Benjamin Busam, Tolga Birdal, Nassir Navab, and Federico Tombari. Explaining the ambiguity of object detection and 6d pose from visual data. In ICCV, 2019.
  34. [Marchand et al., 2015] Eric Marchand, Hideaki Uchiyama, and Fabien Spindler. Pose estimation for augmented re- ality: a hands-on survey. IEEE transact. on visualization and computer graphics, 2015.
  35. [Park et al., 2019] Kiru Park, Timothy Patten, and Markus Vincze. Pix2pose: Pixel-wise coordinate regression of ob- jects for 6d pose estimation. In ICCV, 2019.
  36. [Peng et al., 2019] Sida Peng, Yuan Liu, Qixing Huang, Xi- aowei Zhou, and Hujun Bao. Pvnet: Pixel-wise voting network for 6dof pose estimation. In CVPR, 2019.
  37. [Roy et al., 2019] Subhankar Roy, Aliaksandr Siarohin, En- ver Sangineto, Samuel Rota Bulo, Nicu Sebe, and Elisa Ricci. Unsupervised domain adaptation using feature- whitening and consensus loss. In CVPR, 2019.
  38. [Sock et al., 2020] Juil Sock, Guillermo Garcia-Hernando, Anil Armagan, and Tae-Kyun Kim. Introducing pose con- sistency and warp-alignment for self-supervised 6d object pose estimation in color images. In 3DV, 2020.
  39. [Sohn et al., 2020] Kihyuk Sohn, David Berthelot, Nicholas Carlini, Zizhao Zhang, Han Zhang, Colin A Raffel, Ekin Dogus Cubuk, Alexey Kurakin, and Chun-Liang Li. Fixmatch: Simplifying semi-supervised learning with con- sistency and confidence. NeurIPS, 2020.
  40. [Su et al., 2015] Hao Su, Charles R Qi, Yangyan Li, and Leonidas J Guibas. Render for cnn: Viewpoint estima- tion in images using cnns trained with rendered 3d model views. In ICCV, 2015.
  41. [Sundermeyer et al., 2020] Martin Sundermeyer, Zoltan- Csaba Marton, Maximilian Durner, and Rudolph Triebel. Augmented autoencoders: Implicit 3d orientation learning for 6d object detection. IJCV, 2020.
  42. [Tang et al., 2012] Kevin Tang, Vignesh Ramanathan, Li Fei-Fei, and Daphne Koller. Shifting weights: Adapt- ing object detectors from image to video. NeurIPS, 2012.
  43. [Tekin et al., 2018] Bugra Tekin, Sudipta N Sinha, and Pas- cal Fua. Real-time seamless single shot 6d object pose prediction. In CVPR, 2018.
  44. [Van der Maaten and Hinton, 2008] Laurens Van der Maaten and Geoffrey Hinton. Visualizing data using t-sne. Journal of machine learning research, 2008.
  45. [Wang et al., 2020] Gu Wang, Fabian Manhardt, Jianzhun Shao, Xiangyang Ji, Nassir Navab, and Federico Tombari. Self6d: Self-supervised monocular 6d object pose estima- tion. In ECCV, 2020.
  46. [Wang et al., 2021a] Gu Wang, Fabian Manhardt, Xingyu Liu, Xiangyang Ji, and Federico Tombari. Occlusion- aware self-supervised monocular 6d object pose estima- tion. IEEE TPAMI, 2021.
  47. [Wang et al., 2021b] Gu Wang, Fabian Manhardt, Federico Tombari, and Xiangyang Ji. Gdr-net: Geometry-guided direct regression network for monocular 6d object pose es- timation. In CVPR, 2021.
  48. [Xiang et al., 2018] Yu Xiang, Tanner Schmidt, Venkatra- man Narayanan, and Dieter Fox. Posecnn: A convolu- tional neural network for 6d object pose estimation in clut- tered scenes. RSS, 2018.
  49. [Xie et al., 2020] Qizhe Xie, Minh-Thang Luong, Eduard Hovy, and Quoc V Le. Self-training with noisy student improves imagenet classification. In CVPR, 2020.
  50. [Xu et al., 2019] Zelin Xu, Ke Chen, and Kui Jia. W- posenet: Dense correspondence regularized pixel pair pose regression. arXiv preprint arXiv:1912.11888, 2019.
  51. [Xu et al., 2022] Yan Xu, Kwan-Yee Lin, Guofeng Zhang, Xiaogang Wang, and Hongsheng Li. Rnnpose: Recurrent 6-dof object pose refinement with robust correspondence field estimation and pose optimization. In CVPR, 2022.
  52. [Yang et al., 2021] Zongxin Yang, Xin Yu, and Yi Yang. Dsc-posenet: Learning 6dof object pose estimation via dual-scale consistency. In CVPR, 2021.
  53. [Yue et al., 2021] Xiangyu Yue, Zangwei Zheng, Shanghang Zhang, Yang Gao, Trevor Darrell, Kurt Keutzer, and Al- berto Sangiovanni Vincentelli. Prototypical cross-domain self-supervised learning for few-shot unsupervised domain adaptation. In CVPR, 2021.
  54. [Zakharov et al., 2019] Sergey Zakharov, Ivan Shugurov, and Slobodan Ilic. Dpod: 6d pose object detector and re- finer. In ICCV, 2019.
  55. [Zhang et al., 2021] Shaobo Zhang, Wanqing Zhao, Ziyu Guan, Xianlin Peng, and Jinye Peng. Keypoint-graph- driven learning framework for object pose estimation. In CVPR, 2021.
  56. [Zhou et al., 2019] Yi Zhou, Connelly Barnes, Jingwan Lu, Jimei Yang, and Hao Li. On the continuity of rotation representations in neural networks. In CVPR, 2019.
  57. [Zou et al., 2018] Yang Zou, Zhiding Yu, BVK Kumar, and Jinsong Wang. Unsupervised domain adaptation for se- mantic segmentation via class-balanced self-training. In ECCV, 2018.
  58. [Zou et al., 2021] Longkun Zou, Hui Tang, Ke Chen, and Kui Jia. Geometry-aware self-training for unsupervised domain adaptation on object point clouds. In ICCV, 2021.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards