arXiv reaDer
MI-SegNet: 目に見えないドメインの一般化のための相互情報ベースの US セグメンテーション
MI-SegNet: Mutual Information-Based US Segmentation for Unseen Domain Generalization
ドメイン間での学習ベースの医用画像セグメンテーションの一般化機能は、現在、特に超音波 (US) イメージングの場合、ドメイン シフトによって引き起こされるパフォーマンスの低下によって制限されています。米国の画像の品質は、慎重に調整された音響パラメーターに大きく依存しており、超音波検査者、機械、および設定によって異なります。ドメイン全体の米国画像の一般化可能性を向上させるために、解剖学的およびドメインの特徴表現を明示的に解くための新しい相互情報 (MI) ベースのフレームワークである MI-SegNet を提案します。したがって、堅牢なドメインに依存しないセグメンテーションが期待できます。もつれを解くための関連する特徴を抽出するために、2 つのエンコーダーが使用されます。セグメンテーションは、その予測に解剖学的特徴マップのみを使用します。エンコーダーに意味のある特徴表現を強制的に学習させるために、トレーニング中に相互再構成法が使用されます。ドメインまたは解剖学のいずれかに固有の変換が適用され、それぞれの特徴抽出タスクでエンコーダーをガイドします。さらに、両方の機能マップに存在する MI は、個別の機能空間をさらに促進するために処罰されます。さまざまなパラメーターとマシンを使用したいくつかのデータセットで、提案されたドメインに依存しないセグメンテーション アプローチの一般化可能性を検証します。さらに、最先端のネットワークと比較することにより、事前トレーニング済みモデルとして機能する提案された MI-SegNet の有効性を実証します。
Generalization capabilities of learning-based medical image segmentation across domains are currently limited by the performance degradation caused by the domain shift, particularly for ultrasound (US) imaging. The quality of US images heavily relies on carefully tuned acoustic parameters, which vary across sonographers, machines, and settings. To improve the generalizability on US images across domains, we propose MI-SegNet, a novel mutual information (MI) based framework to explicitly disentangle the anatomical and domain feature representations; therefore, robust domain-independent segmentation can be expected. Two encoders are employed to extract the relevant features for the disentanglement. The segmentation only uses the anatomical feature map for its prediction. In order to force the encoders to learn meaningful feature representations a cross-reconstruction method is used during training. Transformations, specific to either domain or anatomy are applied to guide the encoders in their respective feature extraction task. Additionally, any MI present in both feature maps is punished to further promote separate feature spaces. We validate the generalizability of the proposed domain-independent segmentation approach on several datasets with varying parameters and machines. Furthermore, we demonstrate the effectiveness of the proposed MI-SegNet serving as a pre-trained model by comparing it with state-of-the-art networks.
updated: Tue Feb 06 2024 16:55:14 GMT+0000 (UTC)
published: Wed Mar 22 2023 15:30:44 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト