arXiv reaDer
硬度を意識した動的カリキュラム学習による自己教師あり学習の改善:デジタル病理学への応用
Improving Self-supervised Learning with Hardness-aware Dynamic Curriculum Learning: An Application to Digital Pathology
自己教師あり学習(SSL)は、最近、多くの画像分析タスクに役立つ一般的な視覚表現を学習する大きな可能性を示しています。顕著な成功にもかかわらず、既存のSSLメソッドは、ラベル付けされたトレーニングインスタンスの数が少ない場合、または転送ドメイン間のドメインシフトが大きい場合、ダウンストリームタスクに一般化できません。この論文では、硬さを意識した動的カリキュラム学習(HaDCL)アプローチを提案することにより、カリキュラム学習のレンズを通して自己教師ありの事前訓練された表現を改善しようと試みます。 SSLの堅牢性と一般化可能性を向上させるために、ミニバッチのダウンストリームの微調整中に、ハードになりやすいサンプルとハードに難しいサンプルを介して、プログレッシブで難しい例を動的に活用します。漸進的な段階ごとのカリキュラム学習によって、事前にトレーニングされた表現が大幅に強化され、ドメイン内とドメイン外の両方の分布データに適応できることがわかりました。パッチごとの分類問題とスライドレベルの分類問題の両方について、3つの組織学ベンチマークデータセットに対して広範な検証を実行しました。カリキュラムベースの微調整により、標準の微調整よりも大幅に改善され、ドメイン内およびドメイン外の分布データの曲線下面積(AUC)スコアが1.7%および2.2%最小で改善されます。 、 それぞれ。さらに、私たちのアプローチはより一般的であり、SSLメソッドに適応可能であり、追加のオーバーヘッドの複雑さを課さないことを経験的に示しています。さらに、SSLメソッドのカリキュラムベースの微調整の成功への実用的な洞察を提供するために、組織病理学におけるパッチベースとスライドベースのカリキュラム学習の役割についても概説します。コードはhttps://github.com/srinidhiPY/ICCVCDPATH2021-ID-8でリリースされます
Self-supervised learning (SSL) has recently shown tremendous potential to learn generic visual representations useful for many image analysis tasks. Despite their notable success, the existing SSL methods fail to generalize to downstream tasks when the number of labeled training instances is small or if the domain shift between the transfer domains is significant. In this paper, we attempt to improve self-supervised pretrained representations through the lens of curriculum learning by proposing a hardness-aware dynamic curriculum learning (HaDCL) approach. To improve the robustness and generalizability of SSL, we dynamically leverage progressive harder examples via easy-to-hard and hard-to-very-hard samples during mini-batch downstream fine-tuning. We discover that by progressive stage-wise curriculum learning, the pretrained representations are significantly enhanced and adaptable to both in-domain and out-of-domain distribution data. We performed extensive validation on three histology benchmark datasets on both patch-wise and slide-level classification problems. Our curriculum based fine-tuning yields a significant improvement over standard fine-tuning, with a minimum improvement in area-under-the-curve (AUC) score of 1.7% and 2.2% on in-domain and out-of-domain distribution data, respectively. Further, we empirically show that our approach is more generic and adaptable to any SSL methods and does not impose any additional overhead complexity. Besides, we also outline the role of patch-based versus slide-based curriculum learning in histopathology to provide practical insights into the success of curriculum based fine-tuning of SSL methods. Code will be released at https://github.com/srinidhiPY/ICCVCDPATH2021-ID-8
updated: Mon Aug 16 2021 15:44:48 GMT+0000 (UTC)
published: Mon Aug 16 2021 15:44:48 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アソシエイト