arXiv reaDer
コンピュータ断層撮影画像における大動脈および大血管セグメンテーションのトポロジーを意識した損失
Topology-Aware Loss for Aorta and Great Vessel Segmentation in Computed Tomography Images
セグメンテーション ネットワークは、標準の損失関数を使用してトレーニングされる場合、オブジェクトの形状や複数のオブジェクト間のジオメトリなど、画像のグローバルな不変量を学習するために明示的に適用されません。一方、そのような不変式をネットワーク トレーニングに組み込むと、それらがセグメント化されるオブジェクトの固有の特性である場合、さまざまなセグメント化タスクのパフォーマンスを向上させるのに役立つ可能性があります。一例として、コンピュータ断層撮影 (CT) 画像における大動脈と大血管のセグメンテーションが挙げられます。この場合、血管は人間の解剖学的構造により体内の特定の形状で検出され、2D CT 画像ではほとんどが丸いオブジェクトとして見えます。この論文では、永続的な相同性を通じてグラウンド トゥルースと予測の間のトポロジーの相違にペナルティを与える、新しいトポロジーを意識した損失関数を導入することで、この問題に対処します。予測マップの尤度関数とグラウンド トゥルースの Betti 数にしきい値フィルターを適用する、以前に提案されたセグメンテーション ネットワーク設計とは異なり、この論文では、Vietoris-Rips フィルターを適用して、グラウンド トゥルースとグラウンド トゥルースの両方の永続図を取得することを提案します。予測マップを作成し、対応する永続性ダイアグラム間の Wasserstein 距離との非類似性を計算します。このフィルタリングを使用すると、形状とジオメトリを同時にモデリングできるという利点がありますが、しきい値フィルタリングが適用されている場合にはこのようなことは起こらない可能性があります。 24 人の被験者の 4327 枚の CT 画像に対する実験では、提案されたトポロジー認識損失関数が対応する損失関数よりも優れた結果をもたらし、この使用の有効性を示していることが明らかになりました。
Segmentation networks are not explicitly imposed to learn global invariants of an image, such as the shape of an object and the geometry between multiple objects, when they are trained with a standard loss function. On the other hand, incorporating such invariants into network training may help improve performance for various segmentation tasks when they are the intrinsic characteristics of the objects to be segmented. One example is segmentation of aorta and great vessels in computed tomography (CT) images where vessels are found in a particular geometry in the body due to the human anatomy and they mostly seem as round objects on a 2D CT image. This paper addresses this issue by introducing a new topology-aware loss function that penalizes topology dissimilarities between the ground truth and prediction through persistent homology. Different from the previously suggested segmentation network designs, which apply the threshold filtration on a likelihood function of the prediction map and the Betti numbers of the ground truth, this paper proposes to apply the Vietoris-Rips filtration to obtain persistence diagrams of both ground truth and prediction maps and calculate the dissimilarity with the Wasserstein distance between the corresponding persistence diagrams. The use of this filtration has advantage of modeling shape and geometry at the same time, which may not happen when the threshold filtration is applied. Our experiments on 4327 CT images of 24 subjects reveal that the proposed topology-aware loss function leads to better results than its counterparts, indicating the effectiveness of this use.
updated: Sat Feb 24 2024 10:44:33 GMT+0000 (UTC)
published: Thu Jul 06 2023 17:06:49 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アソシエイト