arXiv reaDer
ヘマトキシリンおよびエオシン染色された組織学的画像における核インスタンスセグメンテーションのための深部神経ネットワーク剪定
Deep Neural Network Pruning for Nuclei Instance Segmentation in Hematoxylin & Eosin-Stained Histological Images
最近、ディープニューラルネットワーク(DNN)のプルーニングは、精度と一般化能力の向上、ネットワークサイズの縮小、および特殊なハードウェアでの推論速度の向上のために多くの注目を集めています。剪定は主にコンピュータービジョンタスクでテストされましたが、医療画像分析のコンテキストでのそのアプリケーションはほとんど検討されていません。この作業では、組織学的画像の核インスタンスのセグメンテーションパフォーマンスに対する、よく知られているプルーニング手法、つまりレイヤーワイズおよびネットワーク全体のマグニチュードプルーニングの影響を調査します。使用されるインスタンスセグメンテーションモデルは、(1)セマンティックセグメンテーションブランチと(2)ディープ回帰ブランチの2つの主要なブランチで構成されています。両方のブランチのパフォーマンスと最終的な核インスタンスのセグメンテーション結果に対する重みの剪定の影響を調査します。 2つの公開されているデータセットで評価した結果、レイヤーごとのプルーニングは、圧縮率(CR)が小さい場合はネットワーク全体のプルーニングよりもわずかに優れたパフォーマンスを提供し、CRが大きい場合はネットワーク全体のプルーニングが優れたパフォーマンスを提供することを示しています。セマンティックセグメンテーション、ディープリグレッション、および最終インスタンスセグメンテーションの場合、モデルの重みの93.75%、95%、および80%は、それぞれのモデルのパフォーマンスの低下が2%未満で、レイヤーごとのプルーニングによってプルーニングできます。
Recently, pruning deep neural networks (DNNs) has received a lot of attention for improving accuracy and generalization power, reducing network size, and increasing inference speed on specialized hardwares. Although pruning was mainly tested on computer vision tasks, its application in the context of medical image analysis has hardly been explored. This work investigates the impact of well-known pruning techniques, namely layer-wise and network-wide magnitude pruning, on the nuclei instance segmentation performance in histological images. Our utilized instance segmentation model consists of two main branches: (1) a semantic segmentation branch, and (2) a deep regression branch. We investigate the impact of weight pruning on the performance of both branches separately and on the final nuclei instance segmentation result. Evaluated on two publicly available datasets, our results show that layer-wise pruning delivers slightly better performance than networkwide pruning for small compression ratios (CRs) while for large CRs, network-wide pruning yields superior performance. For semantic segmentation, deep regression and final instance segmentation, 93.75 %, 95 %, and 80 % of the model weights can be pruned by layer-wise pruning with less than 2 % reduction in the performance of respective models.
updated: Wed Jun 15 2022 09:52:41 GMT+0000 (UTC)
published: Wed Jun 15 2022 09:52:41 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アソシエイト