自然画像や生物医学画像からの微細構造のセグメンテーションでは、ピクセルごとの精度だけが問題となる指標ではありません。血管の接続性や膜の閉鎖などのトポロジーの正確さは、下流の分析タスクにとって非常に重要です。この論文では、より良いトポロジー精度のために深い画像セグメンテーションネットワークを訓練するための新しいアプローチを提案します。特に、離散モース理論(DMT)の力を活用して、トポロジーの精度に重要な1Dスケルトンや2Dパッチなどのグローバル構造を特定します。これらのグローバル構造に基づいた新しい損失でトレーニングされたネットワークパフォーマンスは、特にトポロジ的に困難な場所(接続や膜の弱点など)の近くで大幅に改善されます。多様なデータセットで、私たちの方法はDICEスコアとトポロジーメトリックの両方で優れたパフォーマンスを実現します。
In the segmentation of fine-scale structures from natural and biomedical images, per-pixel accuracy is not the only metric of concern. Topological correctness, such as vessel connectivity and membrane closure, is crucial for downstream analysis tasks. In this paper, we propose a new approach to train deep image segmentation networks for better topological accuracy. In particular, leveraging the power of discrete Morse theory (DMT), we identify global structures, including 1D skeletons and 2D patches, which are important for topological accuracy. Trained with a novel loss based on these global structures, the network performance is significantly improved especially near topologically challenging locations (such as weak spots of connections and membranes). On diverse datasets, our method achieves superior performance on both the DICE score and topological metrics.