arXiv:2605.02853cs.LG2026-05

通过分层检测揭示Transformer训练中的隐藏优化问题

Trust, but Verify: Peeling Low-Bit Transformer Networks for Training Monitoring

论文配图:Trust, but Verify: Peeling Low-Bit Transformer Networks for Training Monitoring
图 1 · 摘自论文原文
  • 逐层构建轻量参考模型,对比中间输出诊断优化状态
  • 在训练中途即发现未充分优化的层,优于整体损失曲线
  • 适用于二值化/量化场景,适合关注模型可靠性研究者

深度神经网络的优化效果难以评估,尤其在非凸训练空间中,传统指标无法反映各层学习质量。对于训练成本高、常被冻结使用的Transformer语言模型,低效层可能悄然降低性能。本文提出分层剥除框架,将每层独立优化于模型中间表示上,通过构造轻量层特定参考解并利用多种排列投影到多个中间输出,获得可实现基准,实现细粒度诊断。在仅解码器的Transformer模型上实验显示,这些分层基准在训练各阶段可达到甚至超越实际模型表现,暴露了聚合损失曲线无法察觉的优化缺陷。进一步证明该方法在二值化与量化设置下仍有效,而此类场景下训练动态尤为脆弱。所有数值结果一致表明,所提边界能区分表观收敛与真实最优性,揭示出仅依赖训练损失时不可见的优化机会。

原文摘要 · Abstract (English)

Understanding whether deep neural networks are effectively optimized remains challenging, as training occurs in highly nonconvex landscapes and standard metrics provide limited visibility into layer-wise learning quality. This challenge is particularly acute for transformer-based language models, where training is expensive, models are often reused in frozen form, and poorly optimized layers can silently degrade performance. We propose a layer-wise peeling framework for monitoring training dynamics, in which each transformer layer is locally optimized against intermediate representations of the trained model. By constructing lightweight, layer-specific reference solutions and projecting layers onto multiple intermediate outputs via different permutations, we obtain achievable baselines that enable fine-grained diagnosis of under-optimized layers. Experiments on decoder-only transformer models show that these layer-wise reference bounds can match or even surpass the trained model at various stages of training, exposing inefficiencies that remain hidden in aggregate loss curves. We further demonstrate that this analysis remains effective under binarization and quantized settings, where training dynamics are particularly fragile. Across all numerical results, the proposed bounds consistently separate apparent convergence from effective optimality, highlighting optimization opportunities that are invisible when relying on training loss alone.

Transformer训练监控量化优化诊断

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