arXiv reaDer
医学的診断のためのユニモーダル均一制約付きWassersteinトレーニング
Unimodal-uniform Constrained Wasserstein Training for Medical Diagnosis
 医療診断タスクのラベルは通常、個別で連続的に配布されます。たとえば、糖尿病性網膜症診断(DR)には、5つの健康リスクレベルが含まれます:DRなし(0)、軽度のDR(1)、中程度のDR(2)、重度のDR(3)および増殖性DR(4)。このラベリングシステムは、医学的疾患に一般的です。以前のメソッドは、通常、マルチバイナリ分類タスクを構築するか、出力ユニットでいくつかの再パラメータースキームを提案します。この論文では、損失関数の観点からこのタスクを対象としています。より具体的には、Wasserstein距離が代替手段として利用され、その地上メトリックを事前定義することでクラス間の相関を明示的に組み込みます。次に、線形、凸または凹の増加関数w.r.tとして機能する地上メトリック。線のユークリッド距離は、最適化の観点から検討されます。一方、この論文では、単峰性の均一混合分布を使用して、インライアノイズとアウトライアノイズをモデル化する平滑化ターゲットラベルを構築することも提案しています。ワンホット設定とは異なり、平滑化されたラベルは、より困難な機能を備えたワッサーシュタイン距離の計算を妨げます。ワンホットまたはスムージングされたターゲットラベルを使用して、このペーパーでは実用的なクローズドフォームソリューションを体系的にまとめています。いくつかの医療診断タスク(例:糖尿病性網膜症および超音波乳房データセット)で方法を評価し、最先端のパフォーマンスを達成します。
The labels in medical diagnosis task are usually discrete and successively distributed. For example, the Diabetic Retinopathy Diagnosis (DR) involves five health risk levels: no DR (0), mild DR (1), moderate DR (2), severe DR (3) and proliferative DR (4). This labeling system is common for medical disease. Previous methods usually construct a multi-binary-classification task or propose some re-parameter schemes in the output unit. In this paper, we target on this task from the perspective of loss function. More specifically, the Wasserstein distance is utilized as an alternative, explicitly incorporating the inter-class correlations by pre-defining its ground metric. Then, the ground metric which serves as a linear, convex or concave increasing function w.r.t. the Euclidean distance in a line is explored from an optimization perspective. Meanwhile, this paper also proposes of constructing the smoothed target labels that model the inlier and outlier noises by using a unimodal-uniform mixture distribution. Different from the one-hot setting, the smoothed label endues the computation of Wasserstein distance with more challenging features. With either one-hot or smoothed target label, this paper systematically concludes the practical closed-form solution. We evaluate our method on several medical diagnosis tasks (e.g., Diabetic Retinopathy and Ultrasound Breast dataset) and achieve state-of-the-art performance.
updated: Sun Nov 03 2019 20:41:14 GMT+0000 (UTC)
published: Sun Nov 03 2019 20:41:14 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アソシエイト