Cosine Mixup: A Mixup Strategy for Point Cloud Contrastive Representation Learning

Guangping Li,Xiang Gao,Chenghui Liu,Dingkai Liang,Bingo Wing-Kuen Ling
DOI: https://doi.org/10.1109/tcsii.2023.3310192
2024-01-01
Abstract:Contrastive representation learning purposes to learn rich point cloud representations from unlabeled data. However, it currently lacks well-designed data augmentation, which can lead to poor regularization and generalization in the point cloud domain. Mixup has been shown to be a simple and effective data augmentation in the image domain. Yet, it was found ineffective to directly applying it to point cloud domain. In this brief, we propose a novel mixup strategy based on the cosine similarity algorithm in point cloud contrastive representation learning. First, we adaptively select pairs of samples with high dissimilarity in the mini-batch for mixing. The Principal Component Analysis (PCA) dimensionality reduction algorithm is used to reduce the cost when calculating the similarity. Then we introduce N-pair contrastive loss that fits to the virtual labels assigned for mixing. Our method outperforms other methods in a diverse range of downstream tasks including 3D object classification and few-shot object classification. Our method can be extended to manifold mixup for better regularization. Meanwhile, the ablation experiments demonstrate that our method is an effective data augmentation strategy.
What problem does this paper attempt to address?