Conference paper Open Access
Some like it tough: Improving model generalization via progressively increasing the training difficulty
Fassold, Hannes
In this work, we propose to progressively increase the training difficulty during learning a neural network model via a novel strategy which we call mini-batch trimming. This strategy makes sure that the optimizer puts its focus in the later training stages on the more difficult samples, which we identify as the ones with the highest loss in the current mini-batch. The strategy is very easy to integrate into an existing training pipeline and does not necessitate a change of the network model. Experiments on several image classification problems show that mini-batch trimming is able to increase the generalization ability (measured via final test error) of the trained model.
| Name | Size | |
|---|---|---|
|
aspai_2021_fassold_pdf_version.pdf
md5:da0eca6acff8e644aa86bdfaff378840 |
349.2 kB | Download |
| All versions | This version | |
|---|---|---|
| Views | 50 | 50 |
| Downloads | 43 | 43 |
| Data volume | 15.0 MB | 15.0 MB |
| Unique views | 45 | 45 |
| Unique downloads | 41 | 41 |
- Publication date:
- October 25, 2021
- DOI:
-
- Grants:
-
European Commission:
- AI4Media - A European Excellence Centre for Media, Society and Democracy (951911)
- Communities:
- License (for files):
- Creative Commons Attribution 4.0 International