Hyperbolic prototype rectification for few-shot 3D point cloud classification

被引:0
|
作者
Feng, Yuan-Zhi [1 ]
Lin, Shing-Ho J. [2 ]
Tang, Xuan [3 ]
Wang, Mu-Yu [1 ]
Zheng, Jian-Zhang [2 ]
He, Zi-Yao [1 ]
Pang, Zi-Yi [1 ]
Yang, Jian [4 ]
Chen, Ming-Song [1 ]
Wei, Xian [1 ]
机构
[1] East China Normal Univ, Software Engn Inst, Shanghai 200062, Peoples R China
[2] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
[3] East China Normal Univ, Sch Commun & Elect Engn, Shanghai 200062, Peoples R China
[4] Informat Engn Univ, Sch Geospatial Informat, Zhengzhou 450052, Henan, Peoples R China
基金
北京市自然科学基金;
关键词
Hyperbolic geometry; Few-shot learning; Point cloud classification; Prototype rectification; Feature enhancement;
D O I
10.1016/j.patcog.2024.111042
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot point cloud classification is a challenging task in 3D computer vision and has received widespread attention from researchers. Most of the works on deep learning models rely heavily on Euclidean spatial metrics. However, point cloud objects often have complex non-Euclidean geometric structures, with underlying inter/intra-class hierarchical structures, which are difficult to capture by current Euclidean-based deep learning models. Moreover, due to the lack of training samples, many few-shot learning methods often suffer from the overfitting problem. Given the Hyperbolic metric of non-Euclidean geometry offering hierarchical structural prior, as we assume to be able to assist FSL task, we propose Hyperbolic Prototype Rectification (HPR) HPR ) for few- shot point cloud classification, without requiring extra learnable parameter. Firstly, point clouds are embedded into hyperbolic space to better describe hierarchical similarity relationships in data. Secondly, the HPR utilizes hyperbolic spatial and distributional information to enhance the feature representation and improve the generalization capability, with more appropriate hyperbolic prototypes. The few-shot classification experiments and further ablation studies conducted on widely used point cloud datasets demonstrate the effectiveness of our method. On the real-world ScanObjectNN(-PB) datasets, the average classification accuracy outperforms the SOTA method by 2.08%(0.66%), respectively, indicating that the proposed HPR has great generalization capability and strong robustness against perturbed data. Our code is available at: https://github.com/JonathanUCAS/HPR.
引用
收藏
页数:12
相关论文
共 50 条
  • [1] Few-shot 3D Point Cloud Semantic Segmentation with Prototype Alignment
    Wei, Maolin
    PROCEEDINGS OF 2023 8TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING TECHNOLOGIES, ICMLT 2023, 2023, : 195 - 200
  • [2] A Closer Look at Few-Shot 3D Point Cloud Classification
    Chuangguan Ye
    Hongyuan Zhu
    Bo Zhang
    Tao Chen
    International Journal of Computer Vision, 2023, 131 : 772 - 795
  • [3] A Closer Look at Few-Shot 3D Point Cloud Classification
    Ye, Chuangguan
    Zhu, Hongyuan
    Zhang, Bo
    Chen, Tao
    INTERNATIONAL JOURNAL OF COMPUTER VISION, 2023, 131 (03) : 772 - 795
  • [4] Prototype distribution correction for few-shot point cloud classification
    Feng, Yuanzhi
    Xia, Yu
    Guo, Jielong
    Shao, Dongheng
    Zhang, Jianfeng
    Wei, Xian
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2024, 39 (09) : 1243 - 1254
  • [5] Generalized Few-Shot 3D Point Cloud Segmentation
    Yang, Shuqian
    Ding, Henhui
    Jiang, Xudong
    2024 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS, ISCAS 2024, 2024,
  • [6] Few-shot 3D Point Cloud Semantic Segmentation
    Zhao, Na
    Chua, Tat-Seng
    Lee, Gim Hee
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 8869 - 8878
  • [7] Enhancing Few-Shot 3D Point Cloud Semantic Segmentation through Bidirectional Prototype Learning
    Guo, Xuehang
    Hu, Hao
    Yang, Xiaoxi
    Deng, Yancong
    PROCEEDINGS OF 2023 9TH INTERNATIONAL CONFERENCE ON ROBOTICS AND ARTIFICIAL INTELLIGENCE, ICRAI 2023, 2023, : 7 - 16
  • [8] Rethinking Few-shot 3D Point Cloud Semantic Segmentation
    An, Zhaochong
    Sun, Guolei
    Liu, Yun
    Liu, Fayao
    Wu, Zongwei
    Wang, Dan
    Van Gool, Luc
    Belongie, Serge
    2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2024, 2024, : 3996 - 4006
  • [9] Noisy Few-shot 3D Point Cloud Scene Segmentation
    Huang, Hao
    Yuan, Shuaihang
    Wen, CongCong
    Hao, Yu
    Fang, Yi
    2024 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA 2024), 2024, : 11070 - 11077
  • [10] Crossmodal Few-shot 3D Point Cloud Semantic Segmentation
    Zhao, Ziyu
    Wu, Zhenyao
    Wu, Xinyi
    Zhang, Canyu
    Wang, Song
    PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, : 4760 - 4768