journal article·2025-02-25Open access

FFMT: Unsupervised RGB-D Point Cloud Registration via Fusion Feature Matching with Transformer

Jiacun Qiu, Zhenqi Han, Lizhaung Liu, Jialu Zhang

Applied Sciences· Volume 15· Issue 5

Abstract

Point cloud registration is a fundamental problem in computer vision and 3D computing, aiming to align point cloud data from different sensors or viewpoints into a unified coordinate system. In recent years, the rapid development of RGB-D sensor technology has greatly facilitated the acquisition of RGB-D data. In previous unsupervised point cloud registration methods based on RGB-D data, there has often been an overemphasis on matching local features, while the potential value of global information has been overlooked, thus limiting the improvement in registration performance. To address this issue, this paper proposes a self-attention-based global information attention module, which learns the global context of fused RGB-D features and effectively integrates global information into each individual feature. Furthermore, this paper introduces alternating self-attention and cross-attention layers, enabling the final feature fusion to achieve a broader global receptive field, thereby facilitating more precise matching relationships. We conduct extensive experiments on the ScanNet and 3DMatch datasets, and the results show that, compared to the previous state-of-the-art methods, our approach reduces the average rotation error by 26.9% and 32% on the ScanNet and 3DMatch datasets, respectively. Our method also achieves state-of-the-art performance on other key metrics.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: 3D Surveying and Cultural Heritage · 3D Shape Modeling and Analysis · Remote Sensing and LiDAR Applications

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Shanghai Advanced Research Institute, Chinese Academy of Sciences, Shanghai 201210, China

University of Chinese Academy of Sciences, Beijing 100049, China

Funding

Shanghai Science and Technology Innovation Project

References (50)

Show source-provided references
1. A method for registration of 3-D shapes10.1109/34.121791 ↗
2. Serafin, J., and Grisetti, G. (October, January 28). NICP: Dense normal based point cloud registration. Proceedings of the 2015 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), Hamburg, Germany.10.1109/IROS.2015.7353455 ↗
3. Segal, A., Haehnel, D., and Thrun, S. (July, January 28). Generalized-icp. Proceedings of the Robotics: Science and Systems, Seattle, WA, USA.
4. Park, J., Zhou, Q.Y., and Koltun, V. (2017, January 22–29). Colored Point Cloud Registration Revisited. Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy.10.1109/ICCV.2017.25 ↗
5. Go-ICP: A globally optimal solution to 3D ICP point-set registration10.1109/TPAMI.2015.2513405 ↗
6. Zhang, X., Yang, J., Zhang, S., and Zhang, Y. (2023, January 17–24). 3D registration with maximal cliques. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, BC, Canada.10.1109/CVPR52729.2023.01702 ↗
7. Zhong, Y. (October, January 27). Intrinsic shape signatures: A shape descriptor for 3D object recognition. Proceedings of the 2009 IEEE 12th International Conference on Computer Vision Workshops, ICCV Workshops, Kyoto, Japan.10.1109/ICCVW.2009.5457637 ↗
8. Rusu, R.B., Blodow, N., and Beetz, M. (2009, January 12–17). Fast point feature histograms (FPFH) for 3D registration. Proceedings of the 2009 IEEE International Conference on Robotics and Automation, Kobe, Japan.10.1109/ROBOT.2009.5152473 ↗
9. Choy, C., Park, J., and Koltun, V. (November, January 27). Fully convolutional geometric features. Proceedings of the IEEE/CVF International Conference on Computer Vision, Seoul, Republic of Korea.
10. Ao, S., Hu, Q., Yang, B., Markham, A., and Guo, Y. (2021, January 20–25). Spinnet: Learning a general surface descriptor for 3d point cloud registration. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.01158 ↗
11. Bai, X., Luo, Z., Zhou, L., Fu, H., Quan, L., and Tai, C.L. (2020, January 13–19). D3feat: Joint learning of dense detection and description of 3d local features. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.00639 ↗
12. Tombari, F., Salti, S., and Di Stefano, L. (2010, January 5–11). Unique signatures of histograms for local surface description. Proceedings of the Computer Vision–ECCV 2010: 11th European Conference on Computer Vision, Heraklion, Crete, Greece. Proceedings, Part III 11.10.1007/978-3-642-15558-1_26 ↗
13. Yew, Z.J., and Lee, G.H. (2022, January 18–24). REGTR: End-to-end Point Cloud Correspondences with Transformers. Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), New Orleans, LA, USA.10.1109/CVPR52688.2022.00656 ↗
14. Choy, C., Dong, W., and Koltun, V. (2020, January 13–19). Deep global registration. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.00259 ↗
15. Qin, Z., Yu, H., Wang, C., Guo, Y., Peng, Y., and Xu, K. (2022, January 18–24). Geometric transformer for fast and robust point cloud registration. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA.10.1109/CVPR52688.2022.01086 ↗
16. Wang, Y., and Solomon, J.M. (November, January 27). Deep closest point: Learning representations for point cloud registration. Proceedings of the IEEE/CVF International Conference on Computer Vision, Seoul, Republic of Korea.
17. Wang, Y., and Solomon, J.M. (2019, January 8–14). Prnet: Self-supervised learning for partial-to-partial registration. Proceedings of the 33rd International Conference on Neural Information Processing Systems, Vancouver, BC, Canada. Advances in Neural Information Processing Systems.
18. Yew, Z.J., and Lee, G.H. (2020, January 13–19). Rpm-net: Robust point matching using learned features. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.01184 ↗
19. Dai, A., Chang, A.X., Savva, M., Halber, M., Funkhouser, T., and Nießner, M. (2017, January 21–26). Scannet: Richly-annotated 3d reconstructions of indoor scenes. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.261 ↗
20. Zeng, A., Song, S., Nießner, M., Fisher, M., Xiao, J., and Funkhouser, T. (2017, January 21–26). 3dmatch: Learning local geometric descriptors from rgb-d reconstructions. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.29 ↗
21. Mei, G., Tang, H., Huang, X., Wang, W., Liu, J., Zhang, J., Van Gool, L., and Wu, Q. (2023, January 17–24). Unsupervised Deep Probabilistic Approach for Partial Point Cloud Registration. Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.01308 ↗
22. El Banani, M., and Johnson, J. (2021, January 10–17). Bootstrap your own correspondences. Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, QC, Canada.10.1109/ICCV48922.2021.00637 ↗
23. Wang, Z., Huo, X., Chen, Z., Zhang, J., Sheng, L., and Xu, D. (2022). Improving RGB-D Point Cloud Registration by Learning Multi-scale Local Linear Transformation. European Conference on Computer Vision, Springer.10.1007/978-3-031-19824-3_11 ↗
24. Yuan, M., Fu, K., Li, Z., Meng, Y., and Wang, M. (2023, January 1–6). PointMBF: A Multi-scale Bidirectional Fusion Network for Unsupervised RGB-D Point Cloud Registration. Proceedings of the 2023 IEEE/CVF International Conference on Computer Vision (ICCV), Paris, France.10.1109/ICCV51070.2023.01622 ↗
25. Vaswani, A. (2017, January 4–9). Attention is all you need. Proceedings of the 31st International Conference on Neural Information Processing Systems, Long Beach, CA, USA. Advances in Neural Information Processing Systems.
26. Sun, J., Shen, Z., Wang, Y., Bao, H., and Zhou, X. (2021, January 20–25). LoFTR: Detector-free local feature matching with transformers. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.00881 ↗
27. Low, K.L. (2004). Linear Least-Squares Optimization for Point-to-Plane ICP Surface Registration, University of North Carolina.
28. Random sample consensus: A paradigm for model fitting with applications to image analysis and automated cartography10.1145/358669.358692 ↗
29. Deng, H., Birdal, T., and Ilic, S. (2018, January 18–23). Ppfnet: Global context aware local features for robust 3d point matching. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00028 ↗
30. Li, Y., Yu, A.W., Meng, T., Caine, B., Ngiam, J., Peng, D., Shen, J., Lu, Y., Zhou, D., and Le, Q.V. (2022, January 18–24). Deepfusion: Lidar-camera deep fusion for multi-modal 3d object detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA.10.1109/CVPR52688.2022.01667 ↗
31. Vora, S., Lang, A.H., Helou, B., and Beijbom, O. (2020, January 13–19). Pointpainting: Sequential fusion for 3d object detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.00466 ↗
32. Jaritz, M., Gu, J., and Su, H. (2019, January 27–28). Multi-view pointnet for 3d scene understanding. Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops, Seoul, Republic of Korea.10.1109/ICCVW.2019.00494 ↗
33. Hui, T.W., and Ngan, K.N. (2014, January 27–30). Depth enhancement using RGB-D guided filtering. Proceedings of the 2014 IEEE International Conference on Image Processing (ICIP), Paris, France.10.1109/ICIP.2014.7025778 ↗
34. Deep bilateral learning for real-time image enhancement10.1145/3072959.3073592 ↗
35. Xia, X., Zhang, M., Xue, T., Sun, Z., Fang, H., Kulis, B., and Chen, J. (2020, January 23–28). Joint bilateral learning for real-time universal photorealistic style transfer. Proceedings of the Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK. Proceedings, Part VIII 16.10.1007/978-3-030-58598-3_20 ↗
36. He, Y., Sun, W., Huang, H., Liu, J., Fan, H., and Sun, J. (2020, January 13–19). Pvn3d: A deep point-wise 3d keypoints voting network for 6dof pose estimation. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.01165 ↗
37. Wu, Z., Gobichettipalayam, S., Tamadazte, B., Allibert, G., Paudel, D.P., and Demonceaux, C. (2022, January 12–16). Robust rgb-d fusion for saliency detection. Proceedings of the 2022 International Conference on 3D Vision (3DV), Prague, Czech Republic.10.1109/3DV57658.2022.00052 ↗
38. Qi, C.R., Yi, L., Su, H., and Guibas, L.J. (2017, January 4–9). Pointnet++: Deep hierarchical feature learning on point sets in a metric space. Proceedings of the 31st International Conference on Neural Information Processing Systems, Long Beach, CA, USA. Advances in Neural Information Processing Systems.
39. Wang, C., Xu, D., Zhu, Y., Martín-Martín, R., Lu, C., Fei-Fei, L., and Savarese, S. (2019, January 15–20). Densefusion: 6d object pose estimation by iterative dense fusion. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Long Beach, CA, USA.10.1109/CVPR.2019.00346 ↗
40. GMF: General Multimodal Fusion Framework for Correspondence Outlier Rejection10.1109/LRA.2022.3220148 ↗
41. Hu, W., Zhao, H., Jiang, L., Jia, J., and Wong, T.T. (2021, January 20–25). Bidirectional projection network for cross dimension scene understanding. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.01414 ↗
42. He, Y., Huang, H., Fan, H., Chen, Q., and Sun, J. (2021, January 20–25). Ffb6d: A full flow bidirectional fusion network for 6d pose estimation. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.00302 ↗
43. Ronneberger, O., Fischer, P., and Brox, T. (2015, January 5–9). U-net: Convolutional networks for biomedical image segmentation. Proceedings of the Medical image computing and computer-assisted intervention–MICCAI 2015: 18th International Conference, Munich, Germany. Proceedings, Part III 18.
44. Thomas, H., Qi, C.R., Deschaud, J.E., Marcotegui, B., Goulette, F., and Guibas, L. (November, January 27). KPConv: Flexible and Deformable Convolution for Point Clouds. Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision (ICCV), Seoul, Republic of Korea.10.1109/ICCV.2019.00651 ↗
45. Dong, H., Pan, J., Xiang, L., Hu, Z., Zhang, X., Wang, F., and Yang, M.H. (2020, January 13–19). Multi-scale boosted dehazing network with dense feature fusion. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.00223 ↗
46. He, K., and Zhang, X. (2016, January 27–30). Deep Residual Learning for Image Recognition. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
47. Luo, S., and Hu, W. (2021, January 20–25). Diffusion probabilistic models for 3d point cloud generation. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.00286 ↗
48. Distinctive image features from scale-invariant keypoints10.1023/B:VISI.0000029664.99615.94 ↗
49. DeTone, D., Malisiewicz, T., and Rabinovich, A. (2018, January 18–22). Superpoint: Self-supervised interest point detection and description. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, Salt Lake City, UT, USA.10.1109/CVPRW.2018.00060 ↗
50. Gojcic, Z., Zhou, C., Wegner, J.D., Guibas, L.J., and Birdal, T. (2020, January 13–19). Learning multiview 3d point cloud registration. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.00183 ↗