1. 程式人生 > >【深度學習】在Dog Breed Identification中使用Inception-V4遇到過擬合的問題解決進展

【深度學習】在Dog Breed Identification中使用Inception-V4遇到過擬合的問題解決進展

在Plant Seedlings Classification中自己根據谷歌的paper用keras搭建了Inception-V4,但在Plant比賽中因為資料集較簡單,沒有出現過擬合問題,模型成功收斂。但在Dog比賽中一開始便出現了過擬合,訓練集loss不斷下降,測試集loss在在3.6附近震盪。先在averagepool之後,fc之前加了0.5的Dropout,效果不理想,測試集loss依舊在3.6附近震盪;之後又在fc上加了0.01的L2正則,現測試集loss已下降至2.9。