arXiv reaDer
スペクトルデカップリングにより、医用画像で転送可能なニューラルネットワークのトレーニングが可能になります
Spectral decoupling allows training transferable neural networks in medical imaging
医用画像用の現在のニューラルネットワークの多くは、トレーニング中に見えないデータに一般化されていません。このような動作は、他の潜在的に有益な機能を無視しながら、学習しやすい、または統計的に優勢な機能をネットワークが過剰適合させることによって引き起こされる可能性があります。たとえば、2つの異なるスキャナーからの画像の鮮明度に見分けがつかない違いがあると、ネットワークのパフォーマンスが大幅に低下する可能性があります。臨床診療を目的としたすべての神経ネットワークは、画像機器、サンプル準備、および患者集団の違いによって引き起こされるデータの変動に対して堅牢である必要があります。これらの課題に対処するために、暗黙のバイアス軽減方法としてのスペクトルデカップリングの有用性を評価します。スペクトルデカップリングは、ネットワークの正規化されていない予測スコアをL2ペナルティで正規化するだけで、ニューラルネットワークがより多くの機能を学習するように促し、追加の計算コストをかけません。スペクトルデカップリングにより、強い疑似相関を持つデータセットでニューラルネットワークをトレーニングでき、データ分散シフトに対するネットワークの堅牢性が向上することを示します。調査結果を検証するために、スペクトルデカップリングの有無にかかわらずネットワークをトレーニングして、胸部X線写真で前立腺癌組織スライドとCOVID-19を検出します。スペクトルデカップリングでトレーニングされたネットワークは、外部データセットで最大9.5パーセント高いパフォーマンスを実現します。私たちの結果は、スペクトルデカップリングがニューラルネットワークに関連する一般化の問題に役立ち、組織学的画像の汚れの正規化など、計算コストの高い明示的なバイアス軽減方法を補完または置き換えることができることを示しています。臨床使用を目的としたニューラルネットワークでは、暗黙のバイアス軽減方法としてスペクトルデカップリングを使用することをお勧めします。
Many current neural networks for medical imaging generalise poorly to data unseen during training. Such behaviour can be caused by networks overfitting easy-to-learn, or statistically dominant, features while disregarding other potentially informative features. For example, indistinguishable differences in the sharpness of the images from two different scanners can degrade the performance of the network significantly. All neural networks intended for clinical practice need to be robust to variation in data caused by differences in imaging equipment, sample preparation and patient populations. To address these challenges, we evaluate the utility of spectral decoupling as an implicit bias mitigation method. Spectral decoupling encourages the neural network to learn more features by simply regularising the networks' unnormalised prediction scores with an L2 penalty, thus having no added computational costs. We show that spectral decoupling allows training neural networks on datasets with strong spurious correlations and increases networks' robustness for data distribution shifts. To validate our findings, we train networks with and without spectral decoupling to detect prostate cancer tissue slides and COVID-19 in chest radiographs. Networks trained with spectral decoupling achieve up to 9.5 percent point higher performance on external datasets. Our results show that spectral decoupling helps with generalisation issues associated with neural networks, and can be used to complement or replace computationally expensive explicit bias mitigation methods, such as stain normalization in histological images. We recommend using spectral decoupling as an implicit bias mitigation method in any neural network intended for clinical use.
updated: Fri Dec 17 2021 17:16:03 GMT+0000 (UTC)
published: Wed Mar 31 2021 15:47:01 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アソシエイト