Other | | arXiv:2607.09371

Spectrally Deconfounded Gradient Boosting

频谱解混淆梯度提升方法

🧭 因果推断中隐混淆鲁棒预测方法
Andrea Nava, Peter Bühlmann, Fabio Sigrist

通过频谱损失与早停相结合,在梯度提升中实现非线性隐藏混淆校正

📌 工作参考价值

  • 推荐系统中常存在未观测的混淆因素(如季节性营销活动同时影响用户点击和曝光),梯度提升模型易学到虚假关联。可直接将频谱损失嵌入 GBDT 训练,配合早停策略,在不增加推理成本的前提下提升分布外稳健性。
  • 光谱收缩(shrinking high-variance directions)相当于将协变量矩阵的主成分中可能携带混淆信息的方向进行弱化,类似在特征空间中做软性掩码,可作为推荐模型特征预筛选的参考思路。
  • 方法中混合模型解释给出的经验贝叶斯调参过程,可用于自动选择频谱损失中的超参数,避免手动搜索,对大规模工业排序模型的自动调优有借鉴意义。
  • 对广义线性似然和非线性混淆的扩展(Laplace 近似、核随机效应)表明该框架可适配分类、回归等多种目标,可尝试在点击率预估模型的 boosting 解决方案中应用。

动机:梯度提升在表格数据上预测能力极强,但对隐藏混淆敏感——当未观测变量同时影响协变量和结果时,模型会学到不稳定的虚假关联,导致分布外鲁棒性下降。现有频谱解混淆方法主要针对线性模型。

方法关键点:提出非线性的频谱解混淆梯度提升框架。核心是用频谱损失(spectral loss)替代普通平方误差损失,通过对协方差矩阵的高方差方向施加收缩,减慢模型在混淆方向上的学习速度。解混淆效果并非仅靠频谱损失,而是由频谱收缩与正则化(尤其是早停)的交互实现。进一步给出混合模型视角,将 LA V A 型收缩解释为随机效应调整,并推导经验贝叶斯调参过程。通过 Laplace 近似和核随机效应,方法被扩展到一般似然和非线性混淆场景。

关键结果:合成与真实数据实验表明,该方法能改善隐藏混淆下的目标函数估计,且比现有非线性频谱解混淆基线在可扩展性上显著更优。