arXiv reaDer
医療画像セグメンテーションのためのオンザフライテスト時間適応
On-the-Fly Test-time Adaptation for Medical Image Segmentation
医用画像の深層学習ベースのソリューションにおける主要な問題の1つは、モデルがトレーニングされたものとは異なるデータ分布でテストされた場合のパフォーマンスの低下です。テスト時にソースモデルをターゲットデータ分布に適合させることは、データシフト問題の効率的な解決策です。以前の方法は、エントロピーの最小化や正則化などの手法を使用してモデルをターゲット分布に適合させることでこれを解決します。これらの方法では、完全なテストデータの分散で教師なし損失を使用したバックプロパゲーションによってモデルが更新されます。実際の臨床設定では、モデルを新しいテストイメージにオンザフライで適合させ、プライバシーの懸念と展開時のコンピューティングリソースの不足により、推論中のモデルの更新を回避する方が理にかなっています。この目的のために、新しい設定を提案します-ゼロショットでエピソード的なオンザフライ適応(つまり、モデルは一度に1つの画像に適応され、テスト時間中にバックプロパゲーションも実行されません) )。これを実現するために、Adaptive UNetと呼ばれる新しいフレームワークを提案します。このフレームワークでは、各畳み込みブロックに、ドメインコードに関して機能を適応させるための適応バッチ正規化レイヤーが装備されています。ドメインコードは、医用画像の大規模なコーパスでトレーニングされた事前トレーニング済みのエンコーダを使用して生成されます。テスト時に、モデルは新しいテストイメージのみを取り込み、ドメインコードを生成して、テストデータに従ってソースモデルの機能を適応させます。 2Dと3Dの両方のデータ分散シフトでパフォーマンスを検証し、以前のテスト時間適応方法と比較してパフォーマンスが向上します。コードはhttps://github.com/jeya-maria-jose/On-The-Fly-Adaptationで入手できます
One major problem in deep learning-based solutions for medical imaging is the drop in performance when a model is tested on a data distribution different from the one that it is trained on. Adapting the source model to target data distribution at test-time is an efficient solution for the data-shift problem. Previous methods solve this by adapting the model to target distribution by using techniques like entropy minimization or regularization. In these methods, the models are still updated by back-propagation using an unsupervised loss on complete test data distribution. In real-world clinical settings, it makes more sense to adapt a model to a new test image on-the-fly and avoid model update during inference due to privacy concerns and lack of computing resource at deployment. To this end, we propose a new setting - On-the-Fly Adaptation which is zero-shot and episodic (i.e., the model is adapted to a single image at a time and also does not perform any back-propagation during test-time). To achieve this, we propose a new framework called Adaptive UNet where each convolutional block is equipped with an adaptive batch normalization layer to adapt the features with respect to a domain code. The domain code is generated using a pre-trained encoder trained on a large corpus of medical images. During test-time, the model takes in just the new test image and generates a domain code to adapt the features of source model according to the test data. We validate the performance on both 2D and 3D data distribution shifts where we get a better performance compared to previous test-time adaptation methods. Code is available at https://github.com/jeya-maria-jose/On-The-Fly-Adaptation
updated: Thu Mar 10 2022 18:51:29 GMT+0000 (UTC)
published: Thu Mar 10 2022 18:51:29 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アソシエイト