{"title":"Diabetic Retinopathy Fundus Image Classification Using Ensemble Methods","authors":"Marina M. Lukashevich","doi":"10.1134/s1054661824700123","DOIUrl":null,"url":null,"abstract":"<h3 data-test=\"abstract-sub-heading\">Abstract</h3><p>Diabetic retinopathy causes damage to the retina of the eye and leads to poor vision in patients with diabetes around the world. It affects the retina of a person’s eye, begins asymptomatically, and can lead to complete loss of vision. Screening for this disease can be done fairly quickly by using machine learning algorithms to analyze retinal images. Early diagnosis is crucial to prevent dangerous consequences such as blindness. This paper presents the results of implementation and comparison of ensemble machine learning algorithms and describes an approach to the selection of hyperparameters for solving screening problems (binary classification) and classifying the stage of diabetic retinopathy (from 0 to 4). Particular attention is paid to the approaches of searching for hyperparameters on a lattice and random search. This study uses a hyperparameter selection mechanism for ensemble algorithms based on a combination of grid search and random search approaches. The selection of hyperparameters, as well as the selection of informative features, made it possible to increase the accuracy of classification of retinal images. The experimental results showed an accuracy of 0.7531 for retinal image classification on the test dataset for the best model (gradient boosting, GB). When considering a binary classification (presence or absence of diabetic retinopathy), an accuracy of 0.9400 (gradient boosting, GB) was achieved.</p>","PeriodicalId":35400,"journal":{"name":"PATTERN RECOGNITION AND IMAGE ANALYSIS","volume":"17 1","pages":""},"PeriodicalIF":0.7000,"publicationDate":"2024-07-04","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"PATTERN RECOGNITION AND IMAGE ANALYSIS","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1134/s1054661824700123","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q4","JCRName":"COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONS","Score":null,"Total":0}
引用次数: 0
Abstract
Diabetic retinopathy causes damage to the retina of the eye and leads to poor vision in patients with diabetes around the world. It affects the retina of a person’s eye, begins asymptomatically, and can lead to complete loss of vision. Screening for this disease can be done fairly quickly by using machine learning algorithms to analyze retinal images. Early diagnosis is crucial to prevent dangerous consequences such as blindness. This paper presents the results of implementation and comparison of ensemble machine learning algorithms and describes an approach to the selection of hyperparameters for solving screening problems (binary classification) and classifying the stage of diabetic retinopathy (from 0 to 4). Particular attention is paid to the approaches of searching for hyperparameters on a lattice and random search. This study uses a hyperparameter selection mechanism for ensemble algorithms based on a combination of grid search and random search approaches. The selection of hyperparameters, as well as the selection of informative features, made it possible to increase the accuracy of classification of retinal images. The experimental results showed an accuracy of 0.7531 for retinal image classification on the test dataset for the best model (gradient boosting, GB). When considering a binary classification (presence or absence of diabetic retinopathy), an accuracy of 0.9400 (gradient boosting, GB) was achieved.
期刊介绍:
The purpose of the journal is to publish high-quality peer-reviewed scientific and technical materials that present the results of fundamental and applied scientific research in the field of image processing, recognition, analysis and understanding, pattern recognition, artificial intelligence, and related fields of theoretical and applied computer science and applied mathematics. The policy of the journal provides for the rapid publication of original scientific articles, analytical reviews, articles of the world''s leading scientists and specialists on the subject of the journal solicited by the editorial board, special thematic issues, proceedings of the world''s leading scientific conferences and seminars, as well as short reports containing new results of fundamental and applied research in the field of mathematical theory and methodology of image analysis, mathematical theory and methodology of image recognition, and mathematical foundations and methodology of artificial intelligence. The journal also publishes articles on the use of the apparatus and methods of the mathematical theory of image analysis and the mathematical theory of image recognition for the development of new information technologies and their supporting software and algorithmic complexes and systems for solving complex and particularly important applied problems. The main scientific areas are the mathematical theory of image analysis and the mathematical theory of pattern recognition. The journal also embraces the problems of analyzing and evaluating poorly formalized, poorly structured, incomplete, contradictory and noisy information, including artificial intelligence, bioinformatics, medical informatics, data mining, big data analysis, machine vision, data representation and modeling, data and knowledge extraction from images, machine learning, forecasting, machine graphics, databases, knowledge bases, medical and technical diagnostics, neural networks, specialized software, specialized computational architectures for information analysis and evaluation, linguistic, psychological, psychophysical, and physiological aspects of image analysis and pattern recognition, applied problems, and related problems. Articles can be submitted either in English or Russian. The English language is preferable. Pattern Recognition and Image Analysis is a hybrid journal that publishes mostly subscription articles that are free of charge for the authors, but also accepts Open Access articles with article processing charges. The journal is one of the top 10 global periodicals on image analysis and pattern recognition and is the only publication on this topic in the Russian Federation, Central and Eastern Europe.