Feature visualisation of classification of diabetic retinopathy using a convolutional neural network



Pratt, H, Coenen, F ORCID: 0000-0003-1026-6649, Harding, SP ORCID: 0000-0003-4676-1158, Broadbent, DM and Zheng, Y ORCID: 0000-0002-7873-0922
(2019) Feature visualisation of classification of diabetic retinopathy using a convolutional neural network. .

[img] Text
kdh-2019_Pratt_Paper_10.pdf - Accepted Version

Download (17MB) | Preview

Abstract

© 2019 for this paper by its authors. Convolutional Neural Networks (CNNs) have been demonstrated to achieve state-of-the-art results on complex computer vision tasks, including medical image diagnosis of Diabetic Retinopathy (DR). CNNs are powerful because they determine relevant image features automatically. However, the current inability to demonstrate what these features are has led to CNNs being considered to be 'black box' methods whose results should not be trusted. This paper presents a method for identifying the learned features of a CNN and applies it in the context of the diagnosis of DR in fundus images using the well-known DenseNet. We train the CNN to diagnose and determine the severity of DR and then successfully extract feature maps from the CNN which identify the regions and features of the images which have led most strongly to the CNN prediction. This feature extraction process has great potential, particularly for encouraging confidence in CNN approaches from users and clinicians, and can aid in the further development of CNN methods. There is also potential for determining previously unidentified features which may contribute to a classification.

Item Type: Conference or Workshop Item
Depositing User: Symplectic Admin
Date Deposited: 17 Sep 2019 08:10
Last Modified: 13 Dec 2019 08:21
URI: http://livrepository.liverpool.ac.uk/id/eprint/3054853
Repository Staff Access