推論中のドメイン シフトの問題を解決することは、ほとんどのディープ ラーニング ベースのソリューションに問題があるため、医用画像処理において不可欠です。実際には、教師なしドメイン適応 (UDA) を実行することでドメイン シフトに対処します。UDA では、ラベル付けされたソース データを活用して、モデルをラベル付けされていないターゲット ドメインに適応させます。医療シナリオでは、データにはプライバシーに関する大きな懸念が伴い、標準の UDA 手法を適用することが難しくなります。したがって、より近い臨床設定はソースフリー UDA (SFUDA) です。ここでは、ソーストレーニングされたモデルにアクセスできますが、適応中にソースデータにはアクセスできません。既存の SFUDA メソッドは、ドメイン シフトに対処するために、疑似ラベル ベースの自己トレーニング手法に依存しています。ただし、これらの疑似ラベルはドメイン シフトによりエントロピーが高くなることが多く、ノイズの多い疑似ラベルを使用してソース モデルを適応させると、最適なパフォーマンスが得られません。この制限を克服するために、ターゲット固有の適応とそれに続くタスク固有の適応で構成される SFUDA の体系的な 2 段階のアプローチを提案します。ターゲット固有の適応では、提案されたアンサンブルエントロピー最小化損失と選択的投票戦略を使用して高エントロピー領域を最小化することにより、疑似ラベル生成を強化します。タスク固有の適応では、学生と教師のフレームワークを使用して強化された疑似ラベルを活用し、ターゲット ドメインのセグメンテーションを効果的に学習します。 2D 眼底データセットと 3D MRI ボリュームで提案された方法を 7 つの異なるドメイン シフトにわたって評価し、医用画像セグメンテーションのための既存の UDA および SFUDA 方法よりも優れたパフォーマンスを発揮します。コードは https://github.com/Vibashan/tt-sfuda で入手できます。
Solving the domain shift problem during inference is essential in medical imaging, as most deep-learning based solutions suffer from it. In practice, domain shifts are tackled by performing Unsupervised Domain Adaptation (UDA), where a model is adapted to an unlabelled target domain by leveraging the labelled source data. In medical scenarios, the data comes with huge privacy concerns making it difficult to apply standard UDA techniques. Hence, a closer clinical setting is Source-Free UDA (SFUDA), where we have access to source-trained model but not the source data during adaptation. Existing SFUDA methods rely on pseudo-label based self-training techniques to address the domain shift. However, these pseudo-labels often have high entropy due to domain shift and adapting the source model with noisy pseudo-labels leads to sub-optimal performance. To overcome this limitation, we propose a systematic two-stage approach for SFUDA comprising of target-specific adaptation followed by task-specific adaptation. In target-specific adaptation, we enhance the pseudo-label generation by minimizing high entropy regions using the proposed ensemble entropy minimization loss and a selective voting strategy. In task-specific adaptation, we exploit the enhanced pseudo-labels using a student-teacher framework to effectively learn segmentation on the target domain. We evaluate our proposed method on 2D fundus datasets and 3D MRI volumes across 7 different domain shifts where we perform better than existing UDA and SFUDA methods for medical image segmentation. Code is available at https://github.com/Vibashan/tt-sfuda.