arXiv reaDer
MetaLR: 医用画像処理における転移学習のための学習率のメタチューニング
MetaLR: Meta-tuning of Learning Rates for Transfer Learning in Medical Imaging
医用画像解析において、転移学習は、ディープ ニューラル ネットワーク (DNN) が限られた医療データを適切に一般化するための強力な方法です。これまでの取り組みは、領域のギャップを埋めるために、肺超音波、胸部 X 線、肝臓 CT などの領域に関する事前トレーニング アルゴリズムの開発に焦点を当ててきました。ただし、モデルの微調整も、医療知識を対象タスクに適応させる上で重要な役割を果たすことがわかりました。一般的な微調整方法は、更新する転送可能なレイヤー (最後の数レイヤーなど) を手動で選択することであり、労力がかかります。この研究では、MetaLR という名前のメタ学習ベースの LR チューナーを提案し、ドメイン間の転送可能性に基づいてさまざまな層を下流のタスクに自動的に共同適応させます。 MetaLR は、さまざまなレイヤーに適切な LR をオンラインで学習し、移行性の高いレイヤーが医療表現能力を忘れることを防ぎ、移行性の低いレイヤーが新しいドメインに積極的に適応するように促します。さまざまな医療アプリケーションに関する広範な実験により、MetaLR が以前の最先端 (SOTA) 微調整戦略よりも優れたパフォーマンスを発揮することが示されています。コードが公開されています。
In medical image analysis, transfer learning is a powerful method for deep neural networks (DNNs) to generalize well on limited medical data. Prior efforts have focused on developing pre-training algorithms on domains such as lung ultrasound, chest X-ray, and liver CT to bridge domain gaps. However, we find that model fine-tuning also plays a crucial role in adapting medical knowledge to target tasks. The common fine-tuning method is manually picking transferable layers (e.g., the last few layers) to update, which is labor-expensive. In this work, we propose a meta-learning-based LR tuner, named MetaLR, to make different layers automatically co-adapt to downstream tasks based on their transferabilities across domains. MetaLR learns appropriate LRs for different layers in an online manner, preventing highly transferable layers from forgetting their medical representation abilities and driving less transferable layers to adapt actively to new domains. Extensive experiments on various medical applications show that MetaLR outperforms previous state-of-the-art (SOTA) fine-tuning strategies. Codes are released.
updated: Mon May 29 2023 10:35:58 GMT+0000 (UTC)
published: Fri Jun 03 2022 06:31:11 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アソシエイト