深い畳み込みニューラル ネットワーク (CNN) は、医療画像のセグメンテーションに広く使用されています。ほとんどの研究では、最終的なセグメンテーション結果を計算するために出力レイヤーのみが利用されており、ディープ ラーニングされた特徴の隠れた表現は十分に理解されていません。この論文では、深い特徴に基づいてバイナリ セグメンテーション マップを計算するためのプロトタイプ セグメンテーション (ProtoSeg) メソッドを提案します。セグメンテーション能力スコア (略して SA スコア) と呼ばれる、特徴セグメンテーション マップとグラウンド トゥルースの間のダイスを計算することにより、特徴のセグメンテーション能力を測定します。対応する SA スコアは、さまざまなレイヤーとユニットの深い特徴のセグメンテーション能力を定量化し、セグメンテーションのためのディープ ニューラル ネットワークを理解することができます。さらに、私たちの方法は、グラウンドトゥルースなしでテスト画像の出力のパフォーマンス推定を与えることができる平均SAスコアを提供できます。最後に、提案された ProtoSeg メソッドを使用して、入力画像のセグメンテーション マップを直接計算し、各入力画像のセグメンテーション能力をさらに理解します。結果は、脳 MRI における腫瘍のセグメンテーション、皮膚画像における病変、CT 画像における COVID 関連の異常、腹部 MRI における前立腺のセグメンテーション、および CT 画像における膵臓の塊のセグメンテーションについて提示されています。私たちの方法は、医療画像セグメンテーションのための解釈と説明可能な AI システムのための新しい洞察を提供できます。コードは https://github.com/shengfly/ProtoSeg で入手できます。
Deep convolutional neural networks (CNNs) have been widely used for medical image segmentation. In most studies, only the output layer is exploited to compute the final segmentation results and the hidden representations of the deep learned features have not been well understood. In this paper, we propose a prototype segmentation (ProtoSeg) method to compute a binary segmentation map based on deep features. We measure the segmentation abilities of the features by computing the Dice between the feature segmentation map and ground-truth, named as the segmentation ability score (SA score for short). The corresponding SA score can quantify the segmentation abilities of deep features in different layers and units to understand the deep neural networks for segmentation. In addition, our method can provide a mean SA score which can give a performance estimation of the output on the test images without ground-truth. Finally, we use the proposed ProtoSeg method to compute the segmentation map directly on input images to further understand the segmentation ability of each input image. Results are presented on segmenting tumors in brain MRI, lesions in skin images, COVID-related abnormality in CT images, prostate segmentation in abdominal MRI, and pancreatic mass segmentation in CT images. Our method can provide new insights for interpreting and explainable AI systems for medical image segmentation. Our code is available on: https://github.com/shengfly/ProtoSeg.