arXiv reaDer
Deep Interpretable Classification and Weakly-Supervised Segmentation of Histology Images via Max-Min Uncertainty
弱教師あり学習(WSL)は、ピクセル単位の注釈の欠如を軽減するため、最近大きな関心を呼んでいます。グローバルな画像ラベルが与えられると、WSLメソッドはピクセルレベルの予測(セグメンテーション)を生成します。これにより、クラスの予測を解釈できます。最近の成功にもかかわらず、主に自然画像で、前景領域と背景領域が同様の視覚的手がかりを持っている場合、そのような方法は重要な課題に直面する可能性があり、挑戦的な組織画像の場合のように、セグメンテーションで高い偽陽性率をもたらします。 WSLトレーニングは通常、モデルの信頼性を暗黙的に最大化し、分類の決定にリンクされた識別領域を特定する標準的な分類損失によって駆動されます。したがって、それらは、明示的に非識別領域をモデル化し、偽陽性率を低減するためのメカニズムを欠いています。モデルが不均衡なセグメンテーションを阻止しながら、非識別領域と識別領域の両方を探すことを可能にする、新しい正則化項を提案します。分類器の決定に影響を与えない非識別領域をローカライズするための基準として高い不確実性を導入し、一様分布からの事後予測の偏差を評価する元のカルバックライブラー(KL)発散損失で説明します。後者が潜在的な非識別領域を入力する場合、私たちのKL項はモデルの高い不確実性を助長します。私たちの損失は次のように統合されます。(i)クラス予測に関するモデルの信頼性が高い前景を求めるクロスエントロピー。 (ii)モデルの不確実性が高い背景を求めるKL正則化。 (iii)不均衡なセグメンテーションを思いとどまらせる対数障壁条件。公開されているGlaS結腸がんデータとCamelyon16パッチベースの乳がんベンチマークに関する包括的な実験とアブレーション研究は、最先端のWSL法に比べて大幅な改善を示しており、新しいレギュラライザーの効果を確認しています。
Weakly-supervised learning (WSL) has recently triggered substantial interest as it mitigates the lack of pixel-wise annotations. Given global image labels, WSL methods yield pixel-level predictions (segmentations), which enable to interpret class predictions. Despite their recent success, mostly with natural images, such methods can face important challenges when the foreground and background regions have similar visual cues, yielding high false-positive rates in segmentations, as is the case in challenging histology images. WSL training is commonly driven by standard classification losses, which implicitly maximize model confidence, and locate the discriminative regions linked to classification decisions. Therefore, they lack mechanisms for modeling explicitly non-discriminative regions and reducing false-positive rates. We propose novel regularization terms, which enable the model to seek both non-discriminative and discriminative regions, while discouraging unbalanced segmentations. We introduce high uncertainty as a criterion to localize non-discriminative regions that do not affect classifier decision, and describe it with original Kullback-Leibler (KL) divergence losses evaluating the deviation of posterior predictions from the uniform distribution. Our KL terms encourage high uncertainty of the model when the latter inputs the latent non-discriminative regions. Our loss integrates: (i) a cross-entropy seeking a foreground, where model confidence about class prediction is high; (ii) a KL regularizer seeking a background, where model uncertainty is high; and (iii) log-barrier terms discouraging unbalanced segmentations. Comprehensive experiments and ablation studies over the public GlaS colon cancer data and a Camelyon16 patch-based benchmark for breast cancer show substantial improvements over state-of-the-art WSL methods, and confirm the effect of our new regularizers.
updated: Sat Oct 09 2021 00:59:44 GMT+0000 (UTC)
published: Sat Nov 14 2020 04:45:07 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト