Related Experiment Videos
Cross-group bidirectional fusion network for 3D model classification
Xueyao Gao1, Yali Shao1, Chunxiang Zhang1
1School of Computer Science and Technology, Harbin University of Science and Technology, Harbin, 150080, China.
Abstract:
3D model classification is an important task in artificial intelligence, which aims to categorize 3D models based on shapes, structures, and other features. Focusing on the issues of insufficient inter-view interaction and inefficient feature fusion in some existing multi-view 3D model classification methods, this paper proposes a 3D model classification method based on Cross-group bidirectional fusion network. Firstly, dual-branch shared Swin Transformer feature extractor is combined with feature pyramid network to achieve multi-scale hierarchical feature fusion. Secondly, the cross-group interaction module constructs an attention-based semantic dependency matrix between the two view groups, enabling context-conditioned cross-view feature association. Thirdly, bidirectional feature-wise linear modulation strategy is proposed to adaptively calibrate cross-view feature responses by dynamically generating affine transformation parameters using the global semantics of the other view group. Fourthly, a stochastic hybrid data augmentation strategy is adopted during training to enhance regularization. Experimental results show that the proposed method achieves a classification accuracy of 95.58% on the ModelNet40 dataset, verifying the effectiveness of the proposed mechanism in facilitating efficient multi-view feature fusion.