Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

4DContrast: Contrastive Learning with Dynamic Correspondences for 3D Scene Understanding

About

We present a new approach to instill 4D dynamic object priors into learned 3D representations by unsupervised pre-training. We observe that dynamic movement of an object through an environment provides important cues about its objectness, and thus propose to imbue learned 3D representations with such dynamic understanding, that can then be effectively transferred to improved performance in downstream 3D semantic scene understanding tasks. We propose a new data augmentation scheme leveraging synthetic 3D shapes moving in static 3D environments, and employ contrastive learning under 3D-4D constraints that encode 4D invariances into the learned 3D representations. Experiments demonstrate that our unsupervised representation learning results in improvement in downstream 3D semantic segmentation, object detection, and instance segmentation tasks, and moreover, notably improves performance in data-scarce scenarios.

Yujin Chen, Matthias Nie{\ss}ner, Angela Dai• 2021

Related benchmarks

TaskDatasetResultRank
3D Object DetectionScanNet V2 (val)--
352
Semantic segmentationScanNet V2 (val)
mIoU2.3
288
Semantic segmentationScanNet (val)
mIoU72.3
231
3D Semantic SegmentationScanNet V2 (val)
mIoU72.3
171
3D Object DetectionScanNet--
123
3D Object DetectionSUN RGB-D--
104
Semantic segmentationS3DIS
mIoU61
88
3D Object DetectionSUN RGB-D v1 (val)--
81
3D Object DetectionSUN RGB-D (test)--
64
Instance SegmentationScanNetV2 (val)
mAP@0.54.2
58
Showing 10 of 16 rows

Other info

Follow for update