Back to Search Start Over

Research on Medical Image Classification Based on Machine Learning

Authors :
Zhihui Hu
Hai Tang
Source :
IEEE Access, Vol 8, Pp 93145-93154 (2020)
Publication Year :
2020
Publisher :
Institute of Electrical and Electronics Engineers (IEEE), 2020.

Abstract

In this paper, we propose a new method for CT pathological image analysis of brain and chest to extract image features and classify images. Because the deep neural network needs a large number of labeled samples to complete the training, and the cost of medical image labeling is very high, the training samples needed to train the deep neural network are insufficient. In this paper, a semi supervised learning based image classification method is proposed, which uses a small amount of labeled pathological image data to train the network model, and then integrates the features extracted by the network to classify the image. The results show that the classification effect of the neural network is better than convolution neural network and other traditional image classification models. To some extent, it can reduce the dependence of neural network on a large number of training samples, and effectively reduce the over fitting phenomenon of the network. Through the analysis of the overall classification accuracy and kappa coefficient of different classification methods under different sample numbers, it is found that the overall classification accuracy and kappa coefficient are increasing with the increasing number of training samples. Especially in the case of a small number of training samples, compared with other deep neural networks and traditional classification methods, the classification accuracy of the counter neural network is about 10% higher than that of other neural networks and traditional classification methods, and the advantages are more obvious.

Details

ISSN :
21693536
Volume :
8
Database :
OpenAIRE
Journal :
IEEE Access
Accession number :
edsair.doi.dedup.....069f8bb5091bf69b01b0cdc1bac0a5fc