arXiv reaDer
小さなデータで糖尿病性網膜症をグレーディングするための多段階転移学習フレームワーク
A Multi-stage Transfer Learning Framework for Diabetic Retinopathy Grading on Small Data
糖尿病性網膜症 (DR) は、現在知られている失明の原因となる主要な疾患の 1 つです。ディープラーニング手法を使用した DR の自動採点は、病気の診断を迅速化するだけでなく、誤診の率も削減します。しかし、小規模な DR データセットにおけるサンプル不足や不均衡なクラス分布などの問題により、グレーディング パフォーマンスの向上が制約されてきました。この論文では、多段階転移学習のアイデアを DR の採点タスクに適用します。新しい転移学習手法では、スケールの異なる複数のデータセットを利用して、モデルがより多くの特徴表現情報を学習できるようにします。一方、小規模な DR データセットの不均衡の問題に対処するために、私たちはクラスバランス損失関数を研究で提示し、そのためのシンプルで実装が簡単なトレーニング方法を採用しました。 IDRiD データセットでの実験結果は、私たちの方法が小規模データのグレーディング パフォーマンスを効果的に向上させることができ、精度と二次加重カッパの点でそれぞれ 0.7961 と 0.8763 のスコアを取得できることを示しています。また、私たちの方法はいくつかの最先端の方法よりも優れています。
Diabetic retinopathy (DR) is one of the major blindness-causing diseases currently 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 distribution in small DR datasets have constrained the improvement of grading performance. In this paper, we apply the idea of multi-stage transfer learning into the grading task of DR. The new transfer learning technique utilizes multiple datasets with different scales to enable the model to learn more feature representation information. Meanwhile, to cope with the imbalanced problem of small DR datasets, we present a class-balanced loss function in our work and adopt a simple and easy-to-implement training method for it. The experimental results on IDRiD dataset show that our method can effectively improve the grading performance on small data, obtaining scores of 0.7961 and 0.8763 in terms of accuracy and quadratic weighted kappa, respectively. Our method also outperforms several state-of-the-art methods.
updated: Wed May 24 2023 10:01:24 GMT+0000 (UTC)
published: Fri Sep 24 2021 08:39:13 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト