生物医学画像の多くのセグメンテーションタスクは、エネルギー関数の最小化としてモデル化でき、最大フローおよび最小カット最適化アルゴリズムのクラスによって解決できます。ただし、従来のエネルギー関数は通常、エネルギー関数で手作りの特徴を使用するため、セグメンテーションの精度は、さまざまなセグメント化オブジェクトのセマンティック特徴の対比に敏感です。これらの制限に対処するために、エンドツーエンドのトレーニング可能なニューラルネットワーク機能をエネルギー関数に組み込むことを提案します。私たちのディープニューラルネットワーク機能は、U-netのスキップ接続を使用してダウンサンプリングレイヤーとアップサンプリングレイヤーから抽出されます。推論段階では、学習された特徴がエネルギー関数に供給されます。そして、セグメンテーションは、ADMMソルバーによって主双対形式で解決されます。トレーニング段階では、デュアルフォームの最適条件から導出される最小カット関数とフロー保存関数の正則化を使用して、プライマリフォームのエネルギー関数を最適化することによってニューラルネットワークをトレーニングします。脳腫瘍のセグメンテーションタスクで、定性的および定量的にメソッドを評価します。エネルギー最小化モデルが感度と滑らかな境界のバランスを実現するにつれて、医師の診断のためのアンサンブル参照として、反復を通じてセグメンテーションの等高線がどのように活発に進化するかを示します。
Many segmentation tasks for biomedical images can be modeled as the minimization of an energy function and solved by a class of max-flow and min-cut optimization algorithms. However, the segmentation accuracy is sensitive to the contrasting of semantic features of different segmenting objects, as the traditional energy function usually uses hand-crafted features in their energy functions. To address these limitations, we propose to incorporate end-to-end trainable neural network features into the energy functions. Our deep neural network features are extracted from the down-sampling and up-sampling layers with skip-connections of a U-net. In the inference stage, the learned features are fed into the energy functions. And the segmentations are solved in a primal-dual form by ADMM solvers. In the training stage, we train our neural networks by optimizing the energy function in the primal form with regularizations on the min-cut and flow-conservation functions, which are derived from the optimal conditions in the dual form. We evaluate our methods, both qualitatively and quantitatively, in a brain tumor segmentation task. As the energy minimization model achieves a balance on sensitivity and smooth boundaries, we would show how our segmentation contours evolve actively through iterations as ensemble references for doctor diagnosis.