arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.11432cs.CVstat.ML

Towards the Localisation of Lesions in Diabetic Retinopathy

African Institute for Mathematical Sciences(非洲数学科学研究所);Stellenbosch University(斯泰伦博斯大学)

Samuel Ofosu Mensah, Bubacarr Bah, Willie Brink

英文摘要:

Convolutional Neural Networks (CNNs) have successfully been used to classify diabetic retinopathy (DR) fundus images in recent times. However, deeper representations in CNNs may capture higher-level semantics at the expense of spatial resolution. To make predictions usable for ophthalmologists, we use a post-attention technique called Gradient-weighted Class Activation Mapping (Grad-CAM) on the penultimate layer of deep learning models to produce coarse localisation maps on DR fundus images. This is to help identify discriminative regions in the images, consequently providing evidence for ophthalmologists to make a diagnosis and potentially save lives by early diagnosis. Specifically, this study uses pre-trained weights from four state-of-the-art deep learning models to produce and compare localisation maps of DR fundus images. The models used include VGG16, ResNet50, InceptionV3, and InceptionResNetV2. We find that InceptionV3 achieves the best performance with a test classification accuracy of 96.07%, and localise lesions better and faster than the other models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑