arXiv reaDer
医療領域拡大のための壊滅的な忘却への対処
Addressing catastrophic forgetting for medical domain expansion
モデルの脆弱性は、実際の医療環境でディープラーニングモデルを展開する際の重要な懸念事項です。ある機関で高いパフォーマンスを発揮するモデルは、他の機関でテストするとパフォーマンスが大幅に低下する可能性があります。複数の機関からのデータセットをプールして再トレーニングすることで簡単な解決策が得られる場合がありますが、それは実行不可能な場合が多く、患者のプライバシーを侵害する可能性があります。別のアプローチは、元の機関でトレーニングした後、後続の機関でモデルを微調整することです。特に、このアプローチは、元の機関でのモデルのパフォーマンスを低下させます。これは、壊滅的な忘却として知られる現象です。この論文では、2つのシナリオでバッチ正規化統計の変調と組み合わせた弾性重量統合に基づく壊滅的な忘却に対処するアプローチを開発します。1つは、あるイメージングシステムのデータから別のイメージングシステムのデータにドメインを拡張するため、2つ目は大規模な複数機関のデータセットから別の単一機関のデータセットにドメインを拡張します。私たちのアプローチが他のいくつかの最先端のアプローチよりも優れていることを示し、バッチ正規化変調の有効性の理論的正当化を提供します。この研究の結果は、一般に、ドメイン拡張を必要とするあらゆる臨床深層学習モデルの展開に適用できます。
Model brittleness is a key concern when deploying deep learning models in real-world medical settings. A model that has high performance at one institution may suffer a significant decline in performance when tested at other institutions. While pooling datasets from multiple institutions and retraining may provide a straightforward solution, it is often infeasible and may compromise patient privacy. An alternative approach is to fine-tune the model on subsequent institutions after training on the original institution. Notably, this approach degrades model performance at the original institution, a phenomenon known as catastrophic forgetting. In this paper, we develop an approach to address catastrophic forget-ting based on elastic weight consolidation combined with modulation of batch normalization statistics under two scenarios: first, for expanding the domain from one imaging system's data to another imaging system's, and second, for expanding the domain from a large multi-institutional dataset to another single institution dataset. We show that our approach outperforms several other state-of-the-art approaches and provide theoretical justification for the efficacy of batch normalization modulation. The results of this study are generally applicable to the deployment of any clinical deep learning model which requires domain expansion.
updated: Wed Mar 24 2021 22:33:38 GMT+0000 (UTC)
published: Wed Mar 24 2021 22:33:38 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アソシエイト