Toward Efficient Image Recognition in Sensor-Based IoT: A Weight Initialization Optimizing Method for CNN Based on RGB Influence Proportion

• Published in: Sensors (Basel, Switzerland) Vol. 20; no. 10; p. 2866
• Main Authors: Deng, Zile, Cao, Yuanlong, Zhou, Xinyu, Yi, Yugen, Jiang, Yirui, You, Ilsun
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI 18.05.2020; MDPI AG
• Subjects: convolution neural network (CNN); image recognition; IoT application; k-nearest neighbor (k-NN); convolution neural network (CNN); IoT application; image recognition; k-nearest neighbor (k-NN)
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s20102866

As the Internet of Things (IoT) is predicted to deal with different problems based on big data, its applications have become increasingly dependent on visual data and deep learning technology, and it is a big challenge to find a suitable method for IoT systems to analyze image data. Traditional deep learning methods have never explicitly taken the color differences of data into account, but from the experience of human vision, colors play differently significant roles in recognizing things. This paper proposes a weight initialization method for deep learning in image recognition problems based on RGB influence proportion, aiming to improve the training process of the learning algorithms. In this paper, we try to extract the RGB proportion and utilize it in the weight initialization process. We conduct several experiments on different datasets to evaluate the effectiveness of our proposal, and it is proven to be effective on small datasets. In addition, as for the access to the RGB influence proportion, we also provide an expedient approach to get the early proportion for the following usage. We assume that the proposed method can be used for IoT sensors to securely analyze complex data in the future.