Proceedings

Reconstruction-Aware Prior Distillation for Semi-supervised Point Cloud Completion

Download BibTeX Cite

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

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

Abstract

Real-world sensors often produce incomplete, irregular, and noisy point clouds, making point cloud completion increasingly important. However, most existing completion methods rely on large paired datasets for training, which is labor-intensive. This paper proposes RaPD, a novel semi-supervised point cloud completion method that reduces the need for paired datasets. RaPD utilizes a two-stage training scheme, where a deep semantic prior is learned in stage 1 from unpaired complete and incomplete point clouds, and a semi-supervised prior distillation process is introduced in stage 2 to train a completion network using only a small number of paired samples. Additionally, a self-supervised completion module is introduced to improve performance using unpaired incomplete point clouds. Experiments on multiple datasets show that RaPD outperforms previous methods in both homologous and heterologous scenarios.

  1. [Berthelot et al., 2019a] David Berthelot, Nicholas Carlini, Ekin D Cubuk, Alex Kurakin, Kihyuk Sohn, Han Zhang, and Colin Raffel. Remixmatch: Semi-supervised learning with distribution alignment and augmentation anchoring. arXiv preprint arXiv:1911.09785, 2019.
  2. [Berthelot et al., 2019b] David Berthelot, Nicholas Carlini, Ian Goodfellow, Nicolas Papernot, Avital Oliver, and Colin A Raffel. Mixmatch: A holistic approach to semi- supervised learning. Advances in Neural Information Pro- cessing Systems, 32, 2019.
  3. [Cai et al., 2022] Yingjie Cai, Kwan-Yee Lin, Chao Zhang, Qiang Wang, Xiaogang Wang, and Hongsheng Li. Learn- ing a structured latent space for unsupervised point cloud completion. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 5543- 5553, 2022.
  4. [Chang et al., 2015] Angel X Chang, Thomas Funkhouser, Leonidas Guibas, Pat Hanrahan, Qixing Huang, Zimo Li, Silvio Savarese, Manolis Savva, Shuran Song, Hao Su, et al. Shapenet: An information-rich 3d model repository. arXiv preprint arXiv:1512.03012, 2015.
  5. [Chapelle et al., 2009] Olivier Chapelle, Bernhard Scholkopf, and Alexander Zien. Semi-supervised learning (chapelle, o. et al., eds.; 2006)[book reviews]. IEEE Transactions on Neural Networks, 20(3):542-542, 2009.
  6. [Chen et al., 2019] Xuelin Chen, Baoquan Chen, and Niloy J Mitra. Unpaired point cloud completion on real scans us- ing adversarial training. arXiv preprint arXiv:1904.00069, 2019.
  7. [Chen et al., 2020] Peibin Chen, Tao Ma, Xu Qin, Weidi Xu, and Shuchang Zhou. Data-efficient semi-supervised learning by reliable edge mining. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 9192-9201, 2020.
  8. [Cheng et al., 2021] Mingmei Cheng, Le Hui, Jin Xie, and Jian Yang. Sspc-net: Semi-supervised semantic 3d point cloud segmentation network. arXiv preprint arXiv:2104.07861, 2021.
  9. [Dai et al., 2017] Zihang Dai, Zhilin Yang, Fan Yang, William W Cohen, and Russ R Salakhutdinov. Good semi- supervised learning that requires a bad gan. Advances in neural information processing systems, 30, 2017.
  10. [Denton et al., 2016] Emily Denton, Sam Gross, and Rob Fergus. Semi-supervised learning with context- conditional generative adversarial networks. arXiv preprint arXiv:1611.06430, 2016.
  11. [Geiger et al., 2013] Andreas Geiger, Philip Lenz, Christoph Stiller, and Raquel Urtasun. Vision meets robotics: The kitti dataset. The International Journal of Robotics Re- search, 32(11):1231-1237, 2013.
  12. [Grandvalet and Bengio, 2004] Yves Grandvalet and Yoshua Bengio. Semi-supervised learning by entropy minimiza- tion. Advances in neural information processing systems, 17, 2004.
  13. [Huang et al., 2020a] Xiaoshui Huang, Guofeng Mei, and Jian Zhang. Feature-metric registration: A fast semi- supervised approach for robust point cloud registra- tion without correspondences. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 11366-11374, 2020.
  14. [Huang et al., 2020b] Zitian Huang, Yikuan Yu, Jiawen Xu, Feng Ni, and Xinyi Le. Pf-net: Point fractal network for 3d point cloud completion. arXiv preprint arXiv:2003.00410, 2020.
  15. [Iscen et al., 2019] Ahmet Iscen, Giorgos Tolias, Yannis Avrithis, and Ondrej Chum. Label propagation for deep semi-supervised learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 5070-5079, 2019.
  16. [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, volume 3, page 896, 2013.
  17. [Li et al., 2019] Ruihui Li, Xianzhi Li, Chi-Wing Fu, Daniel Cohen-Or, and Pheng-Ann Heng. Pu-gan: a point cloud upsampling adversarial network. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pages 7203-7212, 2019.
  18. [Liu et al., 2020] Minghua Liu, Lu Sheng, Sheng Yang, Jing Shao, and Shi-Min Hu. Morphing and sampling net- work for dense point cloud completion. In Proceedings of the AAAI conference on artificial intelligence, volume 34, pages 11596-11603, 2020.
  19. [Ouali et al., 2020] Yassine Ouali, Céline Hudelot, and Myr- iam Tami. Semi-supervised semantic segmentation with cross-consistency training. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 12674-12684, 2020.
  20. [Pan et al., 2021] Liang Pan, Xinyi Chen, Zhongang Cai, Junzhe Zhang, Haiyu Zhao, Shuai Yi, and Ziwei Liu. Vari- ational relational point completion network. In Proceed- ings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 8524-8533, 2021.
  21. [Pan, 2020] Liang Pan. Ecg: Edge-aware point cloud com- pletion with graph convolution. IEEE Robotics and Au- tomation Letters, 5(3):4392-4398, 2020.
  22. [Qi et al., 2017] Charles R Qi, Hao Su, Kaichun Mo, and Leonidas J Guibas. Pointnet: Deep learning on point sets for 3d classification and segmentation. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 652-660, 2017.
  23. [Rasmus et al., 2015] Antti Rasmus, Mathias Berglund, Mikko Honkala, Harri Valpola, and Tapani Raiko. Semi- supervised learning with ladder networks. Advances in neural information processing systems, 28, 2015.
  24. [Sajjadi et al., 2016] Mehdi Sajjadi, Mehran Javanmardi, and Tolga Tasdizen. Regularization with stochastic trans- formations and perturbations for deep semi-supervised learning. Advances in neural information processing sys- tems, 29, 2016.
  25. [Sarmad et al., 2019] Muhammad Sarmad, Hyunjoo Jenny Lee, and Young Min Kim. Rl-gan-net: A reinforcement learning agent controlled gan network for real-time point cloud shape completion. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 5898-5907, 2019.
  26. [Tang et al., 2016] Yuxing Tang, Josiah Wang, Boyang Gao, Emmanuel Dellandréa, Robert Gaizauskas, and Liming Chen. Large scale semi-supervised object detection using visual and semantic knowledge transfer. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 2119-2128, 2016.
  27. [Tchapmi et al., 2019] Lyne P Tchapmi, Vineet Kosaraju, Hamid Rezatofighi, Ian Reid, and Silvio Savarese. Top- net: Structural point cloud decoder. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pat- tern Recognition, pages 383-392, 2019.
  28. [Wang et al., 2020a] Xiaogang Wang, Marcelo H Ang Jr, and Gim Hee Lee. Cascaded refinement network for point cloud completion. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 790-799, 2020.
  29. [Wang et al., 2020b] Yida Wang, David Joseph Tan, Nas- sir Navab, and Federico Tombari. Softpoolnet: Shape descriptor for point cloud completion and classification. arXiv preprint arXiv:2008.07358, 2020.
  30. [Wang et al., 2022] Yida Wang, David Joseph Tan, Nassir Navab, and Federico Tombari. Softpool++: An encoder- decoder network for point cloud completion. International Journal of Computer Vision, 130(5):1145-1164, 2022.
  31. [Xia et al., 2021] Weihao Xia, Yulun Zhang, Yujiu Yang, Jing-Hao Xue, Bolei Zhou, and Ming-Hsuan Yang. Gan inversion: A survey. arXiv preprint arXiv:2101.05278, 2021.
  32. [Xie et al., 2020] Haozhe Xie, Hongxun Yao, Shangchen Zhou, Jiageng Mao, Shengping Zhang, and Wenxiu Sun. Grnet: Gridding residual network for dense point cloud completion. In European Conference on Computer Vision, pages 365-381. Springer, 2020.
  33. [Yan et al., 2022] Xuejun Yan, Hongyu Yan, Jingjing Wang, Hang Du, Zhihong Wu, Di Xie, Shiliang Pu, and Li Lu. Fbnet: Feedback network for point cloud completion. In Computer Vision-ECCV 2022: 17th European Confer- ence, Tel Aviv, Israel, October 23-27, 2022, Proceedings, Part II, pages 676-693. Springer, 2022.
  34. [Yang et al., 2018] Yaoqing Yang, Chen Feng, Yiru Shen, and Dong Tian. Foldingnet: Point cloud auto-encoder via deep grid deformation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 206-215, 2018.
  35. [Yang et al., 2021] Xiangli Yang, Zixing Song, Irwin King, and Zenglin Xu. A survey on deep semi-supervised learn- ing. arXiv preprint arXiv:2103.00550, 2021.
  36. [Yu et al., 2023] Xumin Yu, Yongming Rao, Ziyi Wang, Ji- wen Lu, and Jie Zhou. Adapointr: Diverse point cloud completion with adaptive geometry-aware transformers. arXiv preprint arXiv:2301.04545, 2023.
  37. [Yuan et al., 2018] Wentao Yuan, Tejas Khot, David Held, Christoph Mertz, and Martial Hebert. Pcn: Point com- pletion network. In 2018 International Conference on 3D Vision (3DV), pages 728-737. IEEE, 2018.
  38. [Zhang et al., 2020] Wenxiao Zhang, Qingan Yan, and Chunxia Xiao. Detail preserved point cloud comple- tion via separated feature aggregation. arXiv preprint arXiv:2007.02374, 2020.
  39. [Zhang et al., 2021] Junzhe Zhang, Xinyi Chen, Zhongang Cai, Liang Pan, Haiyu Zhao, Shuai Yi, Chai Kiat Yeo, Bo Dai, and Chen Change Loy. Unsupervised 3d shape completion through gan inversion. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 1768-1777, 2021.
  40. [Zhao et al., 2020] Na Zhao, Tat-Seng Chua, and Gim Hee Lee. Sess: Self-ensembling semi-supervised 3d object de- tection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 11079- 11087, 2020.
  41. [Zhou et al., 2021] Zhongkai Zhou, Xinnan Fan, Pengfei Shi, and Yuanxue Xin. R-msfm: Recurrent multi-scale feature modulation for monocular depth estimating. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pages 12777-12786, 2021.
  42. [Zhu et al., 2020] Xinge Zhu, Yuexin Ma, Tai Wang, Yan Xu, Jianping Shi, and Dahua Lin. Ssn: signature networks for multi-class object detection from point clouds. In Eu- ropean Conference on Computer Vision, pages 581-597. Springer, 2020.
  43. [Zhu, 2005] Xiaojin Jerry Zhu. Semi-supervised learning literature survey. Technical Report 1530, University of Wisconsin-Madison Department of Computer Sciences, 2005.

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

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