arXiv reaDer
グラフフロー:デュアル効率的な医療画像セグメンテーションのためのクロスレイヤーグラフフロー蒸留
Graph Flow: Cross-layer Graph Flow Distillation for Dual Efficient Medical Image Segmentation
深い畳み込みニューラルネットワークの開発により、医療画像のセグメンテーションは、近年、一連のブレークスルーを達成しました。ただし、高性能の畳み込みニューラルネットワークは、常に多数のパラメーターと高い計算コストを意味するため、臨床シナリオでのアプリケーションの妨げになります。一方、大規模な注釈付き医用画像データセットの不足は、高性能ネットワークの適用をさらに妨げています。これらの問題に取り組むために、ネットワーク効率と注釈効率の両方の医療画像セグメンテーションのための包括的な知識蒸留フレームワークであるグラフフローを提案します。具体的には、コアのGraph Flow Distillationは、クロスレイヤーバリエーションの本質を、十分にトレーニングされた面倒な教師ネットワークから、トレーニングを受けていないコンパクトな学生ネットワークに転送します。さらに、教師なし言い換えモジュールは、教師ネットワークの知識を浄化するように設計されており、トレーニング手順の安定化にも役立ちます。さらに、敵対的蒸留とバニラロジット蒸留を統合することにより、統一された蒸留フレームワークを構築します。これにより、コンパクトなネットワークの最終的な予測をさらに洗練させることができます。胃癌セグメンテーションデータセットとシナプス多臓器セグメンテーションデータセットで実施された広範な実験は、これらの異なるモダリティとマルチカテゴリの医療画像データセットで最先端のパフォーマンスを達成する私たちの方法の卓越した能力を示しています。さらに、デュアル効率的な医療画像セグメンテーションのための新しい半教師ありパラダイムを通じて、グラフフローの有効性を示します。私たちのコードはグラフフローで入手できます。
With the development of deep convolutional neural networks, medical image segmentation has achieved a series of breakthroughs in recent years. However, the higher-performance convolutional neural networks always mean numerous parameters and high computation costs, which will hinder the applications in clinical scenarios. Meanwhile, the scarceness of large-scale annotated medical image datasets further impedes the application of high-performance networks. To tackle these problems, we propose Graph Flow, a comprehensive knowledge distillation framework, for both network-efficiency and annotation-efficiency medical image segmentation. Specifically, our core Graph Flow Distillation transfer the essence of cross-layer variations from a well-trained cumbersome teacher network to a non-trained compact student network. In addition, an unsupervised Paraphraser Module is designed to purify the knowledge of the teacher network, which is also beneficial for the stabilization of training procedure. Furthermore, we build a unified distillation framework by integrating the adversarial distillation and the vanilla logits distillation, which can further refine the final predictions of the compact network. Extensive experiments conducted on Gastric Cancer Segmentation Dataset and Synapse Multi-organ Segmentation Dataset demonstrate the prominent ability of our method which achieves state-of-the-art performance on these different-modality and multi-category medical image datasets. Moreover, we demonstrate the effectiveness of our Graph Flow through a new semi-supervised paradigm for dual efficient medical image segmentation. Our code will be available at Graph Flow.
updated: Sat May 07 2022 07:42:47 GMT+0000 (UTC)
published: Wed Mar 16 2022 14:56:02 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アソシエイト