Research Area:  Machine Learning

Abstract:

In this paper, a new federated transfer learning framework, FTL-CDP, is proposed to address the challenges of data scarcity and data privacy faced by modern smart manufacturing with cross-domain applications. Existing applications can share their knowledge through the central server as base models, while new applications can convert a base model to their target domain models with limited application-specific data using transfer learning technique. Meanwhile, the federated learning scheme is deployed among smart devices within the same group to further enhance the accuracy of the application-specific model. The integrated framework allows model sharing across the central server and different smart devices without exposing any raw data, and hence protects data privacy. Two public datasets COCO and PETS2009, which represent the source and target applications, are employed for evaluations. The simulation results show that the proposed method outperforms two state-of-the-art machine learning approaches, by achieving better learning efficiency and accuracy.

Keywords:  

Author(s) Name:  Kevin I-Kai Wang; Xiaokang Zhou; Wei Liang; Zheng Yan; Jinhua She

Publisher name:  IEEE

DOI:  10.1109/TII.2021.3088057

Volume Information:  

Paper Link:   https://ieeexplore.ieee.org/document/9449957