A Tiny Machine Learning Model for Point Cloud Object Classification

Min Zhang,Jintang Xue,Pranav Kadam,Hardik Prajapati,Shan Liu,C.-C. Jay Kuo
2023-03-20
Abstract:The design of a tiny machine learning model, which can be deployed in mobile and edge devices, for point cloud object classification is investigated in this work. To achieve this objective, we replace the multi-scale representation of a point cloud object with a single-scale representation for complexity reduction, and exploit rich 3D geometric information of a point cloud object for performance improvement. The proposed solution is named Green-PointHop due to its low computational complexity. We evaluate the performance of Green-PointHop on ModelNet40 and ScanObjectNN two datasets. Green-PointHop has a model size of 64K parameters. It demands 2.3M floating-point operations (FLOPs) to classify a ModelNet40 object of 1024 down-sampled points. Its classification performance gaps against the state-of-the-art DGCNN method are 3% and 7% for ModelNet40 and ScanObjectNN, respectively. On the other hand, the model size and inference complexity of DGCNN are 42X and 1203X of those of Green-PointHop, respectively.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The paper aims to address the problem of point cloud object classification and design a small machine learning model that can be deployed on mobile and edge devices. Specifically, the paper proposes a new method called Green-PointHop, which reduces complexity by decreasing multi-scale representations and leverages the rich 3D geometric information of point cloud objects to improve performance. The design goal of Green-PointHop is to achieve a model with low computational complexity, making it suitable for advanced tasks and applications in real-time 3D systems. The paper evaluates the performance of Green-PointHop on two datasets: ModelNet40 and ScanObjectNN. The results show that despite its model size being only 64K parameters and requiring only 2.3M floating-point operations (FLOPs) per classification operation, its classification performance is only 3% and 7% lower than the state-of-the-art DGCNN method on ModelNet40 and ScanObjectNN, respectively. In contrast, the model size and inference complexity of DGCNN are 42 times and 1203 times that of Green-PointHop, respectively. Overall, the paper aims to demonstrate that it is possible to design efficient point cloud object classifiers with high classification accuracy even under limited computational resources.