arXiv reaDer
医療画像セグメンテーションにおける教師なしドメイン適応のための最適な潜在ベクトルアラインメント
Optimal Latent Vector Alignment for Unsupervised Domain Adaptation in Medical Image Segmentation
このホワイトペーパーでは、セグメンテーションのドメインシフトの問題について説明します。解決策として、変分オートエンコーダー(VAE)と最適輸送(OT)理論に基づく新規で軽量な教師なしドメイン適応法であるOLVAを提案します。 VAEのおかげで、私たちのモデルは、正規分布に従う共有クロスドメイン潜在空間を学習します。これにより、ドメインシフトが減少します。有効なセグメンテーションを保証するために、共有潜在空間は、強度の変化ではなく形状をモデル化するように設計されています。さらに、潜在空間内の2つのドメイン間の残りの不一致を一致させて調整するために、OT損失に依存しています。ソースドメインが注釈付きの3DMR画像と3DCTのラベルなしのターゲットドメインで構成される、パブリックマルチモダリティ全心臓セグメンテーション(MM-WHS)データセットで、複数の心臓構造のセグメンテーションに対するOLVAの有効性を示します。私たちの結果は、同時生成トレーニングアプローチに比べて12.5%のダイススコアの追加マージンで顕著な改善を示しています。
This paper addresses the domain shift problem for segmentation. As a solution, we propose OLVA, a novel and lightweight unsupervised domain adaptation method based on a Variational Auto-Encoder (VAE) and Optimal Transport (OT) theory. Thanks to the VAE, our model learns a shared cross-domain latent space that follows a normal distribution, which reduces the domain shift. To guarantee valid segmentations, our shared latent space is designed to model the shape rather than the intensity variations. We further rely on an OT loss to match and align the remaining discrepancy between the two domains in the latent space. We demonstrate OLVA's effectiveness for the segmentation of multiple cardiac structures on the public Multi-Modality Whole Heart Segmentation (MM-WHS) dataset, where the source domain consists of annotated 3D MR images and the unlabelled target domain of 3D CTs. Our results show remarkable improvements with an additional margin of 12.5% dice score over concurrent generative training approaches.
updated: Tue Jun 15 2021 14:41:09 GMT+0000 (UTC)
published: Tue Jun 15 2021 14:41:09 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アソシエイト