D3Feat: Joint Learning of Dense Detection and Description of 3D Local Features

TL;DR

D3Feat introduces a joint learning framework for dense detection and description of 3D local features using a 3D fully convolutional network, achieving state-of-the-art registration results and strong generalization across datasets.

Contribution

It proposes a novel joint learning mechanism for 3D feature detection and description, including a keypoint selection strategy and a self-supervised detector loss.

Findings

Achieves state-of-the-art registration accuracy on 3DMatch and KITTI datasets.

Demonstrates strong generalization to ETH dataset.

Enables accurate point cloud alignment with fewer features.

Abstract

A successful point cloud registration often lies on robust establishment of sparse matches through discriminative 3D local features. Despite the fast evolution of learning-based 3D feature descriptors, little attention has been drawn to the learning of 3D feature detectors, even less for a joint learning of the two tasks. In this paper, we leverage a 3D fully convolutional network for 3D point clouds, and propose a novel and practical learning mechanism that densely predicts both a detection score and a description feature for each 3D point. In particular, we propose a keypoint selection strategy that overcomes the inherent density variations of 3D point clouds, and further propose a self-supervised detector loss guided by the on-the-fly feature matching results during training. Finally, our method achieves state-of-the-art results in both indoor and outdoor scenarios, evaluated on 3DMatch and KITTI datasets, and shows its strong generalization ability on the ETH dataset. Towards practical use, we show that by adopting a reliable feature detector, sampling a smaller number of features is sufficient to achieve accurate and fast point cloud alignment.[code release](https://github.com/XuyangBai/D3Feat)