Early diagnosis of retinal diseases such as diabetic retinopathy has had the attention of many researchers. Deep learning through the introduction of convolutional neural networks has become a prominent solution for image-related tasks such as classification and segmentation. Most tasks in image classification are handled by deep CNNs pretrained and evaluated on imagenet dataset. However, these models do not always translate to the best result on other datasets. Devising a neural network manually from scratch based on heuristics may not lead to an optimal model as there are numerous hyperparameters in play. In this paper, we use two nature-inspired swarm algorithms: particle swarm optimization (PSO) and ant colony optimization (ACO) to obtain TDCN models to perform classification of fundus images into severity classes. The power of swarm algorithms is used to search for various combinations of convolutional, pooling, and normalization layers to provide the best model for the task. It is observed that TDCN-PSO outperforms imagenet models and existing literature, while TDCN-ACO achieves faster architecture search. The best TDCN model achieves an accuracy of 90.3%, AUC ROC of 0.956, and a Cohen kappa score of 0.967. The results were compared with the previous studies to show that the proposed TDCN models exhibit superior performance.
翻译:许多研究人员都注意到了糖尿病视网膜病等视网膜疾病早期诊断。通过引入进化神经网络的深层次学习已成为与图像有关的任务,如分类和分化等的突出解决办法。图像分类工作大多由深层CNN对图像网数据集进行预先培训和评价。然而,这些模型并不总是转化成其他数据集的最佳结果。基于脂质的人工从零到零设计神经网络可能不会导致最佳模式,因为有许多超参数在起作用。在本文中,我们使用两种自然激发的热量算法:粒子温优化(PSO)和蚂蚁群优化(ACO),以获得TDCN模型,将基金图像分类为严重程度分类。这些模型的力量被用于寻找各种变压、集合和正常化层的组合,以便为这项任务提供最佳模型。观察到TDCN-PSO超越图像模型和现有文献,而TDCN-CNCO则实现了粒子最迅速的温暖优化模型搜索结果,而TDCN-3和ROC的高级模型则实现了对A/RMR的升级性能测试。