arXiv reaDer
医療画像セグメンテーションの確率とエラーに対する出力活性化関数の影響
Effect of the output activation function on the probabilities and errors in medical image segmentation
シグモイド活性化は、ニューラルネットワークを使用したバイナリ分類およびセグメンテーションにおける標準出力活性化関数です。それでも、医療画像のセグメンテーションの結果を改善する可能性のある、他のさまざまな潜在的な出力活性化関数が存在します。この作業では、さまざまな出力のアクティブ化と損失関数の漸近的な動作が、予測確率と対応するセグメンテーションエラーにどのように影響するかを検討します。クロスエントロピーの場合、活性化関数の変化率が速いほど予測が向上し、変化率が遅いと確率のキャリブレーションが向上することを示します。ダイスロスについては、アークタンジェント活性化関数がシグモイド関数よりも優れていることがわかりました。さらに、医療画像セグメンテーションの分野で任意の出力活性化関数のテストスペースを提供します。 4つの異なる医療画像セグメンテーションタスクで3つの損失関数と組み合わせて7つの活性化関数をテストし、このアプリケーションシナリオに最適な関数の分類を提供しました。
The sigmoid activation is the standard output activation function in binary classification and segmentation with neural networks. Still, there exist a variety of other potential output activation functions, which may lead to improved results in medical image segmentation. In this work, we consider how the asymptotic behavior of different output activation and loss functions affects the prediction probabilities and the corresponding segmentation errors. For cross entropy, we show that a faster rate of change of the activation function correlates with better predictions, while a slower rate of change can improve the calibration of probabilities. For dice loss, we found that the arctangent activation function is superior to the sigmoid function. Furthermore, we provide a test space for arbitrary output activation functions in the area of medical image segmentation. We tested seven activation functions in combination with three loss functions on four different medical image segmentation tasks to provide a classification of which function is best suited in this application scenario.
updated: Thu Sep 02 2021 12:51:14 GMT+0000 (UTC)
published: Thu Sep 02 2021 12:51: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アソシエイト