A Label-Efficient Semi Self-Supervised Learning Framework for IoT Devices in Industrial Process

摘要

The industrial sector has experienced a tremendous advancement in deep supervised learning due to its representation ability, but it comes with high computing and labeled data demands. Recently, the demand for intelligent IoT devices on assembly and disassembly lines has surged. This necessitates algorithms that can use data to make intelligent decisions and a framework that can enable multiple IoT devices to learn collaboratively. Further, huge image generation using IoTs also needs an efficient data annotation scheme for classification problems. WeCollab is one such framework in federated learning that significantly reduces human efforts in data annotation with breakthroughs in self-supervised learning. The proposed framework is generic and can be adapted to any specific image data generated by industrial robots involved in assembly and disassembly lines. Our method outperforms supervised learning by 25% and 20% on the CIFAR-10 and CINIC-10 datasets, respectively, for the labeling task. We generate pseudo labels for the unlabeled part of the data and train a model to achieve 30% better test accuracy on CIFAR-10 and 20% better test accuracy on the CINIC-10 dataset as compared to supervised learning. Extensive experiments unveil the effectiveness of the method and proposed combination of loss functions used by WeCollab.