arXiv reaDer
ノイズの多いラベルを使用したロバストな医療画像セグメンテーションのための共同クラスアフィニティ損失補正
Joint Class-Affinity Loss Correction for Robust Medical Image Segmentation with Noisy Labels
限られた注釈コストで収集されたノイズの多いラベルは、医療画像セグメンテーションアルゴリズムが正確な意味相関を学習することを妨げます。ノイズの多いラベルを使用した学習の以前のセグメンテーション技術は、ピクセル単位のラベル修正などのセマンティクスを保持するためにピクセル単位の方法を実行するだけで、ペア単位の方法を無視します。実際、ピクセル間の親和性関係をキャプチャするペアワイズ方式により、ラベルのノイズ率を大幅に低減できることがわかります。この観察に動機付けられて、監視がそれぞれノイズの多いクラスとアフィニティラベルから派生する、ピクセル単位とペア単位の両方の方法を組み込むことにより、ノイズを軽減するための新しい視点を提示します。ピクセル単位とペア単位の方法を統合して、医療画像セグメンテーションにおけるラベルノイズの問題に対処するための堅牢なジョイントクラスアフィニティセグメンテーション(JCAS)フレームワークを提案します。ペアワイズ方式のアフィニティーにはコンテキスト依存関係が組み込まれていることを考慮して、クラス内およびクラス間のアフィニティー関係について推論することにより、ピクセル単位のセグメンテーション予測を修正するために、差別化アフィニティー推論(DAR)モジュールが考案されています。ノイズ耐性をさらに強化するために、クラスアフィニティ損失補正(CALC)戦略は、クラスおよびアフィニティラベルのモデル化されたノイズラベル分布を介して監視信号を補正するように設計されています。一方、CALC戦略は、理論的に導出された整合性の正則化を通じて、ピクセル単位およびペア単位の方法で相互作用します。合成および実世界のノイズの多いラベルの両方での広範な実験は、提案されたJCASフレームワークの有効性を裏付け、上限パフォーマンスへのギャップを最小限に抑えます。ソースコードはhttps://github.com/CityU-AIM-Group/JCASで入手できます。
Noisy labels collected with limited annotation cost prevent medical image segmentation algorithms from learning precise semantic correlations. Previous segmentation arts of learning with noisy labels merely perform a pixel-wise manner to preserve semantics, such as pixel-wise label correction, but neglect the pair-wise manner. In fact, we observe that the pair-wise manner capturing affinity relations between pixels can greatly reduce the label noise rate. Motivated by this observation, we present a novel perspective for noisy mitigation by incorporating both pixel-wise and pair-wise manners, where supervisions are derived from noisy class and affinity labels, respectively. Unifying the pixel-wise and pair-wise manners, we propose a robust Joint Class-Affinity Segmentation (JCAS) framework to combat label noise issues in medical image segmentation. Considering the affinity in pair-wise manner incorporates contextual dependencies, a differentiated affinity reasoning (DAR) module is devised to rectify the pixel-wise segmentation prediction by reasoning about intra-class and inter-class affinity relations. To further enhance the noise resistance, a class-affinity loss correction (CALC) strategy is designed to correct supervision signals via the modeled noise label distributions in class and affinity labels. Meanwhile, CALC strategy interacts the pixel-wise and pair-wise manners through the theoretically derived consistency regularization. Extensive experiments under both synthetic and real-world noisy labels corroborate the efficacy of the proposed JCAS framework with a minimum gap towards the upper bound performance. The source code is available at https://github.com/CityU-AIM-Group/JCAS.
updated: Thu Jun 16 2022 08:19:33 GMT+0000 (UTC)
published: Thu Jun 16 2022 08:19: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アソシエイト