MHSAN: Multi-view hierarchical self-attention network for 3D shape recognition

被引:5
作者
Cao, Jiangzhong [1 ]
Yu, Lianggeng [1 ]
Ling, Bingo Wing-Kuen [1 ]
Yao, Zijie [1 ]
Dai, Qingyun [2 ,3 ]
机构
[1] Guangdong Univ Technol, Sch Informat Engn, Guangzhou 510006, Peoples R China
[2] Guangdong Polytech Normal Univ, Guangzhou 510665, Peoples R China
[3] Guangdong Prov Key Lab Intellectual Property & Big, Guangzhou 510665, Peoples R China
基金
中国国家自然科学基金;
关键词
3D shape recognition; Self-attention; Multi-view learning; View aggregation;
D O I
10.1016/j.patcog.2024.110315
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Multi-view learning has demonstrated promising performance for 3D shape recognition. However, existing multi-view methods usually focus on fusing multiple views and ignore the structural and discriminative information carried by 2D views. In this paper, we propose a multi-view hierarchical self-attention network (MHSAN) to explore the geometric and discriminative information from complex 2D views. Specifically, MHSAN consists of two self-attention networks. First, a global self-attention network is adopted to exploit the structure information by embedding position information of views. Then, the discriminative self-attention network learns discriminative information from the views with high classification scores. Through the proposed MHSAN, the geometric and discriminative information is condensed as the novel representation of 3D shapes. To validate the effectiveness of our proposed method, extensive experiments have been conducted on three 3D shape benchmarks. Experimental results demonstrate that our method is generally superior to the state-of-the-art methods in 3D shape classification and retrieval tasks.
引用
收藏
页数:12
相关论文
共 40 条
[31]  
Vaswani A, 2017, ADV NEUR IN, V30
[32]   View-GCN: View-based Graph Convolutional Network for 3D Shape Analysis [J].
Wei, Xin ;
Yu, Ruixuan ;
Sun, Jian .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1847-1856
[33]  
Wu ZR, 2015, PROC CVPR IEEE, P1912, DOI 10.1109/CVPR.2015.7298801
[34]   Point clouds learning with attention-based graph convolution networks [J].
Xie, Zhuyang ;
Chen, Junzhou ;
Peng, Bo .
NEUROCOMPUTING, 2020, 402 :245-255
[35]   Enhancing 2D Representation via Adjacent Views for 3D Shape Retrieval [J].
Xu, Cheng ;
Li, Zhaoqun ;
Qiu, Qiang ;
Leng, Biao ;
Jiang, Jingfei .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3731-3739
[36]   Multi-View 3D Shape Recognition via Correspondence-Aware Deep Learning [J].
Xu, Yong ;
Zheng, Chaoda ;
Xu, Ruotao ;
Quan, Yuhui ;
Ling, Haibin .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :5299-5312
[37]   Learning Relationships for Multi-View 3D Object Recognition [J].
Yang, Ze ;
Wang, Liwei .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :7504-7513
[38]   Meta-learning-based adversarial training for deep 3D face recognition on point clouds [J].
Yu, Cuican ;
Zhang, Zihui ;
Li, Huibin ;
Sun, Jian ;
Xu, Zongben .
PATTERN RECOGNITION, 2023, 134
[39]   Multi-view Harmonized Bilinear Network for 3D Object Recognition [J].
Yu, Tan ;
Meng, Jingjing ;
Yuan, Junsong .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :186-194
[40]   Learning directly from synthetic point clouds for "in-the-wild" 3D face recognition [J].
Zhang, Ziyu ;
Da, Feipeng ;
Yu, Yi .
PATTERN RECOGNITION, 2022, 123