arXiv reaDer
AC-Norm: アフィン協調正規化による医用画像解析の効果的なチューニング
AC-Norm: Effective Tuning for Medical Image Analysis via Affine Collaborative Normalization
自己教師あり学習 (SSL) への最新の傾向に後押しされ、限られたアノテーションで臨床アプリケーションのパフォーマンスを向上させるために、「事前トレーニング、その後の微調整」のパラダイムが広範囲に検討されています。モデルの微調整に関するこれまでの文献は、主に正則化項と特定のポリシー モデルに焦点を当ててきましたが、ソース モデルとターゲット モデル間のチャネルの不整合については十分な注目が集まっていませんでした。この研究では、バッチ正規化 (BN) 層のダイナミクスを再検討し、BN のトレーニング可能なアフィン パラメーターがドメイン情報の敏感な指標として機能することを観察しました。したがって、微調整のためにアフィン協調正規化 (AC-Norm) が提案されています。これは、追加のパラメーターを追加せずに、クロスドメインのチャネルごとの相関に従ってターゲット モデル内のチャネルを動的に再調整します。シングルステップのバックプロパゲーションに基づいて、AC-Norm を利用して、事前トレーニングされたモデルの伝達​​性を測定することもできます。糖尿病性網膜症のグレード分類、網膜血管セグメンテーション、CT 肺結節セグメンテーション/分類、CT 肝腫瘍セグメンテーション、 MRI 心臓セグメンテーション タスク。広範な実験により、最先端の手法ではメリットが得られない大幅なドメインシフト下でも、AC-Norm がバニラ微調整よりも最大 4% の改善で優れたパフォーマンスを示すことが実証されました。また、高速転送性推定における AC-Norm の機能も証明します。私たちのコードは https://github.com/EndoluminalSurgicalVision-IMR/ACNorm で入手できます。
Driven by the latest trend towards self-supervised learning (SSL), the paradigm of "pretraining-then-finetuning" has been extensively explored to enhance the performance of clinical applications with limited annotations. Previous literature on model finetuning has mainly focused on regularization terms and specific policy models, while the misalignment of channels between source and target models has not received sufficient attention. In this work, we revisited the dynamics of batch normalization (BN) layers and observed that the trainable affine parameters of BN serve as sensitive indicators of domain information. Therefore, Affine Collaborative Normalization (AC-Norm) is proposed for finetuning, which dynamically recalibrates the channels in the target model according to the cross-domain channel-wise correlations without adding extra parameters. Based on a single-step backpropagation, AC-Norm can also be utilized to measure the transferability of pretrained models. We evaluated AC-Norm against the vanilla finetuning and state-of-the-art fine-tuning methods on transferring diverse pretrained models to the diabetic retinopathy grade classification, retinal vessel segmentation, CT lung nodule segmentation/classification, CT liver-tumor segmentation and MRI cardiac segmentation tasks. Extensive experiments demonstrate that AC-Norm unanimously outperforms the vanilla finetuning by up to 4% improvement, even under significant domain shifts where the state-of-the-art methods bring no gains. We also prove the capability of AC-Norm in fast transferability estimation. Our code is available at https://github.com/EndoluminalSurgicalVision-IMR/ACNorm.
updated: Fri Jul 28 2023 03:27:25 GMT+0000 (UTC)
published: Fri Jul 28 2023 03:27:25 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アソシエイト