因果推断推荐系统工具箱 - Influence Function for Unbiased Recommendation(二)

文章名称

【SIGIR-2020】Influence Function for Unbiased Recommendation

核心要点

继续上节对IF的讨论,并进一步分析如何利用IF进行样本权重调节来实现数据纠偏,讨论IF可以纠偏的基理。简述如何对IF进行求解。

方法细节

问题引入

上一节讨论了IF是什么,以及相对于IPS方法IF方法的优势。那么,1)为什么IF可以进行纠偏,2)又是如何加权样本的(权重怎么求解),3)怎么求解IF的值?

具体做法

如上一节所述,单个样本z_i的损失函数可以表示为如下图所示的式子。

IF

IF to all validation set

因此,训练样本z_i对所有验证集的影响可以表示为如下图所示的式子。如果\epsilon_i是训练样本z_i上的小扰动,\hat{\theta}_{\epsilon_i}是这个扰动后的最优模型参数。那么,利用如下式子近似代替(估计)验证损失的变化。

approximate validation loss change

对所有n个训练样本,验证损失的变化如下图所示。

whole validation loss change

把这个损失变化融合到模型训练的ERM中,其结果如下图所示。可以看到,它与IPS具有相同的权重调节的结构,每个样本的权重为\pi_i = 1 + n \epsilon_i \in [0, B],其中B是为了平衡bias和variance进行的权重截断。

objective function under perturbation

进一步,一个比较好的权重\pi = (\pi_1, ..., \pi_n)应该满足最小化无偏验证损失的条件,可以被形式化的表示为如下图所示的式子。

good sample weighting

从上式可以看出,\phi(z_i)较小的样本z_i具有较大的样本权重\pi_i。公式3表示\phi(z_i)由两部分组成,1)\sum_j \nabla_\theta l(z_j, \hat{\theta})表示在无偏的验证数据集上利用随机梯度下降法得到的参数梯度方向(相对于最优梯度\hat{\theta}来说);2)H_{\hat{\theta}}^{-1}\nabla_\theta l(z_i, \hat{\theta})有偏训练样本z_i\hat{\theta}的梯度方向的更新(因为是二阶导,所以是梯度方向的改变量)。因此,如果这两部分方向更相近,则\phi(z_i)更小。增大这两部分方向更相近的样本的权重,能够促使模型的学习朝向无偏验证集合的方向前进。这两点是IF可以用来纠偏的主要原因和动机,因为不需要求解Propensity Score,所以可以避免overlap(样本重叠,在不同Propensity的情况下都存在类似的或相同的样本)和unconfounderness(无未观测混淆变量)等IPS必备的假设。

理论上,\pi_i是有最优解的,即\pi_i = \{ if \ \phi(z_i) < 0, B; otherwise \ 0\}。之前的方法[1]令\pi_i = max(-\phi(z_i), 0)。但是由于这两种方法将不能保证莱伯尼茨连续,进而导致样本对验证集合过拟合,因此作者提出了如下图所示的权重计算方式,其中\alpha \in R^+,是超参数(有点类似softmax里的\tau)。

\pi

作者表示,这种类似sigmoid的函数是对公式5的一个平滑,而公式5是一个阶梯函数(不方便优化)。此外,经验估计时,\pi = (\pi_1, ..., \pi_n)服从正态分布,因此依据[2]中的做法,加入所以\pi_i和为1的约束,即\frac{1}{n} \sum_{i=1}^n \pi_i = 1

最后也是最重要的问题是如何计算IF。作者采用共轭提的方法(CG[3])计算如下式子中的h。随后,利用-\nabla_\theta l(z_i, \hat{\theta})^\top h计算每一个样本z_i的IF值。不同的损失函数或者模型需要由于,问题突性的原因,需要采用不同的损失函数。作者分析了LR方法,由于是凸优化,可以采用Mixed Preconditioned CG[4]来减少CG的迭代步数。而FM方法,由于非凸导致这种模型下的海森矩阵可能不是正定的,需要利用Gauss-Newton[5]方法来计算。具体的过程可以参见文章的附录部分[6]。

代码实现

IF4Rec的伪代码如下图所示。


pseudo code

心得体会

权重截断

在IPS的方法中,经常会使用到权重截断的方式进行偏差和方差的平衡,该方法也可以被应用到IF的方法中。

文章引用

[1] Ren, M., Zeng, W., Yang, B., and Urtasun, R. Learning to reweight examples for robust deep learning. arXiv preprint arXiv:1803.09050 (2018).

[2] Huang, J., Gretton, A., Borgwardt, K., Schölkopf, B., and Smola, A. J. Cor- recting sample selection bias by unlabeled data. In Advances in neural information processing systems (2007), pp. 601–608.

[4] Hestenes, M. R., and Stiefel, E. Methods of conjugate gradients for solving linear systems. JRNBS 49 (1952), 409–436.

[4] Hsia, C.-Y., Chiang, W.-L., and Lin, C.-J. Preconditioned conjugate gradient methods in truncated newton frameworks for large-scale linear classification. In Asian Conference on Machine Learning (2018), pp. 312–326.

[5] Schraudolph, N. N. Fast curvature matrix-vecctor products for second-order gradient descent. Neural Computation 14 (2002), 1723–1738.

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
禁止转载,如需转载请通过简信或评论联系作者。
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,732评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,496评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,264评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,807评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,806评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,675评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,029评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,683评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,704评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,666评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,773评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,413评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,016评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,204评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,083评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,503评论 2 343

推荐阅读更多精彩内容