Demystifying Training-Time Augmentation for Data-Constrained Language Model Pretraining
解密数据受限下语言模型预训练的训练时数据增强
🧭 训练时增强 · 多轮训练正则化
UC San Diego · RMIT University · Bloomberg AI
系统研究三类训练时增强(token噪声、序列置换、目标偏移)缓解自回归模型在多轮数据训练时的严重过拟合
📌 工作参考价值(4 点)
- 当训练数据有限需反复遍历(如用户行为序列)时,可在自回归下一行为预测中引入随机 token 替换或 masking 作为增强,延迟过拟合,降低验证损失。
- 组合多种增强类别(如随机替换 + 序列逆序 + 偏移预测)能进一步降低最小验证 loss,可作为通用训练 trick 加入推荐模型的预训练或微调流程。
- 目标偏移预测(预测未来第 k 个 token)可迁移到用户长序列建模中,强制模型学习更长期的依赖,减轻短视过拟合。
- 该方法实现简单,不改变模型结构,适合在现有自回归推荐(如 SASRec 类)或 LLM-based 生成式推荐中快速实验。