Kaggle - Carvana Image Masking Challenge1の3位解法2の調査記事です.
3rd place solution (U-Net + Dilated Conv)
Name: lyakaap
Kaggle Discussion: https://www.kaggle.com/c/carvana-image-masking-challenge/discussion/40199
Code: https://github.com/lyakaap/Kaggle-Carvana-3rd-place-solution
Model
- 1536x1024と1920x1280の二種類
- 最良モデル(スコア: 0.997193)
- パラメータ数: 約850万
- KFold(CV=6)
- TTAなし
- アンサンブルはしていない
- 1920x1280
- TTA(オリジナルと反転画像を平均)することで,0.997222を達成
- Optimizer:
- RMSProp(lr=0.0002)
- ReduceLROnPlateau(0.2と0.5の2つでlrを調整)
- Data Augmentation:
- 左右反転
- 拡大縮小
- 平行移動
- HSVの変更
- BatchSize: 1
- 学習時間: 2日
- 損失関数: bce + dice loss
- アンサンブル:
- 5つの平均(1536x1024)
- 6つの平均(1920x1280)
- 2つのモデル自体はLBで加重平均
- しきい値: 0.508(スコアが0.000001改善するらしい)
- Pseudo Labeling
- Cross validation: K-Fold
References
-
Kaggle, Carvana Image Masking Challenge, 2017. ↩
-
lyakaap, 3rd place solution (U-Net + Dilated Conv), 2017. ↩