SIGMA: Scalable Spectral Insights for LLM Collapse
Yi Gu, Lingyou Pang, Xiangkun Ye, Tianyu Wang, Jianyu Lin, Carey E. Priebe, Alexander Aue
TL;DR
The paper tackles model collapse from recursive synthetic-data training by introducing SIGMA, a spectral framework that tracks representation health through the Gram matrix spectrum of embeddings. It develops deterministic and stochastic bounds on the Gram determinant, enabling scalable collapse monitoring even when full eigendecomposition is intractable. Two practical diagnostics, Sigma-UB Track I and Track II, are derived to provide a conservative envelope and a sensitive trend probe, respectively, with a size-corrected baseline to distinguish genuine geometric contraction from dataset growth. Experiments in controlled settings reveal that carrying forward weights across generations accelerates collapse under recursion (S2) compared to restarting from base (S1), illustrating the practical value of SIGMA for monitoring and diagnosing recursive training pipelines. The framework offers a principled, scalable tool for ensuring healthy embedding geometry in large-scale LLMs and informs more robust data-generation and fine-tuning strategies.
Abstract
The rapid adoption of synthetic data for training Large Language Models (LLMs) has introduced the technical challenge of "model collapse"-a degenerative process where recursive training on model-generated content leads to a contraction of distributional variance and representational quality. While the phenomenology of collapse is increasingly evident, rigorous methods to quantify and predict its onset in high-dimensional spaces remain elusive. In this paper, we introduce SIGMA (Spectral Inequalities for Gram Matrix Analysis), a unified framework that benchmarks model collapse through the spectral lens of the embedding Gram matrix. By deriving and utilizing deterministic and stochastic bounds on the matrix's spectrum, SIGMA provides a mathematically grounded metric to track the contraction of the representation space. Crucially, our stochastic formulation enables scalable estimation of these bounds, making the framework applicable to large-scale foundation models where full eigendecomposition is intractable. We demonstrate that SIGMA effectively captures the transition towards degenerate states, offering both theoretical insights into the mechanics of collapse and a practical, scalable tool for monitoring the health of recursive training pipelines.
