Non-IID federated learning via random exchange of local feature maps for textile IIoT secure computing

Bo Peng,Mingmin Chi,Chao Liu
DOI: https://doi.org/10.1007/s11432-021-3423-9
2022-01-01
Science China Information Sciences
Abstract:With the fast development of artificial intelligence (AI) and industrial Internet of Things (IIoT) technologies, it is challenging to deal with the problems of data privacy protection and secure computing. In recent years, federated learning (FL) is proposed to attack the challenges of learning shared models collaboratively while protecting security based on the data from cross-domain clients. However, data in the real environment is usually not independent and identically distributed (Non-IID) due to the differences in business, working environments, and data acquisition, and thus classic federated methods suffer from significant performance degradation. In this paper, a novel federated framework is proposed for secure textile fiber identification (FedTFI) via cross-domain texture representation based on high-definition fabric images. In addition to sharing the gradient of FedTFI, the local patch of feature maps between cross-domain clients is randomly exchanged to build a richer image texture feature distribution while protecting data security simultaneously for secure computing. Furthermore, a texture embedding layer is designed to provide a joint representation through similarity measure between triplet samples in low-dimensional space. To verify the effectiveness of the proposed framework, two textile image datasets, i.e., one public and the other we collected, are utilized to construct four Non-IID scenarios, including label skew, feature skew, and two combined skew scenarios. The experimental results confirm the effectiveness of our model to obtain better detection accuracies than benchmarks in four Non-IID scenarios by keeping data privacy for secure computing in fabric IIoT.
What problem does this paper attempt to address?