arXiv reaDer
CoCo DistillNet:病理学的胃癌セグメンテーションのためのクロスレイヤー相関蒸留ネットワーク
CoCo DistillNet: a Cross-layer Correlation Distillation Network for Pathological Gastric Cancer Segmentation
近年、深い畳み込みニューラルネットワークは病理画像のセグメンテーションに大きな進歩を遂げました。ただし、病理画像のセグメンテーションは、より高性能なネットワークが一般により多くの計算リソースとストレージを必要とするというジレンマに直面します。この現象は、病理画像の固有の高解像度のために、実際のシーンでの高精度ネットワークの使用を制限します。この問題に取り組むために、我々は、病理学的胃癌セグメンテーションのための新しいクロスレイヤー相関(CoCo)知識蒸留ネットワークであるCoCoDistillNetを提案します。知識蒸留は、面倒なネットワークからの知識の伝達を通じてコンパクトなネットワークのパフォーマンスを向上させることを目的とした一般的な手法です。具体的には、CoCo DistillNetは、異なるレイヤー間のチャネル混合空間類似性の相関関係をモデル化し、この知識を事前にトレーニングされた面倒な教師ネットワークからトレーニングされていないコンパクトな学生ネットワークに転送します。さらに、敵対的学習戦略を利用して、敵対的蒸留(AD)と呼ばれる蒸留手順をさらに促進します。さらに、トレーニング手順を安定させるために、教師なしパラフレーズモジュール(PM)を使用して、教師ネットワークの知識の言い換えを強化します。その結果、胃がんセグメンテーションデータセットで実施された広範な実験は、最先端のパフォーマンスを実現するCoCoDistillNetの卓越した能力を実証しています。
In recent years, deep convolutional neural networks have made significant advances in pathology image segmentation. However, pathology image segmentation encounters with a dilemma in which the higher-performance networks generally require more computational resources and storage. This phenomenon limits the employment of high-accuracy networks in real scenes due to the inherent high-resolution of pathological images. To tackle this problem, we propose CoCo DistillNet, a novel Cross-layer Correlation (CoCo) knowledge distillation network for pathological gastric cancer segmentation. Knowledge distillation, a general technique which aims at improving the performance of a compact network through knowledge transfer from a cumbersome network. Concretely, our CoCo DistillNet models the correlations of channel-mixed spatial similarity between different layers and then transfers this knowledge from a pre-trained cumbersome teacher network to a non-trained compact student network. In addition, we also utilize the adversarial learning strategy to further prompt the distilling procedure which is called Adversarial Distillation (AD). Furthermore, to stabilize our training procedure, we make the use of the unsupervised Paraphraser Module (PM) to boost the knowledge paraphrase in the teacher network. As a result, extensive experiments conducted on the Gastric Cancer Segmentation Dataset demonstrate the prominent ability of CoCo DistillNet which achieves state-of-the-art performance.
updated: Fri Nov 12 2021 13:44:42 GMT+0000 (UTC)
published: Fri Aug 27 2021 08:36:00 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アソシエイト