Medical image classification with hybrid convolutional neural network models
Öz
Despite important developments in medicine and technology today, many people die due to false or late diagnosis. It is very important to identify the small details in the images that can be overlooked in the examinations made on medical images in terms of early diagnosis of the disease. Therefore, it is vital in some cases to provide early diagnosis by detecting the details in the images automatically by computer systems. In the study carried out, it was aimed to diagnose the disease through medical images by classifying different types of images. For this purpose, convolutional neural networks, which are among deep learning techniques, were evaluated together with different classifier models. In the applied hybrid model approach, feature extraction was obtained from medical images with the convolutional neural network model. The extracted features are used to train different classification models. In the continuation of the study, the performance results obtained from the classifier models are compared. Two different datasets including brain MR images and lung x-ray images were used in the training and testing of hybrid models. In the study, images were classified into two categories as malignant and benign tumors in order to detect images containing malignant tumors in MR images. In order to identify images with pneumonia, the images are similarly classified into two categories, healthy and pneumonia. At the end of the study, the performance results obtained from the model approaches were compared and the performance evaluation of the models was performed.
Anahtar Kelimeler
Deep learning,Medical diagnosis,Convolutional neural network,Data augmentation,Classification
Kaynakça
- Aghdam, H. H. and Heravi, E. J. (2017). Guide to Convolutional Neural Networks, NY: Springer, New York, USA.
- Afshar, P., Mohammadi, A. and Plataniotis, K. N. (2018). “Brain tumor type classification via capsule networks.” Proc., 25th IEEE International Conference on Image Processing (ICIP), Athens, Greece, pp. 3129-3133. IEEE.
- Bejnordi, B. E., Lin, J., Glass, B., Mullooly, M., Gierach, G. L., Sherman, M. E., Karssemeijer, N., van der Laak, J. and Beck, A. H. (2017). “Deep learning-based assessment of tumor-associated stroma for diagnosing breast cancer in histopathology images.” Proc., IEEE 14th International Symposium on Biomedical Imaging (ISBI 2017), Melbourne, Australia, pp. 929-932. IEEE.
- Breiman, Leo. (2001). "Random forests." Machine Learning, Vol. 45, No. 1, pp. 5-32.
- Chakrabarty, N. (2019). Brain MRI Images for Brain Tumor Detection.
- Chen, X., Xu, Y., Wong, D. W. K., Wong, T. Y., and Liu, J. (2015). “Glaucoma detection based on deep convolutional neural network.” Proc., 37th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Milan, Italy, pp. 715-718.
- Chollet, F. (2017). “Xception: Deep learning with depthwise separable convolutions.” Proc., IEEE Conference on Computer Vision and Pattern Recognition, Hawaiʻi Convention Center, Honolulu, Hawaii, USA, pp. 1251-1258.
- Frid-Adar, M., Diamant, I., Klang, E., Amitai, M., Goldberger, J., and Greenspan, H. (2018). “GAN-based synthetic medical image augmentation for increased CNN performance in liver lesion classification.” Neurocomputing, Vol. 321, 321-331.
- Goodfellow, I., Bengio, Y., and Courville, A. (2016). Deep learning, MIT press, Massachusetts, USA.
- He, K., Zhang, X., Ren, S. and Sun, J. (2016). “Deep residual learning for image recognition.” Proc., IEEE Conference On Computer Vision And Pattern Recognition (CVPR), Las Vegas, NV, USA, pp. 770-778. IEEE.