arXiv reaDer
半教師あり点群セマンティックセグメンテーションのためのガイド付き点対照学習
Guided Point Contrastive Learning for Semi-supervised Point Cloud Semantic Segmentation
3Dセマンティックセグメンテーションの急速な進歩は、トレーニングのために大規模な注釈付きデータに大きく依存するディープネットワークモデルの進歩と切り離せません。 3Dポイントレベルのラベリングの高コストと課題に対処するために、モデルのパフォーマンスを向上させるためのトレーニングで、ラベルなしのポイントクラウドを採用する半教師ありポイントクラウドセマンティックセグメンテーションの方法を紹介します。自己監視タスクにおける最近の対照的な損失に触発されて、半教師あり設定での特徴表現とモデルの一般化能力を強化するためのガイド付きポイントの対照的な損失を提案します。ラベル付けされていない点群のセマンティック予測は、同じカテゴリの負のペアを回避するための損失の疑似ラベルガイダンスとして機能します。また、高品質の特徴学習を保証するための信頼ガイダンスを設計します。さらに、クラスの不均衡の問題を軽減するために、正と負のサンプルを収集するために、カテゴリバランスのとれたサンプリング戦略が提案されています。 3つのデータセット(ScanNet V2、S3DIS、およびSemanticKITTI)での広範な実験は、ラベルのないデータで予測品質を改善するための半教師あり方法の有効性を示しています。
Rapid progress in 3D semantic segmentation is inseparable from the advances of deep network models, which highly rely on large-scale annotated data for training. To address the high cost and challenges of 3D point-level labeling, we present a method for semi-supervised point cloud semantic segmentation to adopt unlabeled point clouds in training to boost the model performance. Inspired by the recent contrastive loss in self-supervised tasks, we propose the guided point contrastive loss to enhance the feature representation and model generalization ability in semi-supervised setting. Semantic predictions on unlabeled point clouds serve as pseudo-label guidance in our loss to avoid negative pairs in the same category. Also, we design the confidence guidance to ensure high-quality feature learning. Besides, a category-balanced sampling strategy is proposed to collect positive and negative samples to mitigate the class imbalance problem. Extensive experiments on three datasets (ScanNet V2, S3DIS, and SemanticKITTI) show the effectiveness of our semi-supervised method to improve the prediction quality with unlabeled data.
updated: Fri Oct 15 2021 16:38:54 GMT+0000 (UTC)
published: Fri Oct 15 2021 16:38:54 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト