Design and Implementation of Distributed Image Recognition App with Federal Learning Techniques

Abstract

In recent years, machine learning technology has been widely used in many fields, such as smart transportation, smart healthcare, smart finance and smart cities. Although machine learning technology has brought people a lot of convenience, the privacy problem of user data has also emerged [1]. Considering that users are not necessarily willing to upload personal privacy data to the cloud for deep learning training, therefore, instead of consuming a lot of bandwidth to upload data to the cloud, it is better to train on the local device and then use the model parameters obtained after training. (For example: weights and bias, etc.) upload to the server for aggregation. This emerging machine learning technology is called federated learning. In this way, the privacy and security of data can be guaranteed, and the purpose of decentralized learning can be achieved through aggregation. This study uses the architecture of federated learning technology and convolutional neural network algorithms to implement distributed image recognition mobile applications. This application allows users to use their mobile devices and the central servers for repeated training. After multiple rounds of repeated training, the convergence will be stabilized, and the accuracy will be significantly improved. At the same time, it can take into account privacy and achieve the machine the purpose of learning.