arXiv:2606.06418cs.LGcs.AI2026-06被引 1

针对推理时误差累积问题,提出双预条件优化方法提升长序列任务表现。

Double Preconditioning (DoPr): Optimization for Test-Time Performance, not Validation Loss

论文配图:Double Preconditioning (DoPr): Optimization for Test-Time Performance, not Validation Loss
图 1 · 摘自论文原文
  • 结合梯度与激活预条件,改进传统优化器在推理时的性能
  • 在语言建模、生成模型等任务中显著提升生成质量与成功率
  • 适合关注推理阶段表现而非验证损失的长序列建模场景

许多现代深度学习应用通过单步预测损失(如L²回归、交叉熵)训练神经网络,但在部署时依赖自身预测结果进行滚动生成。典型场景包括自回归语言建模、基于流的生成建模和机器人策略学习。这类设置会引发测试时反馈(TTF)现象:训练/验证损失与下游指标(如任务成功率、生成质量)之间出现偏差,且随任务长度增长而加剧。尽管已有研究从数据、架构和目标设计角度应对训练-测试差异,本文提出将优化作为新设计维度,以缓解误差累积。我们引入一种名为双预条件(DoPr)的新优化范式,结合梯度预条件(如Adam、Muon)与激活预条件(如KFAC)。实验表明,引入激活预条件可作为即插即用的改进,在多种TTF设置下显著提升下游性能。有趣的是,这些测试时性能提升并不总伴随验证损失改善,这引发了对使用单步监督目标训练模型时评价方式的重新思考。

原文摘要 · Abstract (English)

Many modern applications of deep learning involve training a neural network via a one-step prediction loss (e.g., $L^2$ regression, cross-entropy), but deploy the network by rolling out along its own predictions. Key examples include autoregressive language modeling, flow-based generative modeling, and robot policy learning. It is well-documented that these settings induce a phenomenon we call test-time feedback (TTF): the mismatch between the training/validation loss and downstream metrics of interest, such as task success rate and generation quality, which grows with task length. While data curation, architecture, and objective design have been proposed to combat train-test shift in TTF settings, this paper proposes optimization as a new design axis to mitigate error accumulation. Specifically, we introduce a new optimization paradigm called double-preconditioning (DoPr) uniquely tailored to the challenges of TTF. DoPr combines gradient-wise preconditioning, as in Adam and Muon, with activation-wise preconditioning (AP), such as in KFAC. We show that the addition of AP yields a drop-in intervention for increasing downstream model performance across a range of TTF settings. Interestingly, these gains in test-time performance do not consistently accompany improvements in validation loss, opening new questions about how to properly evaluate models trained with one-step supervised objectives.

优化算法生成模型推理优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。