Other | | arXiv:2609.31560

Generalization behavior of OPTQ and the role of regularization

OPTQ 的泛化行为与正则化作用

🧭 模型量化泛化理论

University of California, San Diego

Erin George, Rayan Saab

为 OPTQ 及其随机变体建立泛化误差上界,并提出更优的正则化参数选择

📌 工作参考价值

  • 在做 LLM 部署量化(如推荐系统的排序大模型、Agent 规划模型做 INT4/INT8 量化)时,OPTQ 校准集的正则项 λ 不要沿用默认值,可参考文中理论选择方式,可能显著降低线上量化误差。
  • 若业务场景数据分布随时间漂移(电商大促、季节性变化),可考虑 stochastic OPTQ 这类随机化量化,其泛化误差界不依赖特定校准集,对分布鲁棒性更强。
  • 论文提供量化误差与校准集大小、分布匹配的理论关系,可用于指导校准集构建:尽量保证校准集与线上分布同源、样本量足够,避免离线效果好但线上掉点。
  • 总体学术性较强,除上述适用点外,大部分理论结果难以直接落地;业务团队可关注其 λ 选择建议或对量化工具的超参数默认值做调整实验。

动机:大型神经网络存储和推理成本高,量化是重要压缩手段。OPTQ 是渐进量化算法,在给定校准数据集上最小化平方量化误差,但校准集误差与测试分布泛化误差之间关系缺乏理论分析,正则化项 λ 的选择也常靠经验。

方法关键点:研究 OPTQ 及随机变体 stochastic OPTQ 的泛化行为,在测试点来自固定分布的设定下推导期望平方误差上界。第一个结果将泛化误差与校准数据集(同分布独立样本)误差关联;第二个结果对任意足够好的分布约束 stochastic OPTQ 的泛化误差,且不依赖具体校准集。两个结果中正则项 λ 起关键作用,基于理论洞察提出新的 λ 选择推荐。

关键结果:新的 λ 选择在实验中优于文献中先前推荐,验证了理论指导的有效性。