arXiv reaDer
医用画像の分布外の一般化を改善するためにウィンドウ統計を使用する単純な正規化手法
A simple normalization technique using window statistics to improve the out-of-distribution generalization on medical images
データの不足とデータの不均一性が医用画像に蔓延しているため、以前の正規化方法を使用して十分にトレーニングされた畳み込みニューラルネットワーク(CNN)は、新しいサイトに展開するとパフォーマンスが低下する可能性があります。ただし、実際の臨床アプリケーションの信頼できるモデルは、分布内(IND)データと分布外(OOD)データ(新しいサイトデータなど)の両方で適切に一般化できる必要があります。この研究では、ウィンドウ正規化(WIN)と呼ばれる新しい正規化手法を提示して、異種の医用画像のモデルの一般化を改善します。これは、既存の正規化手法のシンプルで効果的な代替手段です。具体的には、WINは、フィーチャのウィンドウで計算されたローカル統計を使用して正規化統計を混乱させます。この機能レベルの拡張手法は、モデルを適切に正規化し、OODの一般化を大幅に改善します。その利点を利用して、分類タスクのためのWIN-WINと呼ばれる新しい自己蒸留法を提案します。 WIN-WINは、2回のフォワードパスと整合性制約を使用して簡単に実装できます。これは、既存のメソッドの単純な拡張になります。さまざまなタスク(6つのタスク)とデータセット(24のデータセット)に関する広範な実験結果は、私たちの方法の一般性と有効性を示しています。
Since data scarcity and data heterogeneity are prevailing for medical images, well-trained Convolutional Neural Networks (CNNs) using previous normalization methods may perform poorly when deployed to a new site. However, a reliable model for real-world clinical applications should be able to generalize well both on in-distribution (IND) and out-of-distribution (OOD) data (e.g., the new site data). In this study, we present a novel normalization technique called window normalization (WIN) to improve the model generalization on heterogeneous medical images, which is a simple yet effective alternative to existing normalization methods. Specifically, WIN perturbs the normalizing statistics with the local statistics computed on the window of features. This feature-level augmentation technique regularizes the models well and improves their OOD generalization significantly. Taking its advantage, we propose a novel self-distillation method called WIN-WIN for classification tasks. WIN-WIN is easily implemented with twice forward passes and a consistency constraint, which can be a simple extension for existing methods. Extensive experimental results on various tasks (6 tasks) and datasets (24 datasets) demonstrate the generality and effectiveness of our methods.
updated: Thu Jul 14 2022 03:36:21 GMT+0000 (UTC)
published: Thu Jul 07 2022 15:14:37 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アソシエイト