arXiv reaDer
画像セグメンテーションのためのソースフリードメイン適応
Source-Free Domain Adaptation for Image Segmentation
ドメイン適応(DA)は、ラベル付けされたソースデータでトレーニングされたモデルを適応させて、異なるドメインからのラベル付けされていない、または弱くラベル付けされたターゲットデータで適切に機能する能力に高い関心を集めています。最も一般的なDA手法では、ソースドメインとターゲットドメインの両方の入力イメージに同時にアクセスする必要があります。ただし、実際には、プライバシーの懸念により、適応フェーズでのソース画像の可用性が妨げられることがよくあります。これは、医用画像で非常に頻繁に発生するDAシナリオであり、たとえば、ソース画像とターゲット画像が異なる臨床部位から取得される可能性があります。画像セグメンテーションのためのソースフリードメイン適応を紹介します。私たちの定式化は、ターゲットドメインデータに対して定義されたラベルフリーエントロピー損失を最小化することに基づいています。これは、セグメンテーション領域の前にドメイン不変でさらにガイドします。多くの事前情報は、解剖学的情報から導き出すことができます。ここでは、事前のクラス比が解剖学的知識から推定され、全体的な損失関数のカルバックライブラー(KL)発散の形で統合されています。さらに、ターゲット画像とそのラベル予測の間の相互情報量を最大化するための興味深いリンクを使用して、全体的な損失を動機付けます。脊椎、前立腺、心臓のセグメンテーションなど、さまざまなモダリティとアプリケーションを使用して、さまざまなドメイン適応シナリオで事前に認識されたエントロピー最小化の有効性を示します。私たちの方法は、ソース画像が私たちの適応段階で完全に存在しないため、はるかに少ない情報にアクセスできるにもかかわらず、いくつかの最先端の適応技術に匹敵する結果をもたらします。私たちの単純な適応戦略は、ソースフリーのDA設定には適用できない一般的な敵対的手法とは対照的に、1つのネットワークのみを使用します。私たちのフレームワークは、さまざまなセグメンテーションの問題ですぐに使用でき、コードは一般に公開されています:https://github.com/mathilde-b/SFDA
Domain adaptation (DA) has drawn high interest for its capacity to adapt a model trained on labeled source data to perform well on unlabeled or weakly labeled target data from a different domain. Most common DA techniques require concurrent access to the input images of both the source and target domains. However, in practice, privacy concerns often impede the availability of source images in the adaptation phase. This is a very frequent DA scenario in medical imaging, where, for instance, the source and target images could come from different clinical sites. We introduce a source-free domain adaptation for image segmentation. Our formulation is based on minimizing a label-free entropy loss defined over target-domain data, which we further guide with a domain-invariant prior on the segmentation regions. Many priors can be derived from anatomical information. Here, a class ratio prior is estimated from anatomical knowledge and integrated in the form of a Kullback Leibler (KL) divergence in our overall loss function. Furthermore, we motivate our overall loss with an interesting link to maximizing the mutual information between the target images and their label predictions. We show the effectiveness of our prior aware entropy minimization in a variety of domain-adaptation scenarios, with different modalities and applications, including spine, prostate, and cardiac segmentation. Our method yields comparable results to several state of the art adaptation techniques, despite having access to much less information, as the source images are entirely absent in our adaptation phase. Our straightforward adaptation strategy uses only one network, contrary to popular adversarial techniques, which are not applicable to a source-free DA setting. Our framework can be readily used in a breadth of segmentation problems, and our code is publicly available: https://github.com/mathilde-b/SFDA
updated: Fri Aug 06 2021 14:56:31 GMT+0000 (UTC)
published: Fri Aug 06 2021 14:56:31 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アソシエイト