arXiv reaDer
3D 画像セグメンテーションのためのトポロジーを考慮した焦点損失
Topology-Aware Focal Loss for 3D Image Segmentation
セグメンテーション アルゴリズムの有効性は、領域の重なり、接続の中断、ボイドなどのトポロジ エラーによって頻繁に損なわれます。この問題に取り組むために、新しい損失関数、つまりトポロジー対応焦点損失 (TAFL) を導入します。これは、従来の焦点損失に、グラウンド トゥルースと予測されたセグメンテーション マスクの永続性ダイアグラムの間のワッサースタイン距離に基づくトポロジー制約条件を組み込んだものです。グラウンド トゥルースと同一のトポロジを適用することにより、トポロジ制約はトポロジ エラーを効果的に解決できますが、Focal Loss はクラスの不均衡に対処します。まず、グラウンド トゥルースと予測されたセグメンテーション マスクのフィルター処理された立方体複合体からパーシスタンス ダイアグラムを作成します。その後、Sinkhorn-Knopp アルゴリズムを使用して、2 つの持続性ダイアグラム間の最適なトランスポート プランを決定します。結果として得られる輸送計画は、1 つのディストリビューションから別のディストリビューションに質量を輸送するコストを最小限に抑え、2 つの持続性ダイアグラムのポイント間のマッピングを提供します。次に、この移動計画に基づいてワッサースタイン距離を計算し、グラウンド トゥルースと予測されたマスクの間のトポロジーの非類似性を測定します。 MICCAI Brain Tumor Segmentation (BraTS) チャレンジ検証データセットを使用して 3D U-Net をトレーニングすることにより、アプローチを評価します。これには、悪性脳腫瘍の正確な識別と追跡のためにさまざまなモダリティを統合する 3D MRI スキャンの正確なセグメンテーションが必要です。次に、ペナルティ項としてトポロジー制約を追加して焦点損失を正則化することにより、セグメンテーションパフォーマンスの品質が向上することを示します。
The efficacy of segmentation algorithms is frequently compromised by topological errors like overlapping regions, disrupted connections, and voids. To tackle this problem, we introduce a novel loss function, namely Topology-Aware Focal Loss (TAFL), that incorporates the conventional Focal Loss with a topological constraint term based on the Wasserstein distance between the ground truth and predicted segmentation masks' persistence diagrams. By enforcing identical topology as the ground truth, the topological constraint can effectively resolve topological errors, while Focal Loss tackles class imbalance. We begin by constructing persistence diagrams from filtered cubical complexes of the ground truth and predicted segmentation masks. We subsequently utilize the Sinkhorn-Knopp algorithm to determine the optimal transport plan between the two persistence diagrams. The resultant transport plan minimizes the cost of transporting mass from one distribution to the other and provides a mapping between the points in the two persistence diagrams. We then compute the Wasserstein distance based on this travel plan to measure the topological dissimilarity between the ground truth and predicted masks. We evaluate our approach by training a 3D U-Net with the MICCAI Brain Tumor Segmentation (BraTS) challenge validation dataset, which requires accurate segmentation of 3D MRI scans that integrate various modalities for the precise identification and tracking of malignant brain tumors. Then, we demonstrate that the quality of segmentation performance is enhanced by regularizing the focal loss through the addition of a topological constraint as a penalty term.
updated: Thu Apr 27 2023 15:54:41 GMT+0000 (UTC)
published: Mon Apr 24 2023 16:07:17 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アソシエイト