arXiv reaDer
リアルタイム ポリープ セグメンテーションのためのニューラル ネットワーク プルーニング
Neural Network Pruning for Real-time Polyp Segmentation
コンピューター支援治療は、深層学習モデルの有効性により、医療画像処理の実行可能な応用として浮上しています。リアルタイムの推論速度は、医療従事者を支援するこのようなアプリケーションにとって依然として重要な要件です。一般に、パフォーマンスとモデル サイズの間にはトレードオフが存在しますが、モデル サイズを犠牲にして、ほぼ元のパフォーマンスを維持するために多大な努力が払われてきました。ニューラル ネットワーク プルーニングは、冗長なパラメーターを削除して推論を高速化することを目的としたエキサイティングな分野として浮上しました。この研究では、ポリープのセグメンテーションにおけるニューラル ネットワークの枝刈りの応用を示します。畳み込みフィルタの重要度スコアを計算し、プルーニングのある程度の値まではパフォーマンスを低下させない、最小のスコアを持つフィルタを削除します。重要度スコアを計算するには、特定のフィルターを除去するためのネットワーク出力の変化の Taylor First Order (TaylorFO) 近似を使用します。具体的には、重要度スコアの計算に勾配正規化逆伝播を採用します。ポリープ データセットでの実験を通じて、私たちのアプローチがパラメーター数を大幅に削減し、同等のパフォーマンスを維持できる FLOP を検証しました。
Computer-assisted treatment has emerged as a viable application of medical imaging, owing to the efficacy of deep learning models. Real-time inference speed remains a key requirement for such applications to help medical personnel. Even though there generally exists a trade-off between performance and model size, impressive efforts have been made to retain near-original performance by compromising model size. Neural network pruning has emerged as an exciting area that aims to eliminate redundant parameters to make the inference faster. In this study, we show an application of neural network pruning in polyp segmentation. We compute the importance score of convolutional filters and remove the filters having the least scores, which to some value of pruning does not degrade the performance. For computing the importance score, we use the Taylor First Order (TaylorFO) approximation of the change in network output for the removal of certain filters. Specifically, we employ a gradient-normalized backpropagation for the computation of the importance score. Through experiments in the polyp datasets, we validate that our approach can significantly reduce the parameter count and FLOPs retaining similar performance.
updated: Thu Jun 22 2023 21:03:50 GMT+0000 (UTC)
published: Thu Jun 22 2023 21:03:50 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アソシエイト