arXiv reaDer
医療画像セグメンテーションのための明示的な形状事前分布を使用した学習
Learning with Explicit Shape Priors for Medical Image Segmentation
医用画像のセグメンテーションは、医用画像分析と手術計画の基本的なタスクです。近年、UNet ベースのネットワークが医療画像セグメンテーションの分野で普及しています。しかし、畳み込みニューラル ネットワーク (CNN) は受容野が限られているため、臓器や腫瘍の長距離依存性をモデル化できません。さらに、これらのモデルは、最終セグメンテーション ヘッドのトレーニングに大きく依存しています。そして、既存の方法では、これら 2 つの制限に同時にうまく対処できません。したがって、私たちの研究では、形状事前モジュール (SPM) を提案しました。これは、形状事前モジュールを明示的に導入して、UNet ベースのモデルのセグメンテーション パフォーマンスを向上させることができます。明示的な形状事前分布は、グローバル形状事前分布とローカル形状事前分布から構成されます。前者の粗い形状表現は、ネットワークにグローバル コンテキストをモデル化する機能を提供します。より詳細な形状情報を含む後者は、セグメンテーション パフォーマンスを向上させるための追加のガイダンスとして機能し、セグメンテーション ヘッド内の学習可能なプロトタイプへの大きな依存を軽減します。 SPM の有効性を評価するために、私たちは 3 つの困難な公開データセットで実験を実施しました。そして私たちが提案するモデルは最先端のパフォーマンスを実現します。さらに、SPM は、古典的な CNN と最近の Transformer ベースのバックボーンに対して優れた一般化能力を示し、さまざまなデータセットのセグメンテーション タスクのプラグ アンド プレイ構造として機能します。ソース コードは https://github.com/AlexYouXin/Explicit-Shape-Priors で入手できます。
Medical image segmentation is a fundamental task for medical image analysis and surgical planning. In recent years, UNet-based networks have prevailed in the field of medical image segmentation. However, convolution-neural networks (CNNs) suffer from limited receptive fields, which fail to model the long-range dependency of organs or tumors. Besides, these models are heavily dependent on the training of the final segmentation head. And existing methods can not well address these two limitations at the same time. Hence, in our work, we proposed a novel shape prior module (SPM), which can explicitly introduce shape priors to promote the segmentation performance of UNet-based models. The explicit shape priors consist of global and local shape priors. The former with coarse shape representations provides networks with capabilities to model global contexts. The latter with finer shape information serves as additional guidance to boost the segmentation performance, which relieves the heavy dependence on the learnable prototype in the segmentation head. To evaluate the effectiveness of SPM, we conduct experiments on three challenging public datasets. And our proposed model achieves state-of-the-art performance. Furthermore, SPM shows an outstanding generalization ability on classic CNNs and recent Transformer-based backbones, which can serve as a plug-and-play structure for the segmentation task of different datasets. Source codes are available at https://github.com/AlexYouXin/Explicit-Shape-Priors
updated: Sun Jun 04 2023 15:07:55 GMT+0000 (UTC)
published: Fri Mar 31 2023 11:12:35 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アソシエイト