The machine learning algorithm is based on our earlier work using retinal pixel and lesion classification.
8 9 10 To perform detection and differentiation of bright lesions, if any, in a previously unseen image, the following steps were performed:
-
Each pixel was classified, resulting in a so-called lesion probability map that indicates the probability that a pixel is part of a bright lesion.
-
Pixels with high probability were grouped into probable lesion pixel clusters.
-
Based on cluster characteristics each probable lesion pixel cluster was assigned a probability indicating the likelihood that the pixel cluster was a true bright lesion.
-
Each bright lesion cluster likely to be a bright lesion was classified as exudate, cotton-wool spot or drusen.
Figure 1illustrates these steps. The system performs the classification steps (1, 3, and 4) by using a statistical classifier that, given a training set of labeled examples, can differentiate among different types of pixels or clusters based on so-called
features or numerical characteristics, such as pixel color and cluster area. The features used depend on the type of objects that are to be classified. For each classification step, several statistical classifiers were tested on the training set and the one demonstrating the best performance was used on the test set. A k-nearest neighbor classifier was selected for steps 1 and 3 and a linear discriminant classifier for the third classification step.
16 When presented with an unseen image in the test set, the automated algorithm gave two outputs: whether bright lesions were present or not, and which class each lesion was: exudate, cotton-wool spot, or drusen. A more extensive description of the algorithm is given in the Appendix.