arXiv reaDer
解剖学的に妥当なセグメンテーションのためのコンテキストアウェア仮想敵対トレーニング
Context-aware virtual adversarial training for anatomically-plausible segmentation
卓越した精度にもかかわらず、ディープニューラルネットワークに基づく半教師ありセグメンテーション手法では、たとえば、穴や切断された領域を含むなど、臨床医が解剖学的に不可能と見なす予測を生成できます。この問題を解決するために、解剖学的にもっともらしいセグメンテーションを生成するためのコンテキストアウェア仮想敵対トレーニング(CaVAT)メソッドを提示します。精度のみに焦点を当てたアプローチとは異なり、私たちの方法では、微分可能な損失関数で簡単にモデル化できない接続性などの複雑なトポロジー制約も考慮します。敵対的トレーニングを使用して制約に違反する例を生成するため、ネットワークは新しい例でこのような誤った予測を行わないように学習し、Reinforceアルゴリズムを使用して微分不可能なセグメンテーション制約を処理できます。提案された方法は、任意のセグメンテーションネットワークの上に任意の制約を追加するための一般的で効率的な方法を提供します。 2つの臨床的に関連するデータセットでの実験は、領域の接続性に関して正確で解剖学的に妥当なセグメンテーションを生成する方法を示しています。
Despite their outstanding accuracy, semi-supervised segmentation methods based on deep neural networks can still yield predictions that are considered anatomically impossible by clinicians, for instance, containing holes or disconnected regions. To solve this problem, we present a Context-aware Virtual Adversarial Training (CaVAT) method for generating anatomically plausible segmentation. Unlike approaches focusing solely on accuracy, our method also considers complex topological constraints like connectivity which cannot be easily modeled in a differentiable loss function. We use adversarial training to generate examples violating the constraints, so the network can learn to avoid making such incorrect predictions on new examples, and employ the Reinforce algorithm to handle non-differentiable segmentation constraints. The proposed method offers a generic and efficient way to add any constraint on top of any segmentation network. Experiments on two clinically-relevant datasets show our method to produce segmentations that are both accurate and anatomically-plausible in terms of region connectivity.
updated: Mon Jul 12 2021 16:01:27 GMT+0000 (UTC)
published: Mon Jul 12 2021 16:01:27 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アソシエイト