🤖 AI Summary
Point cloud classification faces dual challenges in balancing local geometric feature extraction with model complexity, while existing Mamba architectures—constrained by unidirectional state-space modeling—struggle to capture local spatial correlations inherent in unordered point clouds. To address this, we propose a Bidirectional Enhanced Mamba (BEM) architecture. Our method introduces geometric-aware local pooling and a geometry-feature coupling mechanism, along with a dual-path collaborative feature enhancer that overcomes the unidirectional limitation of standard state-space models (SSMs). We further incorporate parameter-free dynamic geometric information aggregation and cross-path feature mixing, significantly improving responsiveness to local geometric discontinuities and sparse structural signals. Evaluated on ModelNet40, BEM achieves 95.99% classification accuracy—setting a new state-of-the-art—while adding only 0.03M parameters.
📝 Abstract
The point cloud classification tasks face the dual challenge of efficiently extracting local geometric features while maintaining model complexity. The Mamba architecture utilizes the linear complexity advantage of state space models (SSMs) to overcome the computational bottleneck of Transformers while balancing global modeling capabilities. However, the inherent contradiction between its unidirectional dependency and the unordered nature of point clouds impedes modeling spatial correlation in local neighborhoods, thus constraining geometric feature extraction. This paper proposes Hybrid-Emba3D, a bidirectional Mamba model enhanced by geometry-feature coupling and cross-path feature hybridization. The Local geometric pooling with geometry-feature coupling mechanism significantly enhances local feature discriminative power via coordinated propagation and dynamic aggregation of geometric information between local center points and their neighborhoods, without introducing additional parameters. The designed Collaborative feature enhancer adopts dual-path hybridization, effectively handling local mutations and sparse key signals, breaking through the limitations of traditional SSM long-range modeling. Experimental results demonstrate that the proposed model achieves a new SOTA classification accuracy of 95.99% on ModelNet40 with only 0.03M additional.