arXiv reaDer
糖尿病性網膜症の深層学習に基づく分析のための事前訓練と自己監視のロバスト性について
On the Robustness of Pretraining and Self-Supervision for a Deep Learning-based Analysis of Diabetic Retinopathy
ディープニューラルネットワークに基づく分類アルゴリズムが、人間の医療専門家と競争力のあるパフォーマンスレベルに達する医療ユースケースが増えています。データセットサイズが小さいという課題を軽減するために、これらのシステムは多くの場合、事前トレーニングに依存しています。この作業では、これらのアプローチのより広い意味を評価することを目指しています。例示的なユースケースとしての糖尿病性網膜症の格付けについて、対照学習に基づく最近確立された自己教師あり事前トレーニング方法を含むさまざまなトレーニング手順の影響を比較します。この目的のために、定量的パフォーマンス、学習された特徴表現の統計、解釈可能性、画像の歪みに対するロバスト性など、さまざまな側面を調査します。私たちの結果は、ImageNet事前トレーニングから初期化されたモデルが、パフォーマンス、一般化、および画像の歪みに対する堅牢性の大幅な向上を報告していることを示しています。特に、自己教師ありモデルは、教師ありモデルにさらなる利点を示します。 ImageNet事前トレーニングからの初期化による自己監視モデルは、より高いパフォーマンスを報告するだけでなく、疾患の進行を示す微細な病変を考慮に入れることの改善とともに、大きな病変への過剰適合を減らします。単純なパフォーマンスの比較を超えた広い意味での事前トレーニングの効果を理解することは、この作業で検討されているユースケースを超えて、より広い医用画像コミュニティにとって非常に重要です。
There is an increasing number of medical use-cases where classification algorithms based on deep neural networks reach performance levels that are competitive with human medical experts. To alleviate the challenges of small dataset sizes, these systems often rely on pretraining. In this work, we aim to assess the broader implications of these approaches. For diabetic retinopathy grading as exemplary use case, we compare the impact of different training procedures including recently established self-supervised pretraining methods based on contrastive learning. To this end, we investigate different aspects such as quantitative performance, statistics of the learned feature representations, interpretability and robustness to image distortions. Our results indicate that models initialized from ImageNet pretraining report a significant increase in performance, generalization and robustness to image distortions. In particular, self-supervised models show further benefits to supervised models. Self-supervised models with initialization from ImageNet pretraining not only report higher performance, they also reduce overfitting to large lesions along with improvements in taking into account minute lesions indicative of the progression of the disease. Understanding the effects of pretraining in a broader sense that goes beyond simple performance comparisons is of crucial importance for the broader medical imaging community beyond the use-case considered in this work.
updated: Fri Jun 25 2021 08:32:45 GMT+0000 (UTC)
published: Fri Jun 25 2021 08:32:45 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アソシエイト