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
相关论文
共 44 条
  • [21] Sala F, 2018, PR MACH LEARN RES, V80
  • [22] Snell J, 2017, ADV NEUR IN, V30
  • [23] A Comprehensive Survey of Few-shot Learning: Evolution, Applications, Challenges, and Opportunities
    Song, Yisheng
    Wang, Ting
    Cai, Puyu
    Mondal, Subrota K.
    Sahoo, Jyoti Prakash
    [J]. ACM COMPUTING SURVEYS, 2023, 55 (13S)
  • [24] Multi-view Convolutional Neural Networks for 3D Shape Recognition
    Su, Hang
    Maji, Subhransu
    Kalogerakis, Evangelos
    Learned-Miller, Erik
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 945 - 953
  • [25] Sun J., 2022, ICLR 2022 WORKSH SOC
  • [26] Learning to Compare: Relation Network for Few-Shot Learning
    Sung, Flood
    Yang, Yongxin
    Zhang, Li
    Xiang, Tao
    Torr, Philip H. S.
    Hospedales, Timothy M.
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 1199 - 1208
  • [27] KPConv: Flexible and Deformable Convolution for Point Clouds
    Thomas, Hugues
    Qi, Charles R.
    Deschaud, Jean-Emmanuel
    Marcotegui, Beatriz
    Goulette, Francois
    Guibas, Leonidas J.
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 6420 - 6429
  • [28] Meta-learning approaches for learning-to-learn in deep learning: A survey
    Tian, Yingjie
    Zhao, Xiaoxi
    Huang, Wei
    [J]. NEUROCOMPUTING, 2022, 494 : 203 - 223
  • [29] Revisiting Point Cloud Classification: A New Benchmark Dataset and Classification Model on Real-World Data
    Uy, Mikaela Angelina
    Quang-Hieu Pham
    Binh-Son Hua
    Duc Thanh Nguyen
    Yeung, Sai-Kit
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 1588 - 1597
  • [30] Vinyals Oriol, 2016, Advances in Neural Information Processing Systems, V29