Read this paper write-up in English
Data Quality LSA:预训练协方差、SFT 与 outcome supervision¶
论文信息¶
| 字段 | 内容 |
|---|---|
| 论文链接 | arXiv 2603.01293;固定 v1 全文 |
| 原文标题 | Theoretical Perspectives on Data Quality and Synergistic Effects in Pre- and Post-Training Reasoning Models |
| 公司 / 机构 | 一作 Adel Javanmard:University of Southern California / Google Research |
| 首次公开日期 | 2026-03-01(arXiv v1) |
| 原作者代码 | 原文 v1 与官方机构页未找到/未发布实现仓库;不声称全网没有代码 |
| 本地 adapter / 稳定键 | data-quality-lsa;adapter: null,component_only |
| 本地复现代码目录 | src/auto_research/foundation_models/google16_data_quality_lsa.py;scripts/run_google16_neural.py |
| 完整实验 | GPT2 多网格、多重复长训练成本过高,不复现(审查 DONE) |
原始论文总结¶
背景与主要改动¶
以 in-context 线性回归的权重预测分析预训练与后训练数据协方差。SFT 使用教师中间状态,OS 仅监督最终预测;原文明确 OS 是 outcome-supervised regression,不是包含采样、优势估计和 policy gradient 的完整 RL。§3、§4 给出线性注意力结构及闭式解。
flowchart LR
C[预训练协方差] --> I[population初始V和W]
X[线性回归上下文xy] --> Z[分块prompt]
Z --> A[无softmax线性self attention]
I --> A
A --> O[自主多步rollout与最终OS]
T[训练权重标签] --> S[教师中间状态SFT]
S --> A
X --> F[最近初始化SFT闭式解]
原论文关键图¶
Figure 4(a),LSA的SFT样本规模结果:这里只展示原图的(a)子图,非整张六面板图,非本地实验。原文未提供模型结构图,本地不伪造原图。版权归原作者;来源与hash。
核心公式¶
按 Eq.3.2–3.8 保留输入、标签、预测权重、常量四个分块。OS 每步追加实际生成 token,最终损失为 \(\|\widehat w_k-w^*\|^2/(2B)\);SFT 采用 \(w_i=[1-(1-\eta)^i]w^*\),对 \(i=0,\ldots,k\) 的下一 token 求和。
Eq.4.1 的非零 value block 为 \(\widetilde V_*=-\eta\Omega\Phi^\dagger-\Gamma_0^{-1}(I-\Phi\Phi^\dagger)\),固定 \(\widetilde W=I\)。实现保留初始化在数据零空间的分量,而不是仅返回最小范数拟合。
论文离线与线上效果¶
原文以合成协方差偏移和约9.5M参数 GPT2结构验证理论趋势;未报告线上 A/B。本地仅验证公式、梯度与闭式解正规方程,不宣称验证普遍的“大模型 RL 数据质量规律”。
本地复现¶
population_initialization、context_embedding、lsa_forward、rollout、sft_loss、outcome_supervision_loss 与 closest_sft_value 可组合执行。自主 rollout 没有真实权重参数,只读取公开任务上下文;训练目标只进入明确命名的训练 loss。
PYTHONPATH=src python scripts/run_google16_neural.py \
--device cpu --seed 42 --output runs/google16-neural-cpu.json
runner 使用 d=4、6个生成 prompt 的数学诊断,fixture_only: true,没有 benchmark train/test 能力结论。独立测试核验矩阵乘法重排、首步预条件协方差预测、SFT/OS监督边界、最优解正规方程和零空间。CUDA 回执不改变 L0/L1 资格;没有 Evolve 执行映射。
成本与未复现范围¶
原文 §8 包含 d=100 LSA长训练、12层8头d256的GPT2结构,SFT100步、OS12K步及十次重复,并遍历多种数据规模与协方差。这不是本轮可用单卡的短时组件任务;完整网格成本过高,不下载或试跑。保留精确低维数学核心,不以缩小fixture冒充完整实验。
