arXiv reaDer
限られたバッチサイズのセグメンテーションニューラルネットワークの正規化方法の比較
Comparing Normalization Methods for Limited Batch Size Segmentation Neural Networks
バッチ正規化が広く使用されているため、より深いニューラルネットワークのトレーニングが可能になり、より安定した高速な結果が得られます。ただし、バッチ正規化は、トレーニング中に大きなバッチサイズを使用すると最適に機能します。また、最先端のセグメンテーション畳み込みニューラルネットワークアーキテクチャは非常にメモリを必要とするため、現在のハードウェアでは大きなバッチサイズを実現できないことがよくあります。 3DCTスキャンからのバイナリスパインセグメンテーションの問題でこの問題を解決するために提案された代替の正規化方法を評価します。私たちの結果は、限られたバッチサイズのニューラルネットワークトレーニング環境でのインスタンス正規化の有効性を示しています。比較したすべての方法の中で、インスタンスの正規化はダイス係数= 0.96で最高の結果を達成しました。これは、トレーニング時間が長い、より深いネットワークによって達成された以前の結果に匹敵します。また、この実験で使用されたインスタンス正規化の実装は、正規化方法のないネットワークと比較した場合、計算時間が効率的であることも示しています。
The widespread use of Batch Normalization has enabled training deeper neural networks with more stable and faster results. However, the Batch Normalization works best using large batch size during training and as the state-of-the-art segmentation convolutional neural network architectures are very memory demanding, large batch size is often impossible to achieve on current hardware. We evaluate the alternative normalization methods proposed to solve this issue on a problem of binary spine segmentation from 3D CT scan. Our results show the effectiveness of Instance Normalization in the limited batch size neural network training environment. Out of all the compared methods the Instance Normalization achieved the highest result with Dice coefficient = 0.96 which is comparable to our previous results achieved by deeper network with longer training time. We also show that the Instance Normalization implementation used in this experiment is computational time efficient when compared to the network without any normalization method.
updated: Mon Nov 23 2020 17:13:24 GMT+0000 (UTC)
published: Mon Nov 23 2020 17:13:24 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アソシエイト