← 全部深度解读
📖 深度解读 · PERSONALIZATION

Hierarchical Exponential-Gaussian Mixtures for Watch-Time Distribution Prediction

Sofia Gulevskaia, Mikhail Trapeznikov, Aleksandr Poslavsky, Alexander D'yakonov · arXiv 2608.23356 · 发表 2026-08-25
11 分钟阅读时长 4280字 4.80综合分 第 1 名08-26 Personalization 日报
相关 5新颖 4实用 5证据 5时效 5
观看时长不应只回归一个点值,分布建模更适合短视频中“秒退、正常观看、长时观看”并存的反馈。HEGM 的层级分解和稳定性正则可以直接接入多目标精排,但落地时应先检查混合分量是否塌缩,再用线上分桶验证长尾用户与新内容的收益。

这篇论文从短视频推荐线上最实际的反馈信号 watch time(WT)出发。全屏自动播放场景里,用户没有必要先点击,CTR 因而不再是足够可靠的满意度代理;观看时长是每次曝光都能得到的连续信号,但它并不是普通回归问题。论文在工业数据上明确观察到近零膨胀、右偏长尾和多峰:大量内容会在几秒内被划走,也有部分视频被看完、重播或长时间观看。视频长度还会带来 duration bias,长视频机械地拥有更大的可观测 WT。因而精排更关心候选之间的相对顺序、超过某个观看阈值的概率和不确定性,而不是只拟合一个平均时长。

作者复现前作 EGMN 时发现,最大似然训练的高斯混合会出现三个生产级故障:方差趋近于零形成尖峰,多个高斯分量合并或长期不活跃,以及结果对初始化非常敏感。最直接的证据是工业数据上基础 EGMN 的 XAUC 只有 0.6585,甚至低于简单 MSE 回归的 0.7070;混合分量数量增加并没有带来有效容量。论文的动机因此不是再加一个更大的 backbone,而是诊断分布式预测头为什么在真实精排中退化,并把 skip、partial watch、completion、replay 等行为模式用一个稳定、可解释、可以直接替换旧 head 的分布表示出来。

2. 方法详解

论文保留 EGMN 的基本建模直觉:一个指数分布负责近零的快速划走,多个高斯分布负责不同的有效观看状态。对用户、视频和上下文特征 x,EGMN 先由共享 backbone 得到 h,再分别预测指数率 lambda、高斯均值 mu_k、标准差 sigma_k 和混合权重 w_k。其密度可写为 p(y|x)=w_0 f_exp(y|lambda)+sum_k w_k f_gauss(y|mu_k,sigma_k^2),期望 WT 可直接解析计算为 w_0/lambda+sum_k w_k mu_k。这样一份分布可以同时导出期望观看时长、P(Y>tau)、按视频长度归一化的完成概率和预测方差。

HEGM 的第一个关键改动是把“跳过”和“真正观看”分成两级。sigmoid gate 先输出 p_skip(x)=sigmoid(W_skip h+b_skip),完整密度变成 p_skip f_exp+(1-p_skip) sum_k w_k f_gauss。也就是说,指数分量不再和所有高斯分量争夺一个扁平 softmax;门控值可以直接解释为快速跳过倾向,剩下的概率质量专门在有效观看的多峰分布中分配。高斯权重仍用 softmax 归一化,保留简单的多模式解释。

第二个关键是全局归一化与结构化初始化。目标先按数据集级常数 s 做 y=y_raw/s,不采用按用户、按视频或按时长的归一化,因为后者会破坏跨候选的绝对排序关系。高斯均值在归一化支持上均匀初始化,论文实际使用确定性均匀初始化;标准差初始化为 1.5/K,指数率用短观看先验 lambda=1/0.05,其中 0.05 在工业尺度约对应 9 秒。这个初始化使不同高斯从训练第一轮就分开,避免所有分量从同一点开始而互相吞并。

第三个关键是删除 EGMN 的两个设计。HEGM 不再把每个高斯均值强行平移到指数均值 1/lambda 之上,而让有效观看分量自由适应真实数据;也不再加入熵正则,因为实验显示强行鼓励权重均匀反而降低效果。对大 K 仍可能出现 sigma_k 过小,于是加入按视频时长分桶的 KL 方差正则。对包含样本的视频长度桶 b,先计算训练集参考方差 sigma_bar_b^2,再对每个预测高斯加入 D_KL(N(mu_k,sigma_k^2)||N(mu_k,sigma_bar_b^2))。这个项只约束方差,不约束均值,所以可以阻止尖峰,却不会抹平不同观看模式。最终损失是 NLL+lambda_reg L_reg+lambda_KL L_KL,L_reg 为 MAE。

实验使用固定 backbone 的替换式 head 评估。公开数据用 PLE 编码数值特征、embedding 编码类别特征和 DCNv2 并行堆栈;工业 backbone 是带 wide logits、user embedding 以及 click、like、share、WT 等任务塔的多任务结构。所有模型用 Adam 最多训练 30 epochs,batch size 4096;高斯分量 K 在 3、6、9、12 中选择,KL 权重在 0.05、0.1、0.5、1.0 中选择,回归权重在 0.5、1.0、2.0 中选择,方差正则桶数 M=18。验证集只用 XAUC 选 checkpoint,最终测试结果平均 3 个随机种子,数据按时间切分以避免未来信息泄漏。

3. 核心技术贡献

第一,论文把混合模型失败模式变成了可定位的工程问题:EGMN 的高斯方差塌缩会记忆单点,组件合并会让 K 变大却不增加表达能力,纯指数退化又会丢失多峰行为。HEGM 用行为层级和初始化分别处理“跳过/观看分离”和“分量可分离”。

第二,sigmoid skip gate 让概率结构与业务行为结构对齐。它不只是换一个激活函数,而是把近零质量和有效观看质量放到不同的条件分布中;因此可以同时做 session depth、deep watch、completion 和 uncertainty 的下游预测。

第三,KL 方差正则把视频长度条件下的统计尺度注入混合头。它在 K=12 等过参数化配置下提供了防塌缩的安全阀,同时保留各分量的均值自由度;这种“只正则不确定性尺度,不强迫兴趣中心一致”的设计比增加熵正则更贴近 WT 的异质性。

第四,作者把可复现离线比较、消融和线上验证串成完整闭环:代码、预处理、划分、配置和 checkpoint 公开,线上只替换 WT head,检索和特征生成保持不变,从而尽量隔离模型头的因果效果。

4. 实验结果分析

论文的三套数据规模很有代表性:KuaiRec 为 7,176 个用户、10,728 个视频和 12,530,806 次交互;VK-LSVD 为 100,000 个用户、171,106 个视频和 38,404,921 次交互;工业集为 6,390,738 个用户、2,484,528 个视频和 282,443,525 次交互。KuaiRec 以 60 秒分位数归一化,VK-LSVD 以 157 秒归一化,工业集以最大 180 秒归一化;三者都按时间切分。

数据集/模型XAUCMAEMSE
KuaiRec MSE-VR0.54674.6559.17
KuaiRec HEGM(K=3)0.56224.4959.29
VK-LSVD EGMN(K=3)0.617316.07480.70
VK-LSVD HEGM(K=6)0.620615.74478.92
Industrial MSE-VR0.707013.12569.99
Industrial EGMN(K=3)0.658513.51730.74
Industrial HEGM(K=3)0.718812.97578.61

工业集最能说明问题:HEGM 的 XAUC 0.7188 高于 MSE-VR 的 0.7070,且比 EGMN 的 0.6585 高 0.0603;MAE 也降到 12.97。KuaiRec 上 HEGM 的 K=3 XAUC 0.5622,优于 MSE-VR 的 0.5467 和 EGMN 的 0.5504;VK-LSVD 上 K=6 的 0.6206 优于 EGMN K=6 的 0.6163。它没有在每个数据集、每个 K 上同时最优的 MSE,但论文把生产排序目标放在 XAUC 和阈值事件上,而不是只追求点误差。

消融结果进一步显示,最重要的不是 KL,而是结构化初始化。工业 K=3 完整 HEGM 的 XAUC/MAE/MSE 为 0.7188/12.97/578.60;去掉结构化初始化后变成 0.6841/13.54/644.15,下降幅度明显。改回扁平混合后 XAUC 为 0.7164,恢复强制均值平移为 0.7171,加入熵损失为 0.7153,说明这三个 EGMN 设计都拖累结果。KL 在 K=3 时略微把 XAUC 降到 0.7179,却对大 K 的稳定性关键:在 sigma<10^-3 的标准下,未正则 HEGM 的 K=12 塌缩率为 47.88%,HEGM+KL 在 K=3、6、9、12 全部为 0%。

线上实验在生产精排系统中只替换 WT head,配置 K=3、lambda_reg=1.0、lambda_KL=0。1.5 个月用户级随机 A/B 中,处理组和基线组各承接 5% 流量,每组约 1,100 万请求/日,并采用先正向一个月、再反转两周的 forward-reverse 设计;CUPED 校正 t 检验的所有报告指标 p<0.05。非短观看提升 4.59%,深度观看提升 5.75%,跳过下降 6.23%,session depth 从 28.52 增至 31.16 个视频、提升 9.26%;分享提升 14.59%,不喜欢下降 8.34%。总观看时长仅变化 -0.03%,点赞仅 +0.10%,这说明收益主要来自减少即时跳过和增加有效观看事件,而不是简单拉长总消费时长。

5. 局限与未来方向

作者明确承认三点局限:HEGM 不是显式因果去偏方法,无法单独解决 duration bias 或曝光策略造成的反事实偏差;未来可以寻找更适合 WT 支持集和形状的其它分布族;现有评估还没有全面覆盖概率质量,需要进一步考察校准、sharpness 等指标。论文还显示,K=12 即使加入 KL 前也会塌缩,说明正则强度与混合规模需要配套调参。作者没有把工业数据公开,这使完整复现实验受限;此外,线上结果来自单一短视频平台,外推到社区 UGC 的观看长度、内容类型和反馈节奏仍需验证。

6. 工业实践意义

对社区搜索和推荐,HEGM 最值得借鉴的是“一个分布,多种业务信号”。同一用户—帖子—上下文表示可以输出期望停留、超过 3 秒/10 秒的概率、读完率以及不确定性,避免为每个阈值单独训练一套模型。实际接入可以先保持现有多任务 backbone,只将 WT/停留时长 head 替换为 K=3 的层级混合头;K=3 在论文中没有出现塌缩,因此第一期不必上更大的 K。

训练时应保留全局时长归一化和按时间切分,重点监控 XAUC、阈值 AUC、skip rate、session depth 与 guardrail,而不只看 MAE。上线前先做混合分量健康检查:sigma 是否逼近零、权重是否长期只剩一个分量、不同视频时长桶是否共享同一方差。若要扩展到 K>6,再开启以视频长度为条件的 KL 方差正则。论文的部署成本相对可控:一次训练约 8 小时,使用 4 张 H100;TensorRT 服务侧在 10 亿请求评估上将中位延迟增加 1.22ms,p99 从 7.9ms 到 8.2ms,远低于 30ms 约束,参数量只从 1.3M 增至 1.5M。要避免的做法是直接把总观看时长当唯一目标,因为线上结果表明总时长几乎不变,真正改变的是跳过、深度观看和分享结构。

← 全部深度解读 出现于报告:2026-08-26 · Personalization #1 (4.80) 原文 ↗