arXiv reaDer
カテゴリレベルの正則化による教師なしドメイン適応眼底画像セグメンテーション
Unsupervised Domain Adaptive Fundus Image Segmentation with Category-level Regularization
敵対的学習に基づく既存の教師なしドメイン適応法は、いくつかの医用画像処理タスクで優れたパフォーマンスを達成しています。ただし、これらの方法は、グローバルな分布の適応にのみ焦点を当てており、カテゴリレベルでの分布の制約を無視します。これにより、適応のパフォーマンスが最適化されなくなります。このホワイトペーパーでは、カテゴリの分布を3つの観点から正則化する、カテゴリレベルの正則化に基づく教師なしドメイン適応フレームワークを紹介します。具体的には、ドメイン間カテゴリの正則化のために、適応型プロトタイプアラインメントモジュールが提案され、ソースドメインとターゲットドメインで同じカテゴリの機能プロトタイプをアラインメントします。さらに、ドメイン内カテゴリの正則化については、ソースドメインとターゲットドメインにそれぞれ正則化手法を調整しました。ソースドメインでは、クラス内のコンパクト性とクラス間の分離可能性を適用することにより、従来の教師あり損失を補完するものとして、より識別的な特徴表現を学習するために、プロトタイプに基づく識別損失が提案されます。ターゲットドメインでは、拡張された/拡張されていないターゲット画像の一貫した予測をモデルに強制するために、拡張された整合性カテゴリの正則化損失が提案されます。これにより、意味的に類似した領域に同じラベルが付けられます。 2つの公的眼底データセットに関する広範な実験は、提案されたアプローチが他の最先端の比較アルゴリズムを大幅に上回っていることを示しています。
Existing unsupervised domain adaptation methods based on adversarial learning have achieved good performance in several medical imaging tasks. However, these methods focus only on global distribution adaptation and ignore distribution constraints at the category level, which would lead to sub-optimal adaptation performance. This paper presents an unsupervised domain adaptation framework based on category-level regularization that regularizes the category distribution from three perspectives. Specifically, for inter-domain category regularization, an adaptive prototype alignment module is proposed to align feature prototypes of the same category in the source and target domains. In addition, for intra-domain category regularization, we tailored a regularization technique for the source and target domains, respectively. In the source domain, a prototype-guided discriminative loss is proposed to learn more discriminative feature representations by enforcing intra-class compactness and inter-class separability, and as a complement to traditional supervised loss. In the target domain, an augmented consistency category regularization loss is proposed to force the model to produce consistent predictions for augmented/unaugmented target images, which encourages semantically similar regions to be given the same label. Extensive experiments on two publicly fundus datasets show that the proposed approach significantly outperforms other state-of-the-art comparison algorithms.
updated: Fri Jul 08 2022 04:34:39 GMT+0000 (UTC)
published: Fri Jul 08 2022 04:34: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アソシエイト