Classification method of dermoscopic image based on hierarchical convolution neural network

Abstract

In order to solve the problem of insufficient number of dermoscopic image and the imbalance of image data among various diseases, a classification method of dermoscopic image based on class weighted cross entropy loss function and hierarchical convolution neural network was proposed.Firstly, the dermoscopic image was processed by color constancy to eliminate the ambient light noise.Then, the hierarchical convolution neural network based on ResNet 50 was constructed, and the two classification and multi classification convolution neural network models were constructed respectively, and the class weighted cross entropy loss function was set according to the quantitative characteristics of the dermatoscopic image.The experimental results show that the method achieves good classification effect, and the classification accuracy reaches 85.94%.Compared with the improved classification model ResNet 50, the test accuracy is improved by 5.752%.

Keywords

• ResNet 50;classification of dermoscopic image;hierarchical convolution neural network;over fitting