糖尿病性網膜症(DR)は、現在知られている主要な失明の原因となる病気の1つです。ディープラーニング手法を使用したDRの自動グレーディングは、病気の診断をスピードアップするだけでなく、誤診の割合を減らします。ただし、サンプルの不足やDRデータセット内のクラス分布の不均衡などの問題により、評価パフォーマンスの向上が制約されています。この論文では、DRの評価タスクに多段階転送のアイデアを紹介します。新しい伝達学習手法は、異なるスケールの複数のデータセットを活用して、モデルがより多くの特徴表現情報を学習できるようにします。一方、不均衡なDRデータセットに対処するために、自然画像分類タスクでうまく機能するクラス均衡損失関数を提示し、そのためのシンプルで実装が容易なトレーニング方法を採用します。実験結果は、多段転送とクラスバランス損失関数の適用が、精度や二次加重カッパなどのグレーディングパフォーマンスメトリックを効果的に改善できることを示しています。実際、私たちの方法は2つの最先端の方法を上回り、IDRiDサブチャレンジ2のDRグレーディングタスクで最高の結果を達成しました。
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.