arXiv reaDer
MRIスキャンの半教師付きセグメンテーションのための二重不確実性ガイダンスを備えたハイブリッドデュアル平均教師ネットワーク
Hybrid Dual Mean-Teacher Network With Double-Uncertainty Guidance for Semi-Supervised Segmentation of MRI Scans
半教師あり学習は、医療画像のセグメンテーションにおいて大きな進歩を遂げました。ただし、既存の方法は主に 1 次元 (2D/3D) から取得した情報を利用するため、複数のオブジェクトを使用した磁気共鳴画像法 (MRI) スキャンや高度な異方性解像度など、困難なデータに対して最適なパフォーマンスが得られません。この問題に対処するために、非常に効果的な半教師ありセグメンテーションを実現するために、ハイブリッド、半教師あり、およびマルチタスク学習を備えたハイブリッド デュアル平均教師 (HD-Teacher) モデルを提示します。 HD-Teacher は、2D および 3D 平均教師ネットワークを使用して、両方の次元でキャプチャされたハイブリッド情報からセグメンテーション ラベルと符号付き距離フィールドを生成します。このハイブリッド学習メカニズムにより、HD-Teacher は、2D、3D、または両方の次元から抽出された機能を利用して、「両方の長所」を組み合わせて、適切と思われる出力を生成できます。 2D および 3D の教師モデルからの出力も、個々の不確実性スコアに基づいて動的に結合され、ハイブリッドの不確実性が推定される単一のハイブリッド予測になります。次に、ハイブリッド正則化モジュールを提案して、両方の学生モデルが不確実性加重ハイブリッド予測に近い結果を生成するようにします。ハイブリッドの不確実性は、ハイブリッド予測における信頼できない知識を抑制し、ネットワーク パフォーマンスをさらに改善するために役立つ情報のみを残します。 3 つの MRI データセットに対して行われたバイナリおよびマルチクラス セグメンテーションの広範な実験は、提案されたフレームワークの有効性を示しています。コードは https://github.com/ThisGame42/Hybrid-Teacher で入手できます。
Semi-supervised learning has made significant progress in medical image segmentation. However, existing methods primarily utilize information acquired from a single dimensionality (2D/3D), resulting in sub-optimal performance on challenging data, such as magnetic resonance imaging (MRI) scans with multiple objects and highly anisotropic resolution. To address this issue, we present a Hybrid Dual Mean-Teacher (HD-Teacher) model with hybrid, semi-supervised, and multi-task learning to achieve highly effective semi-supervised segmentation. HD-Teacher employs a 2D and a 3D mean-teacher network to produce segmentation labels and signed distance fields from the hybrid information captured in both dimensionalities. This hybrid learning mechanism allows HD-Teacher to combine the `best of both worlds', utilizing features extracted from either 2D, 3D, or both dimensions to produce outputs as it sees fit. Outputs from 2D and 3D teacher models are also dynamically combined, based on their individual uncertainty scores, into a single hybrid prediction, where the hybrid uncertainty is estimated. We then propose a hybrid regularization module to encourage both student models to produce results close to the uncertainty-weighted hybrid prediction. The hybrid uncertainty suppresses unreliable knowledge in the hybrid prediction, leaving only useful information to improve network performance further. Extensive experiments of binary and multi-class segmentation conducted on three MRI datasets demonstrate the effectiveness of the proposed framework. Code is available at https://github.com/ThisGame42/Hybrid-Teacher.
updated: Thu Mar 09 2023 09:16:39 GMT+0000 (UTC)
published: Thu Mar 09 2023 09:16:39 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アソシエイト