arXiv reaDer
隣人を信頼する: モデルのキャリブレーションに対するペナルティベースの制約
Trust your neighbours: Penalty-based constraints for model calibration
深いネットワークから信頼できる信頼スコアを確保することは、重要な意思決定システム、特に医療分野において非常に重要です。ディープ セグメンテーション ネットワークのキャリブレーションに関する最近の文献は大きな進歩をもたらしましたが、その不確実性は通常、個々のピクセルの情報を利用してモデル化され、対象オブジェクトの局所構造は無視されます。特に、最近の Spatially Varying Label Smoothing (SVLS) アプローチのみが、離散空間ガウス カーネルを使用してピクセル ラベルの割り当てを緩和することでこの問題に対処しています。この作業では、最初に SVLS の制約付き最適化の観点を提示し、それが周囲のピクセルのソフト クラスの比率に暗黙の制約を適用することを示します。さらに、私たちの分析では、SVLS には制約の寄与と主な目的のバランスをとるメカニズムがなく、最適化プロセスを妨げる可能性があることが示されています。これらの観察に基づいて、ロジット値の等式制約に基づく原則的で単純なソリューションを提案します。これにより、強制された制約とペナルティの重みの両方を明示的に制御でき、柔軟性が向上します。さまざまなよく知られているセグメンテーション ベンチマークでの包括的な実験は、提案されたアプローチの優れたパフォーマンスを示しています。
Ensuring reliable confidence scores from deep networks is of pivotal importance in critical decision-making systems, notably in the medical domain. While recent literature on calibrating deep segmentation networks has led to significant progress, their uncertainty is usually modeled by leveraging the information of individual pixels, which disregards the local structure of the object of interest. In particular, only the recent Spatially Varying Label Smoothing (SVLS) approach addresses this issue by softening the pixel label assignments with a discrete spatial Gaussian kernel. In this work, we first present a constrained optimization perspective of SVLS and demonstrate that it enforces an implicit constraint on soft class proportions of surrounding pixels. Furthermore, our analysis shows that SVLS lacks a mechanism to balance the contribution of the constraint with the primary objective, potentially hindering the optimization process. Based on these observations, we propose a principled and simple solution based on equality constraints on the logit values, which enables to control explicitly both the enforced constraint and the weight of the penalty, offering more flexibility. Comprehensive experiments on a variety of well-known segmentation benchmarks demonstrate the superior performance of the proposed approach.
updated: Sat Mar 11 2023 01:10:26 GMT+0000 (UTC)
published: Sat Mar 11 2023 01:10:26 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アソシエイト