arXiv reaDer
LightVessel: 類似性知識蒸留による軽量冠動脈血管セグメンテーションの探索
LightVessel: Exploring Lightweight Coronary Artery Vessel Segmentation via Similarity Knowledge Distillation
近年、深層畳み込みニューラル ネットワーク (DCNN) は、冠動脈血管セグメンテーションにおいて大きな展望を達成しています。ただし、高性能アプローチには過剰なパラメーターと高い計算コストがあるため、複雑なモデルを臨床シナリオに展開することは困難です。この問題に取り組むために、軽量の冠動脈血管セグメンテーションのために、Similarity Knowledge Distillation Framework である LightVessel を提案します。主に、セマンティック シフト モデリングのための特徴ごとの類似性蒸留 (FSD) モジュールを提案します。具体的には、エンコーダーとデコーダーからの対称レイヤー間の特徴の類似性を計算します。次に、類似性は、面倒な教師ネットワークからトレーニングされていない軽量の学生ネットワークに知識として転送されます。一方、学生モデルがより多くのピクセル単位のセマンティック情報を学習することを奨励するために、Adversarial Similarity Distillation (ASD) モジュールを導入します。具体的には、ASD モジュールは、それぞれ教師モデルと生徒モデルからの注釈と予測の間の空間的な敵対的相関関係を構築することを目的としています。 ASD モジュールを介して、学生モデルは、冠状動脈血管の細粒度の微妙なエッジ セグメント化された結果を取得します。臨床冠動脈血管データセットで実施された広範な実験は、LightVessel がさまざまな知識蒸留の対応物よりも優れていることを示しています。
In recent years, deep convolution neural networks (DCNNs) have achieved great prospects in coronary artery vessel segmentation. However, it is difficult to deploy complicated models in clinical scenarios since high-performance approaches have excessive parameters and high computation costs. To tackle this problem, we propose LightVessel, a Similarity Knowledge Distillation Framework, for lightweight coronary artery vessel segmentation. Primarily, we propose a Feature-wise Similarity Distillation (FSD) module for semantic-shift modeling. Specifically, we calculate the feature similarity between the symmetric layers from the encoder and decoder. Then the similarity is transferred as knowledge from a cumbersome teacher network to a non-trained lightweight student network. Meanwhile, for encouraging the student model to learn more pixel-wise semantic information, we introduce the Adversarial Similarity Distillation (ASD) module. Concretely, the ASD module aims to construct the spatial adversarial correlation between the annotation and prediction from the teacher and student models, respectively. Through the ASD module, the student model obtains fined-grained subtle edge segmented results of the coronary artery vessel. Extensive experiments conducted on Clinical Coronary Artery Vessel Dataset demonstrate that LightVessel outperforms various knowledge distillation counterparts.
updated: Wed Nov 02 2022 05:49:19 GMT+0000 (UTC)
published: Wed Nov 02 2022 05:49:19 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アソシエイト