arXiv reaDer
ロングテール網膜疾患認識のためのリレーショナルサブセット知識蒸留
Relational Subsets Knowledge Distillation for Long-tailed Retinal Diseases Recognition
現実の世界では、医療データセットはロングテールのデータ分布を示すことが多く(つまり、ほとんどのクラスがほとんどのデータを占めるのに対し、ほとんどのクラスはほとんどサンプルを持たない)、その結果、困難な不均衡学習シナリオが発生します。たとえば、罹患率が変動する40種類以上の網膜疾患が推定されていますが、30以上の状態では、世界の患者コホートからは非常にまれであり、深層学習ベースのスクリーニングの典型的な長期学習問題が発生します。モデル。本研究では、地域や表現型情報などの事前知識に応じて、ロングテールデータを複数のクラスサブセットに分割することにより、クラスサブセット学習を提案します。これは、サブセット固有の知識の学習に焦点を合わせるようにモデルを強制します。より具体的には、固定網膜領域に存在するいくつかの関係クラスが存在するか、またはいくつかの一般的な病理学的特徴が多数派および少数派の両方の状態で観察されます。これらのサブセットで学習した教師モデルを使用して、複数の教師モデルを、加重知識蒸留損失のある統合モデルに抽出できます。提案されたフレームワークは、ロングテール網膜疾患認識タスクに効果的であることが証明されました。 2つの異なるデータセットでの実験結果は、私たちの方法が柔軟であり、大幅な改善により他の多くの最先端技術に簡単にプラグインできることを示しています。
In the real world, medical datasets often exhibit a long-tailed data distribution (i.e., a few classes occupy most of the data, while most classes have rarely few samples), which results in a challenging imbalance learning scenario. For example, there are estimated more than 40 different kinds of retinal diseases with variable morbidity, however with more than 30+ conditions are very rare from the global patient cohorts, which results in a typical long-tailed learning problem for deep learning-based screening models. In this study, we propose class subset learning by dividing the long-tailed data into multiple class subsets according to prior knowledge, such as regions and phenotype information. It enforces the model to focus on learning the subset-specific knowledge. More specifically, there are some relational classes that reside in the fixed retinal regions, or some common pathological features are observed in both the majority and minority conditions. With those subsets learnt teacher models, then we are able to distill the multiple teacher models into a unified model with weighted knowledge distillation loss. The proposed framework proved to be effective for the long-tailed retinal diseases recognition task. The experimental results on two different datasets demonstrate that our method is flexible and can be easily plugged into many other state-of-the-art techniques with significant improvements.
updated: Thu Apr 22 2021 13:39:33 GMT+0000 (UTC)
published: Thu Apr 22 2021 13:39:33 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アソシエイト