arXiv reaDer
自然および医療用X線胸部画像のドメイン内およびドメイン間フルショットおよび少数ショット転送学習に対する事前トレーニングスケールの影響
Effect of Pre-Training Scale on Intra- and Inter-Domain Full and Few-Shot Transfer Learning for Natural and Medical X-Ray Chest Images
転移学習は、事前にトレーニングされたモデルを活用して、さまざまなダウンストリームタスクとデータセットでより効率的なフォローアップトレーニングを行い、小さなデータでもトレーニングを成功させることを目的としています。最近、事前トレーニングでモデル、データ、および計算予算のスケールを増やすと、転移学習とモデルの一般化に大きな改善が見られました。ドメイン内およびドメイン間のフルショット転送と数ショット転送の両方でスケールの効果を比較するために、この研究では、初めて大規模な公開医療用X線胸部イメージングデータセットを組み合わせて、ImageNet-1kに匹敵するデータセットスケールに到達します。次に、事前トレーニングを実施し、ネットワークサイズとソースデータのスケールとドメインを変えながら、さまざまな自然または医療ターゲットに転送します。これは、大規模な自然(ImageNet-1k / 21k)または大規模な医療用胸部X線データセットです。ドメイン内の自然-自然および医療-医療転送の事前トレーニングスケールが大きくなるため、大幅な改善が見られます。ドメイン間自然医療転送の場合、フルショットレジームの大きなX線ターゲットでの事前トレーニングスケールが大きいために改善が見られますが、小さいターゲットおよび少数ショットレジームでは改善は見られません。驚くべきことに、非常に大きな自然のImageNet-21kで事前トレーニングされた大規模なネットワークは、大規模なX線ターゲットへの転送を実行するときに利用可能な最大の医療用X線データで事前トレーニングされたネットワークと同等またはそれ以上です。ドメイン間転送の高品質モデルは、モデルと一般的な天然ソースデータの規模を大幅に拡大し、事前トレーニングでの大規模なドメイン固有の医療ソースデータの必要性を排除することによっても得られると結論付けています。コードは次の場所で入手できます:https://github.com/SLAMPAI/large-scale-pretraining-transfer}
Transfer learning aims to exploit pre-trained models for more efficient follow-up training on wide range of downstream tasks and datasets, enabling successful training also on small data. Recently, strong improvement was shown for transfer learning and model generalization when increasing model, data and compute budget scale in the pre-training. To compare effect of scale both in intra- and inter-domain full and few-shot transfer, in this study we combine for the first time large openly available medical X-Ray chest imaging datasets to reach a dataset scale comparable to ImageNet-1k. We then conduct pre-training and transfer to different natural or medical targets while varying network size and source data scale and domain, being either large natural (ImageNet-1k/21k) or large medical chest X-Ray datasets. We observe strong improvement due to larger pre-training scale for intra-domain natural-natural and medical-medical transfer. For inter-domain natural-medical transfer, we find improvements due to larger pre-training scale on larger X-Ray targets in full shot regime, while for smaller targets and for few-shot regime the improvement is not visible. Remarkably, large networks pre-trained on very large natural ImageNet-21k are as good or better than networks pre-trained on largest available medical X-Ray data when performing transfer to large X-Ray targets. We conclude that high quality models for inter-domain transfer can be also obtained by substantially increasing scale of model and generic natural source data, removing necessity for large domain-specific medical source data in the pre-training. Code is available at: https://github.com/SLAMPAI/large-scale-pretraining-transfer}
updated: Thu Sep 16 2021 23:51:26 GMT+0000 (UTC)
published: Mon May 31 2021 21:55:56 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アソシエイト