arrow
Return

EP-Net: Improving Point Cloud Learning Efficiency Through Feature Decoupling

delete2024-01-01
delete0
PRE
AI
H
Hao Deng
陈盛梅 cover
陈盛梅 (Shengmei Chen)
B
Bo Jiang
K
Kunlei Jing *
王琳 cover
王琳 (Lin Wang) *
DOI:10.1109/TIM.2024.3451587delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Point cloud analysis has been significantly influenced by the PointNet family, which has achieved remarkable accuracy across various benchmarks. However, enhancing the efficiency of these models without compromising performance remains a challenging task. In this article, we investigate the efficiency of the state-of-the-art (SOTA) point-based method PointNeXt without degrading its performance and propose a more efficient and powerful network named efficient PointNet (EP-Net). In particular, we propose a decoupled feature aggregation (DFA) module that decouples the learning of geometric and semantic features to capture more discriminative information, rather than the vanilla set abstraction (SA) module aggregating them for coarse joint processing. Besides, we develop a more effective sampling strategy for EP-Net, compressing processing complexity from O(n log(n)) to linear. It selects points nearest to the center of each nonempty grid cell as the new sampling points. Impressively, we achieve a dual increase in both speed and accuracy, attributed to the more refined feature aggregation and more efficient sampling. Furthermore, we demonstrate the extensibility of our proposal toward other point cloud networks, from the perspective either of boost efficiency or performance prompt. Conditioned on extensive experiments on point cloud classification and segmentation tasks, EP-Net reports an SOTA performance with 73.3% mean IoU on S3DIS Area5, boosting their baseline PointNeXt by 2.5%, with only 10.6% FLOPs and 37% parameters. On extensibility experiments, our strategy boosts PointMLP with an improvement of 0.9% mAcc on the ScanObjectNN dataset.
Keywords:
Classification
deep learning
efficient architecture
point cloud
segmentation

Journal

IEEE Transactions on Instrumentation and Measurement cover
IEEE Transactions on Instrumentation and Measurement
IF:
5.9
Papers:
1.9W
Citations:
5.8W

Organization

N
northwest university xi'an
Scholars:
1.8W
Papers: 1.2W
Citations: 22