고정밀 이미지 분류기를 위한 엣지 강화 신경망

Abstract

Among the several deep learning research areas, image classification is a fundamental area and has been widely applied to many practical applications. Image classification is to determine which category the given image belongs to. Since the image classification is a typical supervised learning, test images and the corresponding answers, i.e., labels are also given. And, with the given test images and labels, a neural network is trained to minimize a loss function defined by error between the label and the inference result. Therefore, as the loss decreases, the inference accuracy increases. As a result, the accuracy is a criteria of the performance of the neural network in image classification. In this paper, a new method for high accuracy image classification is suggested. Authors thinks that recognizing things mean seeing the shape of the things. With the intuition, in the proposed method, additional edge information is applied to the image and around 3% accuracy improvement is achieved in the experiment. In order to clarify the improvement, compared feature maps in the hidden layers are visualized and analyzed. And, it is also confirmed that the feature map of the proposed method is more clear and sharp than that of the conventional one. Another merit of the proposed method is that this method can easily improve the accuracy of the conventionally existing neural networks through the transfer learning because the proposed method just modifies the first layer of the conventional neural networks.