Few-shot Learning Based on Multi-stage Transfer and Class-Balanced Loss
  for Diabetic Retinopathy Grading
  Diabetic retinopathy (DR) is one of the major blindness-causing diseases
current-ly known. Automatic grading of DR using deep learning methods not only
speeds up the diagnosis of the disease but also reduces the rate of
misdiagnosis. However, problems such as insufficient samples and imbalanced
class distribu-tion in DR datasets have constrained the improvement of grading
performance. In this paper, we introduce the idea of multi-stage transfer into
the grading task of DR. The new transfer learning technique leverages multiple
datasets with differ-ent scales to enable the model to learn more feature
representation information. Meanwhile, to cope with imbalanced DR datasets, we
present a class-balanced loss function that performs well in natural image
classification tasks, and adopt a simple and easy-to-implement training method
for it. The experimental results show that the application of multi-stage
transfer and class-balanced loss function can effectively improve the grading
performance metrics such as accuracy and quadratic weighted kappa. In fact, our
method has outperformed two state-of-the-art methods and achieved the best
result on the DR grading task of IDRiD Sub-Challenge 2.
