Reconstruction-Aware Prior Distillation for Semi-supervised Point Cloud Completion
Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. Main Track. Pages 726-735.
https://doi.org/10.24963/ijcai.2023/81
Keywords
- Computer Vision: CV: 3D computer vision
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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [Grandvalet and Bengio, 2004] Yves Grandvalet and Yoshua Bengio. Semi-supervised learning by entropy minimiza- tion. Advances in neural information processing systems, 17, 2004.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
- [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.
Full text extracted automatically from the PDF; figures, tables and equations may differ. The PDF is the version of record.
1 Introduction
Point cloud can represent the 3D geometric details of a scene or an object well with high resolution, hence attracting extensive research interests and becoming increasingly important in many applications, such as autonomous driving, robotics, and augmented reality. However, point clouds scanned from real-world sensors are always incomplete, irregular, and noisy, making many algorithms designed for point loud-based applications, such as 3D object detection [Zhu et al., 2020], unreliable. Therefore, point cloud completion, aiming at transforming a partially observed incomplete point cloud into its complete counterpart, becomes more and more popular in recent years. PCN [Yuan et al., 2018] is a pioneering work for deep learning-based point cloud completion, which utilizes PointNet [Qi et al., 2017] to learn global feature of the incomplete point cloud. Then, pointwise features of the complete point cloud are folded [Yang et al., 2018] from the global feature to generate the final complete point cloud. Due to its disability in extracting local feature, PCN fails to recover the geometric details of the object. Therefore, following works [Liu et al., 2020; Pan, 2020; Wang et al., 2020a; Xie et al., 2020; Yan et al., 2022;

(b) Semi-supervised Prior Distillation
Figure 1: An illustration of our main idea. We train our point cloud completion model in two stages. In training stage 1, reconstructionaware pretraining is introduced to learn the semantic prior. In training stage 2, the above prior is distilled into the completion network, and a self-supervised completion module is proposed to further improve the network's power. In our work, only a small number of paired training samples are required.
Yu et al., 2023; Cai et al., 2022; Wang et al., 2022] provide better completion results by preserving observed geometric details from the incomplete point shape using local features [Pan et al., 2021]. Though existing methods have achieved very promising performance, most of them require taking a large number of well-aligned complete-incomplete point cloud pairs to train their models, which is labor exhausted. In some wild scenarios, annotating the data is nearly impossible. Therefore, a question arises: can we train a point cloud completion model using less paired data?
The answer is absolutely yes. In this paper, we propose a novel deep learning model named RaPD for semi-supervised point cloud completion. RaPD achieves the point cloud completion goal by utilizing a small number of paired point cloud samples compared to previous methods. Hence, it is much more annotation friendly and gains a significant reduction in annotation cost. The intuition behind our method is that though paired samples are hard to collect, unpaired point clouds are easy to access. For example, manually designed clean complete point clouds from shape analysis datasets like ShapeNet [Chang et al., 2015], and noisy incomplete point clouds from autonomous driving datasets like KITTI [Geiger et al., 2013]. The semantic prior of a category of objects consisting in these unpaired point clouds is very available and we should take full advantage of it. To this end, we propose a two-stage training scheme for RaPD to achieve the goal. Fig. 1 illustrates the main idea of our method.
In training stage 1, a reconstruction-aware pretraining process is introduced. The insight behind this design is that we believe that completion-dependent features and object-shapedependent features can be decoupled and obtained by reconstructing a point cloud using different deep learning-based models. To this end, two auto-encoders are trained for complete and incomplete point clouds separately. The one trained using unpaired complete point clouds would characterize the completion-dependent features into its decoder and the latent global code Zc, while the other one trained using unpaired incomplete point clouds would characterize the object-shapedependent features into its encoder and latent global code Zi. The above-mentioned encoder, decoder, and latent code Zc are regarded as the semantic prior to the next training stage.
Then, in training stage 2, we distill the learned semantic prior into a point cloud completion network to enable semi-supervised training. Specifically, weights of the abovelearned encoder and decoder are adopted as a good initiation. Then, object-shape-dependent features and completiondependent features are utilized as a balanced signal to guide the network training process. Thereby, taking advantages of this kind of prior, the network can learn the ability of competing a point cloud very well utilizing only a small number of paired training samples (10% or below than full-supervised methods in this paper). What's more, to make usage of a large amount of unpaired incomplete point clouds, we further propose a self-supervised learning module. This module would benefit from a novel category-level latent prior and an adversarial training strategy, which further increases the performance of RaPD.
To the best of our knowledge, we are the first work to design semi-supervised models for the point cloud completion task. to verify the effectiveness of our method, extensive experiments are conducted on several widely used datasets. Experimental results show that our method can achieve excellent performance in both homologous and heterologous scenarios. When only a small number of paired training samples are available, RaPD achieves state-of-the-art performance. And our method even achieves comparable performance to fullysupervised methods. Our contribution can be summarized as: 1) We propose the first semi-supervised point cloud completion method RaPD, which achieves state-of-the-art performance in the data shortage setting. 2) We propose a two-stage training scheme, which consists of a reconstruction-aware pretraining process and a semi-supervised prior distillation process, enabling effective semi-supervised training. 3) We conduct extensive experiments to verify the effectiveness of our method in both homologous and heterologous scenarios. Contributions of each key design are also well-studied.
2 RelatedWork
2.1 Point Cloud Completion
Point cloud completion has obtained significant development in recent years. It plays an important role in autonomous driving, robotic grasping, and computer graphics. PCN [Yuan et al., 2018] pioneers the work of deep learning-based point cloud completion. It proposes a simple encoder-decoder network for dense and complete point set generation. Since only global features are utilized, PCN is not good at recovering the geometric details. Therefore, methods such as [Sarmad et al., 2019; Wang et al., 2020a; Tchapmi et al., 2019; Huang et al., 2020b] are proposed to learn local features better. For example, Wang et al. [Wang et al., 2020a] propose a coarse-to-fine strategy to synthesize the detailed object shapes by considering the local details of partial input with the global shape information together. There are also some methods [Wang et al., 2020b; Zhang et al., 2020; Xie et al., 2020] propose to learn better semantics for generating more complicated and cleaner complete point clouds. For example, SoftPoolNet [Wang et al., 2020b] proposes a soft pooling operation to adaptively extract better semantics features for point cloud generation. To learn finer details, VRCNet [Pan et al., 2021] proposes a variational framework that consists of a probabilistic modeling module and a relational enhancement module to improve performance. Though promising, these methods are limited in that they all require a large number of paired complete-incomplete point cloud pairs for training, which is cost unfriendly. To tackle the issue, Zhang et al. [Zhang et al., 2021] propose to use GANInversion [Xia et al., 2021] for unsupervised point cloud completion. Though unpaired samples are no longer required, time-consuming inference time optimization is needed during testing in [Zhang et al., 2021], making it in-efficient. Besides, in [Zhang et al., 2021], one model only works for a particular category, further limiting its broader application. Most similar to our work is [Chen et al., 2019], which also use an encoder-decoder structure for point cloud completion. Howecer, firstly, our approach is semi-supervised, whereas [Chen et al., 2019] is not. Secondly, we use a discriminator on incomplete point clouds with the help of a degradation module, while the [Chen et al., 2019] uses a GAN for mapping two spaces on the completion point cloud. Thirdly, we propose a semisupervised prior distillation technique that distills different features as prior, which is not discussed in t [Chen et al., 2019]. Finally, we address the case when pair-data is limited, which is not discussed in [Chen et al., 2019].
In contrast to all previous works, we propose a novel reconstruction-aware prior distillation method named RaPD for semi-supervised point cloud completion. To tackle the data shortage problem, RaPD takes full advantage of the semantic prior hidden in unpaired point clouds and distills the learned prior into the final completion network. Our method achieves superior performances than both previous supervised and unsupervised methods only taking a small number of paired point clouds for training.

2.2 Semi-supervised Learning
Semi-supervised learning is a mixed research task between supervised and unsupervised tasks [Chapelle et al., 2009; Zhu, 2005]. It targets predicting more accurately with the aid of unlabeled data than supervised learning that uses only labeled data [Yang et al., 2021]. Semi-supervised methods can be divided into generative methods [Denton et al., 2016; Dai et al., 2017], consistency regularization methods [Rasmus et al., 2015; Sajjadi et al., 2016], graph-based methods [Iscen et al., 2019; Chen et al., 2020], pseudo-labelling methods [Grandvalet and Bengio, 2004; Lee and others, 2013], and hybrid methods [Berthelot et al., 2019b; Berthelot et al., 2019a]. With the development of deep learning, semisupervised learning has been deployed into a wide range of downstream tasks. For example, Tang et al. [Tang et al., 2016] propose a semi-supervised object detection method using visual and semantic knowledge transfer. Zhou et al. [Zhou et al., 2021] introduce a recurrent multi-scale feature modulation approach for monocular depth estimation. Quali et al. [Ouali et al., 2020] present a method that utilizes cross-consistency training for semi-supervised semantic segmentation. In the field of point cloud processing, Huang et al. [Huang et al., 2020a] propose a fast and robust semisupervised approach for point cloud registration. Zhao et at. [Zhao et al., 2020] propose a self-ensembling technique in their work for semi-supervised 3D object detection. Cheng et al. [Cheng et al., 2021] propose to divide a point cloud into superpoints and build superpoint graphs for semi-supervised point cloud segmentation. Though semi-supervised learning has attracted massive attention, to our knowledge, there is no existing method tailored for point cloud completion.
In this paper, we propose RaPD, which is tailored for the point cloud completion task. In our work, discriminative semantic prior hidden in unpaired point clouds are distilled and utilized to achieve our semi-supervised training goal.
3 Method
3.1 Overview
Problem Statement: We aim at training a deep learning model that takes a incomplete point cloud Pi as input, it can output Pi's complete counterpart Pc. In this paper, in order to reduce the dependence on paired completeincomplete samples, we study the semi-supervised point cloud completion task. In our setting, there exists two training datasets Dpaired = {(P 1 i , P 1 c ), (P 2 i , P 2 c ), · · · , (P K i , P K c )} and Dunpaired = {P 1 i , P 2 i , · · · , P N i , P 1 c , P 2 c , · · · , P M c }, where K, N and M are number of point clouds. Point clouds in Dpaired are well-paired that can be utilized for supervised training, while point clouds in Dunpaired are unpaired samples with great diversity. A constrain between the two datasets is that K<<N(M), Which means Dpaired is small scale. The deep learning model should be trained utilizing Dpaired and Dunpaired. The expectation is that the final trained model can do point cloud completion excellently even though only a small number of training samples are labeled (paired).
Framework of the Approach: To solve the semi-supervised point cloud completion problem, we propose a novel method named RaPD. Fig. 2 illustrates the framework. RaPD adopts a two-stage training scheme. In the first stage, we conduct reconstruction-aware pretraining to learn the semantic prior from Dunpaired. Specifically, two auto-encoders are trained using the complete point clouds and incomplete point clouds, respectively. After trained, network weights and embedded latent codes of point clouds are regarded as the semantic prior, which would be exploited in the next stage. Then, in the second stage, we conduct semi-supervised prior distillation. This stage consists of a prior distillation module and a self-supervised completion module, both of which would take Dunpaired and Dpaired as training data and jointly learn an excellent model for the final point cloud completion task.
3.2 Stage 1: Reconstruction-Aware Pretraining
Since it is very hard to annotate complete-incomplete point cloud pairs in real-world scenarios, only a small number of paired point clouds (K in Dpaired) can be leveraged to train a model in the semi-supervised training setting. However, we argue that though paired data is rare, there are abundant unpaired point clouds that can be leveraged. For complete point clouds, there exist human-designed object datasets such as ShapeNet [Chang et al., 2015]. For incomplete point clouds, we can extract samples from the wild datasets such as KITTI [Geiger et al., 2013]. They consist of Dunpaired. We note here that the semantic characteristics in these unpaired point clouds are very valuable and should be exploited to benefit the semi-supervised point cloud completion task. Therefore, we propose to learn the semantic prior from them. In fact, there are two kinds of characteristics that would benefit our task: the completeness information of an object and the shape information of an object. We term the two characteristics as completion-dependent features and object-shape-dependent features respectively. And we find that reconstruction-aware pretraining can learn them effectively.
Learning completion-dependent features: To learn the completion-dependent features, we train an auto-encoder using the M complete point clouds in Dunpaired. As shown in Fig. 2, the auto-encoder consists of an encoder and a decoder. Taking a complete point cloud Pc as input, the encoder would embed it into a latent code Zc. Then, the decoder would take Zc as input and recover the shape of Pc, termed ˆ Pc. After training, the latent code Zc would contain the completeness information of Pc since ˆ Pc is recovered only utilizing Zc, the predicting of which is learned from massive complete point clouds without disturbed by any incomplete noise. Besides, since the decoder Dc works for recovering a completion point cloud, the network weights of the decoder would also contain rich implicit completeness-related information. Therefore, the decoder and Zc together consist of the completiondependent features. And we regard the latent code Zc and the decoder as part of the expected semantic prior.
Learning Object-Shape-Dependent Features: The objectshape-dependent features are learned in the similar way. The difference is that the auto-encoder is trained by using the N incomplete point clouds in Dunpaired . Similarly, after training, the encoder Ei and latent code Zi of the input point cloud would be very expressive in reflecting the shape of the object in an implicit way, i.e, they are actually the objectshape-dependent features that we expect to learn. It is also very straightforward to regard the two as part of the semantic prior. However, since Zi is generated by Ei, they are essentially redundant. Therefore, we only take the encoder Ei as the semantic prior. In another word, The encoder Ei is used to encode the input point cloud into a compact and low-dimensional feature representation, which contains the object-shape-dependent features. These features capture the overall shape information of the object, which we believe is crucial for accurate 3D point cloud completion.
3.3 Stage 2: Semi-supervised Prior Distillation
In this stage, we focus on training an encoder-decoder network to achieve the final point cloud completion goal taking Dpaired, Dunpaired and the semantic prior as training data. To do this, we design two modules. The first module is the distillation module, which mainly utilizes Dpaired as input and focuses on distilling information from the semantic prior to better train the model. The second module is the selfsupervised completion module, which mainly makes usage of the incomplete point clouds in Dunpaired to further benefit the network.
Prior Distillation: The distillation module quantifies the semantic prior into two categories. The first category is named weights distillation. In particular, we build the encoderdecoder network architecture to be the same as the autoencoders in training stage 1. And we use the weights of the trained Ei and Dc as the initialization of the encoder E and decoder D. Since weights of Ei and Dc are equipped with the ability of representing the shape-related information and completeness-related information, such a kind of distillation would flow very valuable prior information from training stage 1 to training stage 2. Besides, since Ei and Dc are already well capable of point cloud reconstruction, initiating the network using them would help the model coverage faster, further increasing the robustness of the model and avoiding the network from falling into local-minimal.
Besides distilling model weights, we also propose a features distillation step. Specifically, one observation is that since we initialize the decoder using Dc, if the latent code Z was the same as or at least be very similar to Zc, the decoder can nearly perfectly output a complete point cloud. The reason is natural that Dc is already trained for recovering a complete point cloud in stage 1 and can perform well if the latent code is sampled from the feature space of Zc. Therefore, to promote the model to learn such a kind of prior, we encourage the distribution of Z to be close to that of Zc during training. To achieve so, we adopt the Kullback-Leibler divergence as a loss function:
Then, for point cloud in Dpaired, we use the chamfer distance loss as the point cloud similarity loss.
where γ is a balance term.
| 1% | 2% | 5% | 10% | |||||
|---|---|---|---|---|---|---|---|---|
| CD | F1-score | CD | F1-score | CD | F1-score | CD | F1-score | |
| TopNet [Tchapmi et al., 2019] | 20.73 | 0.186 | 16.49 | 0.220 | 14.37 | 0.236 | 13.62 | 0.251 |
| PCN [Yuan et al., 2018] | 20.80 | 0.199 | 16.16 | 0.229 | 12.52 | 0.272 | 10.99 | 0.299 |
| VRCNet [Pan et al., 2021] | 30.65 | 0.214 | 20.36 | 0.252 | 13.36 | 0.319 | 9.38 | 0.415 |
| RaPD* | 15.19 | 0.263 | 12.80 | 0.287 | 10.90 | 0.315 | 10.63 | 0.325 |
| RaPD | 12.12 | 0.313 | 11.07 | 0.319 | 10.07 | 0.328 | 9.64 | 0.335 |
| Methods | CD | F1-score | |
|---|---|---|---|
| fully supervised | PCN [Yuan et al., 2018] | 9.77 | 0.320 |
| TopNet [Tchapmi et al., 2019] | 10.11 | 0.308 | |
| MSN [Liu et al., 2020] | 7.90 | 0.432 | |
| ECG [Pan, 2020] | 6.64 | 0.476 | |
| Wang et al. [Wang et al., 2020a] | 7.25 | 0.434 | |
| VRCNet [Pan et al., 2021] | 5.96 | 0.499 | |
| semi-supervised | RaPD* | 10.63 | 0.325 |
| RaPD | 9.64 | 0.335 |
Self-supervised completion: During training, beyond using Dpaired, we also take the incomplete point clouds in Dunpaired as training data. To take advantage of them, we propose a self-training strategy. Specifically, for point cloud Pi in Dunpaired, it is first fed into the network to predict a point cloud Pn. Then, since there is no ground truth for Pn, we can only make use of the input Pi as a training signal. To do so, we should first degrade Pn to Pi, termed ˆ Pi. For the degradation, we adopt the K-Mask Degradation Module in [Zhang et al., 2021]. Then, the loss function for this signal is set as the chamfer distance loss too:
Lcd,unpaired would encourage Pn to be as similar as its incomplete counterpart Pi as possible during training. However, Lcd,unpaired only supervises the similarity in a low-level manner, which is not powerful enough for the completion task. To further benefit the encoder-decoder network, we add a discriminator D to guide the training process. The discriminator plays the role of encouraging ˆ Pi to be as similar Pi as possible. It would predict a probability for each of its input, where close to 0 means it regards the point cloud as a fake incomplete one, while close to 1 means it regards the point cloud as a real incomplete one. In this way, the network would be forced to generate a more realistic ˆ Pi . Hence, the high-level shape semantics between Pn and Pi would also be aligned. The loss function for the discriminator is:
Though straightforward, only using Lcd,unpaired would cause over-fitting since there is no signal account for the completeness of Pn. To encourage Pn to also be as complete as possible, we use the following loss to make the distribution of the latent code Z fall into the completeness-related space:
where ˆ Zc is the category-level mean latent code of all complete training point clouds in training stage 1. This loss can also be regarded as distilling a part of semantic prior from stage 1, further taking advantage of Dunpaired.
For training stage 1. We train two auto-encoders to formulate the reconstruction-aware pretraining. The loss functions of training both networks are chamfer distance loss. For training stage 2, the loss function for the encoder-decoder network is the weighted sum of each component of the training process:
where λ1 to λ5 are balance terms. For the discriminator, it is optimized using Ld independently.
4 Experiment
4.1 Implementation Details
To verify the superiority of RaPD, we conduct experiments on three widely used public datasets: MVP [Pan et al., 2021], CRN [Wang et al., 2020a] and KITTI [Geiger et al., 2013]. We set our encoder-decoder network architecture to be the same as PCN [Yuan et al., 2018] for a fair comparison. The architecture of the discriminator follows the basic architecture of [Li et al., 2019]. The evaluation metrics are chamfer distance (CD) and the F1 score, following [Pan et al., 2021]. More details please refer to the SuppMat.
4.2 Main Results
Results on MVP dataset: We first compare our method with PCN [Yuan et al., 2018], TopNet [Tchapmi et al., 2019] and VRCNet [Pan et al., 2021] on the MVP dataset. Since there is no existing semi-supervised point cloud completion method, for a fair comparison, we re-train their model using the same number of paired point clouds to report their performance under our semi-supervised training setting. We train all models on the MVP training set and test them on the MVP testing set. Therefore, the training and testing datasets are homologous. Table 1 and Fig. 3 show the results. RaPD* is a version of RaPD that discard the self-supervised completion module during training. It can be seen that with the increase of paired training data, performances of all models are promoted. However, no matter whether we adopt the self-supervised completion module, our method outperforms all existing methods by a large margin. These results greatly demonstrate the advantage of our idea of distilling semantic prior from large-scale unpaired data over existing supervised methods. The performance gain of RaPD over RaPD* also demonstrates the superiority of the self-supervised completion module in improving the performance. We find when using 10% data, RaPD is outperformed by VRCNet, that is because RaPD is designed for small number of labeled data. Here 10% data reaches around 6000 paired point clouds, which breaks the small data rule in practice, so it would benefit VRCNet more. But note, in real world scenarios, it is hard to collect so many paired data. Some visualization results are shown in Fig. 4. From the visualization, it can be seen that RaPD can obviously recover better geometric details than PCN and RaPD*. For example, the leg of the complete chair predicted by RaPD is obviously more clear and more slender than the one predicted by PCN.
We also compare RaPD with existing state-of-the-art supervised methods. All compared methods are training using all the paired data in the training set while RaPD and RaPD* are only trained using 10% of paired data. The result is shown in Table 2. It can be seen that though we only use 10% of paired training data, our method can achieve comparable performance with fully-supervised methods. Note that RaPD even outperforms PCN and TopNet. As mentioned before, RaPD shares the same network architecture with PCN. However, it outperforms PCN using only a small number of data, which further demonstrates the superiority of our semisupervised training scheme.
Results on CRN: As mentioned before, the training data and testing data of training stage 1 and training stage 2 are homologous when we evaluate our method on the MVP dataset. To verify the performance of our method in the heterologous situation, we conduct experiments on CRN dataset. Specifically, we conduct the reconstruction-aware pretraining on the MVP training set and conduct the semi-supervised prior distillation on the CRN training set, and test the trained model on the CRN testing set. The results are shown in Table 3, Fig. 3, and Fig. 5. It can be seen that RaPD* and RaPD significantly outperform the competitors, benefited from our prior distillation design and the semi-supervised completion module. This experiment demonstrates that our method not only performs well in homologous scenario, but also generates good generalization ability in heterologous scenario. That is, when generalizing to a new domain, the model only needs fine-tune or retrain in a semi-supervised prior distillation way based on the learned semantic prior to achieve an excellent performance.


Results on KITTI: We also compare our method with TopNet [Tchapmi et al., 2019], PCN [Yuan et al., 2018], and VRCNet [Pan et al., 2021] in the wild KITTI dataset. Since there is no complete ground truth for objects in the KITTI dataset, we only show the qualitative comparison in Fig. 6. It can be seen that the complete point clouds generated by our method are much better than the mentioned three methods. For example, the rear of the car is more smooth than that of PCN, TopNet, and VRCNet. The results demonstrate that RaPD can generalize well in wide scenarios. With less training overhead and better performance, RaPD has the potential of benefiting downstream applications like autonomous driving.
4.3 Ablation Study
In this section, we conduct extensive ablation study to verify the effectiveness of our key designs. All experiments are conducted on the MVP dataset. 5% paired training samples are used for training. More ablation study can be found in the SuppMat.
| 1% | 2% | 5% | 10% | |||||
|---|---|---|---|---|---|---|---|---|
| CD | F1-score | CD | F1-score | CD | F1-score | CD | F1-score | |
| TopNet [Tchapmi et al., 2019] | 39.91 | 0.125 | 27.77 | 0.157 | 22.05 | 0.171 | 18.86 | 0.193 |
| PCN [Yuan et al., 2018] | 36.76 | 0.130 | 25.65 | 0.165 | 18.34 | 0.195 | 15.50 | 0.215 |
| VRCNet [Pan et al., 2021] | 52.28 | 0.115 | 32.24 | 0.137 | 18.56 | 0.215 | 13.13 | 0.274 |
| RaPD* | 26.35 | 0.182 | 20.92 | 0.206 | 15.98 | 0.229 | 14.21 | 0.246 |
| RaPD | 17.57 | 0.229 | 15.70 | 0.235 | 13.96 | 0.244 | 12.47 | 0.272 |

| RaPD* | RaPD | |||
|---|---|---|---|---|
| CD | F1-score | CD | F1-score | |
| w/o features distillation | 11.61 | 0.299 | 43.86 | 0.109 |
| w/o decoder distillation | 11.95 | 0.283 | 10.74 | 0.321 |
| w/o encoder distillation | 11.88 | 0.292 | 10.77 | 0.315 |
| w/o weights distillation | 12.52 | 0.272 | 11.66 | 0.309 |
| full model | 10.90 | 0.315 | 10.07 | 0.328 |
Impact of Prior Distillation: In our work, we present two kinds of prior distillation for the final completion network: the weights distillation (including encoder weights distillation and decoder weights distillation) and the features distillation. The two kinds of distillations are both one of the most critical factors for the success of our semi-supervised point cloud completion network. Table 4 shows the impact of them. For features distillation, when it is discarded, the performance drops a lot, which demonstrates that it is very necessary to encourage the distribution of latent code of the input incomplete point cloud to be similar to that of the large scale unpaired complete point clouds. Then, we find that both the encoder weights distillation and decoder weights distillation are beneficial to our completion network. Since any of them is removed, the performance will significantly drop. When both of them are removed, the performance drop is even more serious. In a world, it can be found that both kinds of distillation play a significant role in ensuring the final completion performance. The absence of any of them would lead to a sharp performance decrease. Impact of Self-Supervised Completion: In our work, we introduce a self-supervised completion module to make use of the abundant unpaired incomplete point clouds in training stage 2. The performance difference between the RaPD* and RaPD has demonstrated that this module is very important for performance gain since it helps the model learn more information from a large scale of incomplete unpaired point clouds in an effective way. In this module, we use a degradation method to get a degraded point cloud for loss computation. We find that the choice of the degradation method we use is one of the key components that may affect the performance. In Table 6, we study the impact of several different degradation methods. It can be found that the K-Mask method is the best choice. The reason may be that K-Mask can better preserve the geometric details of point cloud after degradation. What's more, we use a discriminator for adversarial training to further increase the realism of the predicted point cloud. In Table 5, we illustrate the performance of discarding the discriminator and the corresponding adversarial loss. It can be found that without them, the performance drops a lot. Therefore, it is necessary to adopt the discriminator during training.
| RaPD | ||
|---|---|---|
| CD | F1-score | |
| w/o discriminator | 11.83 | 0.307 |
| full model | 10.07 | 0.328 |
| RaPD | ||
|---|---|---|
| CD | F1-score | |
| random downsample | 82.29 | 0.037 |
| voxel mask | 10.81 | 0.312 |
| tau mask | 14.46 | 0.311 |
| k mask | 10.07 | 0.328 |

[Cai et al., 2022] Yingjie Cai, Kwan-Yee Lin, Chao Zhang, Qiang Wang, Xiaogang Wang, and Hongsheng Li. Learning a structured latent space for unsupervised point cloud completion. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 55435553, 2022.
[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.
[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.
[Chen et al., 2019] Xuelin Chen, Baoquan Chen, and Niloy J Mitra. Unpaired point cloud completion on real scans using adversarial training. arXiv preprint arXiv:1904.00069, 2019.
[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.
[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.
[Dai et al., 2017] Zihang Dai, Zhilin Yang, Fan Yang, William W Cohen, and Russ R Salakhutdinov. Good semisupervised learning that requires a bad gan. Advances in neural information processing systems, 30, 2017.
[Denton et al., 2016] Emily Denton, Sam Gross, and Rob Fergus. Semi-supervised learning with contextconditional generative adversarial networks. arXiv preprint arXiv:1611.06430, 2016.
5 Conclusion
Point cloud completion is an important but challenging task for many real-world applications. To reduce the dependency on paired data when training a point cloud completion model, this paper proposes a novel semi-supervised method named RaPD. A two-stage training scheme is introduced in RaPD. The first stage learns powerful semantic prior through reconstruction-aware pretraining, while the second stage learns the final completion deep model by prior distillation and self-supervised completion learning. Experiments on several widely used datasets have demonstrated the superiority of RaPD over previous supervised methods. In the future, we will research efficient unsupervised point cloud completion to further ease the training requirement.
Acknowledgements
This work was supported in part by the National Key Research and Development Program of China under Grant No. 2020YFB2104101 and the National Natural Science Foundation of China (NSFC) under Grant Nos. 62172421, and 62072459.
Contribution Statement
Zhaoxin Fan and Yulin He contributed equally to this work. Jun He is the corresponding author.
@inproceedings{ijcai2023p81,
title = {Reconstruction-Aware Prior Distillation for Semi-supervised Point Cloud Completion},
author = {Fan, Zhaoxin and He, Yulin and Wang, Zhicheng and Wu, Kejian and Liu, Hongyan and He, Jun},
booktitle = {Proceedings of the Thirty-Second International Joint Conference on
Artificial Intelligence, {IJCAI-23}},
publisher = {International Joint Conferences on Artificial Intelligence Organization},
editor = {Edith Elkind},
pages = {726--735},
year = {2023},
month = {8},
note = {Main Track},
doi = {10.24963/ijcai.2023/81},
url = {https://doi.org/10.24963/ijcai.2023/81},
}
TY - CPAPER TI - Reconstruction-Aware Prior Distillation for Semi-supervised Point Cloud Completion AU - Fan, Zhaoxin AU - He, Yulin AU - Wang, Zhicheng AU - Wu, Kejian AU - Liu, Hongyan AU - He, Jun T2 - Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence, IJCAI-23 PB - International Joint Conferences on Artificial Intelligence Organization PY - 2023 SP - 726 EP - 735 DO - 10.24963/ijcai.2023/81 UR - https://doi.org/10.24963/ijcai.2023/81 ER -Copied
Machine-readable versions of this paper, for tools and AI agents: