Towards Transmission-Friendly and Robust CNN Models over Cloud and Device

TL;DR

This paper introduces a cloud-assisted CNN training framework that reduces parameter transmission and enhances robustness by using nonlearnable filters generated through filter generation functions, suitable for resource-limited IoT devices.

Contribution

The proposed framework employs MonoCNN with nonlearnable filters generated by rules, significantly reducing transmission and improving robustness compared to existing methods.

Findings

Reduces model parameter transmission by sending only learnable filters and seeds.

Improves robustness and performance on corrupted data by approximately 2.2%.

Demonstrates effectiveness over state-of-the-art methods in IoT scenarios.

Abstract

Deploying deep convolutional neural network (CNN) models on ubiquitous Internet of Things (IoT) devices has attracted much attention from industry and academia since it greatly facilitates our lives by providing various rapid-response services. Due to the limited resources of IoT devices, cloud-assisted training of CNN models has become the mainstream. However, most existing related works suffer from a large amount of model parameter transmission and weak model robustness. To this end, this paper proposes a cloud-assisted CNN training framework with low model parameter transmission and strong model robustness. In the proposed framework, we first introduce MonoCNN, which contains only a few learnable filters, and other filters are nonlearnable. These nonlearnable filter parameters are generated according to certain rules, i.e., the filter generation function (FGF), and can be saved and reproduced by a few random seeds. Thus, the cloud server only needs to send these learnable filters and a few seeds to the IoT device. Compared to transmitting all model parameters, sending several learnable filter parameters and seeds can significantly reduce parameter transmission. Then, we investigate multiple FGFs and enable the IoT device to use the FGF to generate multiple filters and combine them into MonoCNN. Thus, MonoCNN is affected not only by the training data but also by the FGF. The rules of the FGF play a role in regularizing the MonoCNN, thereby improving its robustness. Experimental results show that compared to state-of-the-art methods, our proposed framework can reduce a large amount of model parameter transfer between the cloud server and the IoT device while improving the performance by approximately 2.2% when dealing with corrupted data. The code is available at https://github.com/evoxlos/mono-cnn-pytorch.