早停
Early Stopping常用验证集表现不再变好时就停止训练,取表现最好的那一版模型。
早停是最常用的正则化手段之一。训练时定期在验证集(不参与训练、专门用来挑模型的数据)上评估:训练损失还在降、验证损失却开始回升,说明模型开始过拟合,这时停止训练并退回验证表现最好的检查点。实践中常设一个「耐心值」,连续若干轮没有改善才停,避免被波动误导。机器人模仿学习要特别注意:验证损失低不等于真机成功率高,robomimic 的研究指出训练目标和评测目标不一致,停在哪一步对结果影响很大,很多工作会存多个检查点,再用仿真或真机推演来挑。注意别和强化学习里的「提前终止」(回合中摔倒等情况就结束该回合)混淆。
例子假设训练一个抓取策略,每个 epoch 算一次验证集动作误差:第 30 轮最低,之后连续 10 轮都没再降,于是停止训练,取第 30 轮的检查点。
- 也叫
- 提前停止
- 相关
- 过拟合、训练集 / 验证集 / 测试集、检查点、正则化、训练轮次、提前终止
- 来源
- Wikipedia: Early stopping
What Matters in Learning from Offline Human Demonstrations for Robot Manipulation (robomimic)
Google Machine Learning Glossary