arXiv reaDer
半教師あり3D左心房セグメンテーションのためのパラメータデカップリング戦略
Parameter Decoupling Strategy for Semi-supervised 3D Left Atrium Segmentation
一貫性トレーニングは、高度な半教師ありフレームワークであることが証明されており、入力のさまざまなビューに対して予測の不変性を適用することにより、医療画像セグメンテーションタスクで有望な結果を達成しました。ただし、モデルパラメータの反復更新により、モデルは結合状態に達する傾向があり、最終的にはラベルのないデータを活用する機能が失われます。この問題に対処するために、パラメータデカップリング戦略に基づく新しい半教師ありセグメンテーションモデルを提示して、さまざまなビューからの一貫した予測を促進します。具体的には、まず2分岐ネットワークを採用して、各画像の予測を同時に生成します。トレーニングプロセス中に、2次コサイン距離によって2つの予測分岐パラメーターを分離して、潜在空間に異なるビューを構築します。これに基づいて、特徴抽出器は、多様な特徴の下で分類器によって生成された確率マップの一貫性を促進するように制約されます。全体的なトレーニングプロセスでは、特徴抽出器と分類器のパラメーターが、一貫性の正則化操作とデカップリング操作によって交互に更新され、モデルの一般化パフォーマンスが徐々に向上します。私たちの方法は、心房セグメンテーションチャレンジデータセットの最先端の半教師あり方法に対して競争力のある結果を達成しており、私たちのフレームワークの有効性を示しています。コードはhttps://github.com/BX0903/PDCで入手できます。
Consistency training has proven to be an advanced semi-supervised framework and achieved promising results in medical image segmentation tasks through enforcing an invariance of the predictions over different views of the inputs. However, with the iterative updating of model parameters, the models would tend to reach a coupled state and eventually lose the ability to exploit unlabeled data. To address the issue, we present a novel semi-supervised segmentation model based on parameter decoupling strategy to encourage consistent predictions from diverse views. Specifically, we first adopt a two-branch network to simultaneously produce predictions for each image. During the training process, we decouple the two prediction branch parameters by quadratic cosine distance to construct different views in latent space. Based on this, the feature extractor is constrained to encourage the consistency of probability maps generated by classifiers under diversified features. In the overall training process, the parameters of feature extractor and classifiers are updated alternately by consistency regularization operation and decoupling operation to gradually improve the generalization performance of the model. Our method has achieved a competitive result over the state-of-the-art semi-supervised methods on the Atrial Segmentation Challenge dataset, demonstrating the effectiveness of our framework. Code is available at https://github.com/BX0903/PDC.
updated: Tue Nov 09 2021 04:59:18 GMT+0000 (UTC)
published: Mon Sep 20 2021 14:51:42 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アソシエイト