arXiv reaDer
ECL: 尾部皮膚病変分類のためのクラス強化対照学習
ECL: Class-Enhancement Contrastive Learning for Long-tailed Skin Lesion Classification
皮膚画像データセットはデータの分布が不均衡であることが多く、コンピューター支援による皮膚疾患診断の困難さをさらに悪化させています。最近の研究の中には、この長期にわたる課題に対して教師あり対比学習 (SCL) を活用したものもあります。これらの SCL ベースのメソッドは、顕著なパフォーマンスを達成しているにもかかわらず、先頭クラスに重点を置き、末尾クラスでの情報の利用を無視しています。この論文では、少数クラスの情報を強化し、異なるクラスを平等に扱うクラス強化対照学習 (ECL) を提案します。情報強化のために、クラス依存のプロキシを生成するハイブリッド プロキシ モデルを設計し、パラメータ最適化のためのサイクル更新戦略を提案します。バランス型ハイブリッド プロキシ損失は、異なるクラスが同等に扱われるサンプルとプロキシ間の関係を利用するように設計されています。 「不均衡なデータ」と「不均衡な診断難易度」の両方を考慮して、カリキュラム学習スケジュールに従ってバランスのとれた重み付けされたクロスエントロピー損失をさらに提示します。不均衡な皮膚病変データの分類に関する実験結果は、私たちの方法の優位性と有効性を実証しました。
Skin image datasets often suffer from imbalanced data distribution, exacerbating the difficulty of computer-aided skin disease diagnosis. Some recent works exploit supervised contrastive learning (SCL) for this long-tailed challenge. Despite achieving significant performance, these SCL-based methods focus more on head classes, yet ignoring the utilization of information in tail classes. In this paper, we propose class-Enhancement Contrastive Learning (ECL), which enriches the information of minority classes and treats different classes equally. For information enhancement, we design a hybrid-proxy model to generate class-dependent proxies and propose a cycle update strategy for parameters optimization. A balanced-hybrid-proxy loss is designed to exploit relations between samples and proxies with different classes treated equally. Taking both "imbalanced data" and "imbalanced diagnosis difficulty" into account, we further present a balanced-weighted cross-entropy loss following curriculum learning schedule. Experimental results on the classification of imbalanced skin lesion data have demonstrated the superiority and effectiveness of our method.
updated: Sun Jul 09 2023 09:29:15 GMT+0000 (UTC)
published: Sun Jul 09 2023 09:29:15 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アソシエイト