arXiv reaDer
カラー眼底画像からの硬性滲出液セグメンテーションのためのデュアルサンプリング変調ダイスロスを備えたデュアルブランチネットワーク
Dual-Branch Network with Dual-Sampling Modulated Dice Loss for Hard Exudate Segmentation from Colour Fundus Images
カラー眼底画像における硬い滲出液の自動セグメンテーションは、極端なクラスの不均衡と巨大なサイズ変動の問題のために挑戦的なタスクです。この論文は、これらの問題に取り組むことを目的とし、デュアルサンプリング変調ダイス損失を備えたデュアルブランチネットワークを提案します。これは、2つのブランチで構成されています。大きなハード滲出液バイアス学習ブランチと小さなハード滲出液バイアス学習ブランチです。それらの両方が別々に彼ら自身の義務に責任があります。さらに、提案されたデュアルブランチネットワークが異なるサイズの硬い滲出液をセグメント化できるように、トレーニング用のデュアルサンプリング変調ダイス損失を提案します。詳細には、最初のブランチでは、均一なサンプラーを使用して、ダイス損失計算用の予測セグメンテーションマスクからピクセルをサンプリングします。これにより、ダイス損失は大きなハードの誤認により大きなコストを生成するため、このブランチは自然に大きなハード滲出液にバイアスされます。小さな硬い滲出液よりも滲出液。 2番目のブランチでは、リバランスされたサンプラーを使用して、ハード滲出ピクセルをオーバーサンプリングし、損失計算のためにバックグラウンドピクセルをアンダーサンプリングします。このようにして、小さな硬い滲出液の誤認のコストが増大し、これにより、2番目のブランチのパラメータが小さな硬い滲出液にうまく適合します。大きな硬性滲出液は小さな硬性滲出液よりも正確に識別しやすいことを考慮して、2つの分岐の損失を適応的に変調することにより、困難な学習戦略を提案します。 2つの公開データセットで提案された方法を評価し、結果は、私たちが最先端のパフォーマンスを達成していることを示しています。
Automated segmentation of hard exudates in colour fundus images is a challenge task due to issues of extreme class imbalance and enormous size variation. This paper aims to tackle these issues and proposes a dual-branch network with dual-sampling modulated Dice loss. It consists of two branches: large hard exudate biased learning branch and small hard exudate biased learning branch. Both of them are responsible for their own duty separately. Furthermore, we propose a dual-sampling modulated Dice loss for the training such that our proposed dual-branch network is able to segment hard exudates in different sizes. In detail, for the first branch, we use a uniform sampler to sample pixels from predicted segmentation mask for Dice loss calculation, which leads to this branch naturally be biased in favour of large hard exudates as Dice loss generates larger cost on misidentification of large hard exudates than small hard exudates. For the second branch, we use a re-balanced sampler to oversample hard exudate pixels and undersample background pixels for loss calculation. In this way, cost on misidentification of small hard exudates is enlarged, which enforces the parameters in the second branch fit small hard exudates well. Considering that large hard exudates are much easier to be correctly identified than small hard exudates, we propose an easy-to-difficult learning strategy by adaptively modulating the losses of two branches. We evaluate our proposed method on two public datasets and results demonstrate that ours achieves state-of-the-art performances.
updated: Thu May 06 2021 06:59:31 GMT+0000 (UTC)
published: Thu Dec 03 2020 02:37: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アソシエイト