arXiv reaDer
DOMINO++: 深層学習の一般化のためのドメイン認識損失正則化
DOMINO++: Domain-aware Loss Regularization for Deep Learning Generalizability
Out-of-distribution (OOD) 一般化は、最新のディープラーニング (DL) にとって深刻な課題を引き起こします。 OOD データは、モデルのトレーニング データとは大きく異なるテスト データで構成されます。 DL モデルはドメイン内テスト データでは良好にパフォーマンスしますが、OOD データでは苦戦する可能性があります。この矛盾を克服することは、DL を確実に展開するために不可欠です。モデルを適切にキャリブレーションすると、モデルの特徴とクラス出力の間で行われる偽の接続の数が減少します。したがって、調整された DL は、それぞれのクラスを真に示す特徴のみを学習することで、OOD の一般化を向上させることができます。以前の研究では、DL キャリブレーションを改善するためにドメイン認識モデル キャリブレーション (DOMINO) が提案されましたが、OOD データに対するモデルの一般化性に関する設計が不足していました。この研究では、OOD の一般化性に焦点を当てたデュアル ガイダンスで動的ドメイン認識損失正則化である DOMINO++ を提案します。 DOMINO++ は、専門家に基づいた知識とデータに基づいた知識を正規化に統合します。固定のスケーリングと正則化レートを課す DOMINO とは異なり、DOMINO++ は動的なスケーリング係数と適応的な正則化レートを設計します。包括的な評価では、DOMINO++ を DOMINO および OOD データの磁気共鳴画像 (MRI) からの頭部組織セグメンテーションのベースライン モデルと比較します。 OOD データは、ノイズを含む合成データセットと回転データセット、および別のサイトの別の MRI スキャナーを使用した実際のデータで構成されます。 DOMINO++ の優れたパフォーマンスは、実際の臨床データに対する DL の信頼できる展開を向上させる可能性を示しています。
Out-of-distribution (OOD) generalization poses a serious challenge for modern deep learning (DL). OOD data consists of test data that is significantly different from the model's training data. DL models that perform well on in-domain test data could struggle on OOD data. Overcoming this discrepancy is essential to the reliable deployment of DL. Proper model calibration decreases the number of spurious connections that are made between model features and class outputs. Hence, calibrated DL can improve OOD generalization by only learning features that are truly indicative of the respective classes. Previous work proposed domain-aware model calibration (DOMINO) to improve DL calibration, but it lacks designs for model generalizability to OOD data. In this work, we propose DOMINO++, a dual-guidance and dynamic domain-aware loss regularization focused on OOD generalizability. DOMINO++ integrates expert-guided and data-guided knowledge in its regularization. Unlike DOMINO which imposed a fixed scaling and regularization rate, DOMINO++ designs a dynamic scaling factor and an adaptive regularization rate. Comprehensive evaluations compare DOMINO++ with DOMINO and the baseline model for head tissue segmentation from magnetic resonance images (MRIs) on OOD data. The OOD data consists of synthetic noisy and rotated datasets, as well as real data using a different MRI scanner from a separate site. DOMINO++'s superior performance demonstrates its potential to improve the trustworthy deployment of DL on real clinical data.
updated: Mon Aug 21 2023 03:58:04 GMT+0000 (UTC)
published: Mon Aug 21 2023 03:58:04 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アソシエイト