arXiv Machine Learning By Shanbin Yu, Shaoyang Guo, Haoran Zhao, Danni Yu, Ziming Liu

Forking: Sudden Overfitting Under Replay

Read the original on arXiv Machine Learning →

The paper investigates a phenomenon called forking, a type of generalization failure observed in NanoGPT and similar autoregressive models. Forking manifests as a sharp divergence between training and validation loss at epoch boundaries, especially when models employ over‑encoding n‑gram memory modules and data replay. The authors reproduce the effect in both vanilla NanoGPT and a DeepSeek-style model, attributing it to repeated updates that sharpen seen continuations while suppressing unseen ones, with low‑frequency contexts contributing most to the loss gap.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv AI
Aug 20

Demystifying Training-Time Augmentation for Data-Constrained Language Model Pretraining

The paper investigates training-time data augmentation as a regularizer for autoregressive language model pretraining in data‑constrained, compute‑abundant settings. It introduces three orthogonal augmentation categories—token‑level noise, sequence permutations, and target offset prediction—and shows through systematic ablations that each category delays overfitting and reduces validation loss, with random token replacement performing best individually. Combining augmentation categories further lowers the minimum validation loss, demonstrating that such augmentations mitigate data inefficiency in autoregressive pretraining.

By Michael K. Chen, Xikun Zhang, Fan Bai, Zhengding Hu, Zhen Wang