arXiv reaDer
アトラスに注目してください: 堅牢な 3D 医用画像セグメンテーションのためのアトラスガイドによるテスト時間適応方法
Pay Attention to the Atlas: Atlas-Guided Test-Time Adaptation Method for Robust 3D Medical Image Segmentation
畳み込みニューラル ネットワーク (CNN) は、トレーニング (ソース) データ分布とは異なるターゲット データでテストすると、パフォーマンスが低下することがよくあります。特に医療画像アプリケーションでは、さまざまな臨床現場やスキャナーにわたる画像プロトコルの違いにより画像の外観が異なります。ただし、教師なしドメイン適応のためにソース トレーニング データに再アクセスしたり、モデル微調整のために追加のテスト データにラベルを付けたりすることは、それぞれプライバシーの問題とラベル付けのコストが高いため、難しい場合があります。この問題を解決するために、我々は、AdaAtlas と呼ばれる、堅牢な 3D 医用画像セグメンテーションのための新しいアトラスガイドによるテスト時間適応 (TTA) 方法を提案します。 AdaAtlas は、ラベルのないテスト サンプルを 1 つだけ入力として受け取り、アトラスベースの損失を最小限に抑えることでセグメンテーション ネットワークを適応させます。具体的には、ネットワークは、登録後の予測がアトラス空間内の学習されたアトラスと一致するように適応されており、テスト時の解剖学的セグメンテーション エラーの削減に役立ちます。さらに、セグメンテーション ネットワーク内のバッチ正規化ブロックのみへの適応を制限するほとんどの既存の TTA 手法とは異なり、テスト時の適応性を向上させるためにチャネルと空間アテンション ブロックの使用をさらに活用します。異なるサイトからの複数のデータセットに対する広範な実験により、アテンション ブロックを適応させた AdaAtlas (AdaAtlas-tention) が優れたパフォーマンス向上を達成し、他の競合 TTA 手法を大きく上回るパフォーマンスを示すことが示されました。
Convolutional neural networks (CNNs) often suffer from poor performance when tested on target data that differs from the training (source) data distribution, particularly in medical imaging applications where variations in imaging protocols across different clinical sites and scanners lead to different imaging appearances. However, re-accessing source training data for unsupervised domain adaptation or labeling additional test data for model fine-tuning can be difficult due to privacy issues and high labeling costs, respectively. To solve this problem, we propose a novel atlas-guided test-time adaptation (TTA) method for robust 3D medical image segmentation, called AdaAtlas. AdaAtlas only takes one single unlabeled test sample as input and adapts the segmentation network by minimizing an atlas-based loss. Specifically, the network is adapted so that its prediction after registration is aligned with the learned atlas in the atlas space, which helps to reduce anatomical segmentation errors at test time. In addition, different from most existing TTA methods which restrict the adaptation to batch normalization blocks in the segmentation network only, we further exploit the use of channel and spatial attention blocks for improved adaptability at test time. Extensive experiments on multiple datasets from different sites show that AdaAtlas with attention blocks adapted (AdaAtlas-Attention) achieves superior performance improvements, greatly outperforming other competitive TTA methods.
updated: Tue Feb 27 2024 13:46:11 GMT+0000 (UTC)
published: Sun Jul 02 2023 22:08:24 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アソシエイト