arXiv reaDer
教師なしドメイン適応セグメンテーションのための分布整合拡散およびプロトタイプ誘導ネットワーク
Distribution Aligned Diffusion and Prototype-guided network for Unsupervised Domain Adaptive Segmentation
拡散確率モデル (DPM) は、コンピューター ビジョンの分野で非常に効果的な生成モデルとして登場しました。その中間潜在ベクトルは豊富なセマンティック情報を提供するため、セグメンテーションや検出などのさまざまなダウンストリーム タスクにとって魅力的なオプションとなっています。その可能性をさらに探求するために、私たちは一歩前進し、特に教師なしの適応条件下で、医用画像ドメインのより複雑なシナリオを検討しました。この目的のために、教師なしドメイン適応セグメンテーションのための拡散ベースおよびプロトタイプ誘導ネットワーク (DP-Net) を提案します。具体的には、DP-Net は次の 2 つの段階で構成されます。1) DPM によって生成された中間特徴間の差異を最小化するようにドメイン弁別器をトレーニングすることを含む、分布整列拡散 (DADiff)。これにより、ドメイン間の分布を整列させます。 2) プロトタイプ ガイド付き整合性学習 (PCL) は、フィーチャ セントロイドをプロトタイプとして利用し、プロトタイプ ガイド付き損失を適用して、セグメンターがソース ドメインとターゲット ドメインの両方から一貫したコンテンツを確実に学習するようにします。私たちのアプローチは、一連の実験を通じて眼底データセットで評価され、提案された方法のパフォーマンスが信頼性が高く、最先端の方法よりも優れていることが示されています。私たちの研究は、複雑な医用画像シナリオで DPM を使用するための有望な方向性を示しており、医用画像のさらなる研究に新たな可能性を切り開いています。
The Diffusion Probabilistic Model (DPM) has emerged as a highly effective generative model in the field of computer vision. Its intermediate latent vectors offer rich semantic information, making it an attractive option for various downstream tasks such as segmentation and detection. In order to explore its potential further, we have taken a step forward and considered a more complex scenario in the medical image domain, specifically, under an unsupervised adaptation condition. To this end, we propose a Diffusion-based and Prototype-guided network (DP-Net) for unsupervised domain adaptive segmentation. Concretely, our DP-Net consists of two stages: 1) Distribution Aligned Diffusion (DADiff), which involves training a domain discriminator to minimize the difference between the intermediate features generated by the DPM, thereby aligning the inter-domain distribution; and 2) Prototype-guided Consistency Learning (PCL), which utilizes feature centroids as prototypes and applies a prototype-guided loss to ensure that the segmentor learns consistent content from both source and target domains. Our approach is evaluated on fundus datasets through a series of experiments, which demonstrate that the performance of the proposed method is reliable and outperforms state-of-the-art methods. Our work presents a promising direction for using DPM in complex medical image scenarios, opening up new possibilities for further research in medical imaging.
updated: Sat Mar 25 2023 04:59:00 GMT+0000 (UTC)
published: Wed Mar 22 2023 05:03:14 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アソシエイト