1. Diversified Point Cloud Classification Using Personalized Federated Learning
- Author
-
Jianzong Wang, Anshun Xue, Xinghua Zhu, and Xiao Jing
- Subjects
Independent and identically distributed random variables ,Information privacy ,Artificial neural network ,Computer science ,Point cloud ,Data Protection Act 1998 ,Unstructured data ,Data mining ,Object (computer science) ,computer.software_genre ,computer ,Data modeling - Abstract
As a comprehensive and unstructured data source describing 3 dimensional (3D) objects, point cloud data has attracted wide interest from researchers. Neural networks (NNs) provide flexible, end-to-end solutions to point cloud processing tasks. Yet, one of the unavoidable problems in training the NNs is the difficulty in collecting and labeling high-quality point cloud data. Existing point cloud datasets are often collected at different scales and densities. Directly combining highly heterogeneous datasets might violate data protection regulations, but also lead to degradation of model performance. In our experiments, naive horizontal federated learning has been applied to alleviate the data privacy concern. However, when the datasets are not identically distributed, the federated model's performance falls far behind the state-of-the-art of centralized models. This paper investigates the union of highly heterogeneous point cloud datasets for the object classification task. In the proposed framework, the federated model is split into shared and personalized modules. The former improves generalizability by bringing multiple data sources into play, while the latter adapts local distribution and avoids convergence at false optima. Experiments show that the proposed framework FedPCN achieves 35.1% higher accuracy than FedAvg. At local test samples, FedPCN also outperforms the locally trained model by 0.5%.
- Published
- 2021
- Full Text
- View/download PDF