Augmented semi-supervised learning for salient object detection with edge computing

Abstract

[EN] Salient object detection (SOD) from raw sensor images in the edge networks can effectively speed up the decision-making process in the complex environments, because it simulates the mechanism of human attention to identify salient objects from images. The success of supervised deep learning approaches have been widely proved SOD field. However, the imbalanced and limited training data at each edge device pose a huge challenge for us to deploy deep learning methods in the edge computing environments. In this article, we propose a cloud-edge distributed augmented semi-supervised learning architecture for SOD over the edge networks. The framework consists of two components: the base classification networks are employed in different edge nodes, and the reverse augmented network is employed in cloud. First, the base classification networks are trained with data from edge nodes while the reverse augmented network is trained with the whole data. Then, we concatenate each base classification network with reverse augmented network, thus the latter network can help the training of former network. Finally, we integrate the outputs of all base classification network to generate the pseudo-labels, which are used for semi-supervised learning of the augment network. We demonstrated a convincing performance of our semi-supervised learning framework on four bench-marked data-sets. These results show that our augmented semi-supervised learning framework can outperform other optimization strategies on deep learning for the edge computing.