画像セグメンテーションは画像処理の基本的なトピックであり、何十年にもわたって研究されてきました。深層学習ベースの教師ありセグメンテーションモデルは、最先端のパフォーマンスを実現していますが、それらのほとんどは、幾何学的制約のないトレーニングにピクセル単位の損失関数を使用することによって制限されています。オイラーのエラスティカモデルとディープラーニングの分野に導入された最近のアクティブコンターモデルに触発され、エラスティカ(曲率と長さ)と領域情報を画像セグメンテーションの幾何学的に自然な制約として組み込んだエラスティカ(ACE)損失関数を備えた新しいアクティブコンターを提案しますタスク。 ACE損失関数で曲率を表す前のより効果的な画像として、平均曲率、つまりすべての主曲率の平均を紹介します。さらに、平均曲率の定義に基づいて、3D画像セグメンテーションにラプラス演算子を使用して3次元(3D)でACE損失を概算する高速ソリューションを提案します。 4つの2Dおよび3Dの自然および生物医学画像データセットでACE損失関数を評価します。我々の結果は、提案された損失関数が、異なるセグメンテーションネットワーク上の他の主流の損失関数よりも優れていることを示しています。ソースコードはhttps://github.com/HiLab-git/ACELossで入手できます。
Image segmentation is a fundamental topic in image processing and has been studied for many decades. Deep learning-based supervised segmentation models have achieved state-of-the-art performance but most of them are limited by using pixel-wise loss functions for training without geometrical constraints. Inspired by Euler's Elastica model and recent active contour models introduced into the field of deep learning, we propose a novel active contour with elastica (ACE) loss function incorporating Elastica (curvature and length) and region information as geometrically-natural constraints for the image segmentation tasks. We introduce the mean curvature i.e. the average of all principal curvatures, as a more effective image prior to representing curvature in our ACE loss function. Furthermore, based on the definition of the mean curvature, we propose a fast solution to approximate the ACE loss in three-dimensional (3D) by using Laplace operators for 3D image segmentation. We evaluate our ACE loss function on four 2D and 3D natural and biomedical image datasets. Our results show that the proposed loss function outperforms other mainstream loss functions on different segmentation networks. Our source code is available at https://github.com/HiLab-git/ACELoss.