arXiv reaDer
画像セグメンテーションのための形状モーメントによるテスト時間適応
Test-Time Adaptation with Shape Moments for Image Segmentation
教師あり学習は、分布シフトの下で一般化に失敗することがよく知られています。典型的な臨床設定では、ソースデータにアクセスできず、ターゲット分布は少数のサンプルで表されます。適応は、テスト時に少数または単一の被験者でのみ発生する可能性があります。セグメンテーションのためのテスト時間の単一被験者の適応を調査し、このタスクに取り組むための形状誘導エントロピー最小化の目的を提案します。単一のテスト対象の推論中に、バッチ正規化のスケールとバイアスパラメーターに関して損失が最小限に抑えられます。もっともらしいソリューションへの適応を導くためにさまざまな形状の事前情報を統合する可能性を示し、心臓セグメンテーションのMRIからCTへの適応と前立腺セグメンテーションのクロスサイト適応という2つの挑戦的なシナリオで私たちの方法を検証します。私たちのアプローチは、既存のテスト時間適応方法よりも大幅に優れたパフォーマンスを示します。さらに驚くべきことに、適応中に追加のターゲットデータのトレーニングを行わないものの、最先端のドメイン適応方法よりも優れています。私たちの結果は、セグメンテーション適応におけるターゲットデータのトレーニングの有用性に疑問を投げかけ、テスト時間の推論に対する形状の事前の実質的な効果を示しています。私たちのフレームワークは、さまざまな事前情報を統合したり、任意のセグメンテーションネットワークを適応させたりするために簡単に使用でき、コードを利用できます。
Supervised learning is well-known to fail at generalization under distribution shifts. In typical clinical settings, the source data is inaccessible and the target distribution is represented with a handful of samples: adaptation can only happen at test time on a few or even a single subject(s). We investigate test-time single-subject adaptation for segmentation, and propose a Shape-guided Entropy Minimization objective for tackling this task. During inference for a single testing subject, our loss is minimized with respect to the batch normalization's scale and bias parameters. We show the potential of integrating various shape priors to guide adaptation to plausible solutions, and validate our method in two challenging scenarios: MRI-to-CT adaptation of cardiac segmentation and cross-site adaptation of prostate segmentation. Our approach exhibits substantially better performances than the existing test-time adaptation methods. Even more surprisingly, it fares better than state-of-the-art domain adaptation methods, although it forgoes training on additional target data during adaptation. Our results question the usefulness of training on target data in segmentation adaptation, and points to the substantial effect of shape priors on test-time inference. Our framework can be readily used for integrating various priors and for adapting any segmentation network, and our code is available.
updated: Mon May 16 2022 20:47:13 GMT+0000 (UTC)
published: Mon May 16 2022 20:47:13 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アソシエイト