ファインチューニングについて
解決したいこと
事前学習済みモデルを用いてファインチューニングを行いたいのですが,モデルの一部を変えた場合もファインチューニングを行えますか?
例えばResnetにSEブロックを追加した自作モデルに,事前学習済みResnetを用いてファインチューニングできますか?
0
事前学習済みモデルを用いてファインチューニングを行いたいのですが,モデルの一部を変えた場合もファインチューニングを行えますか?
例えばResnetにSEブロックを追加した自作モデルに,事前学習済みResnetを用いてファインチューニングできますか?
できるかどうか論で聞いていらっしゃるのであれば,「できます.」
@gazounu
Questioner
こんにちは、お手伝いできて嬉しいです。日本語が分からないので、翻訳ソフトを通じてコミュニケーションを取っていますので、質問が多くなるかもしれませんが、どうぞお気軽にお聞きください。最初の質問は完全に理解できます。例えば、ResNetは、元の論文ではImageNet-1kデータセットで訓練されており、このデータセットは1000クラスのタスクです。自分で実装する際には、おそらくこれほど多くの分類カテゴリはないでしょうから、一般的なファインチューニングの手法としては、最後の全結合層を再構築し、前のCNN層のパラメータをすべて凍結して更新しないようにすることです。
これは論文のアドレスです:https://openaccess.thecvf.com/content_cvpr_2016/papers/He_Deep_Residual_Learning_CVPR_2016_paper.pdf