工业推荐的优化重心,已经从单一交互指标转向整体eCPM(每千次曝光收益)。
一次曝光之后,用户可能点击、下载、付费、关注……这些目标分布在一条“转化漏斗”上,越往下越稀疏,但往往越值钱。
业界的标准做法是多任务学习(MTL):让这些目标共享表示、一起训练。
论文《Personalized Task Dependency Graphs for Mitigating Signal Erosion in Multi-Task Recommendation》则关注一个具体问题:
沿着固定漏斗逐级传递信息时,信号会一层层衰减,最需要帮助的深层稀疏任务反而拿到最弱的信号。
论文把这种现象叫作“信号侵蚀”(signal erosion),并提出PTDG(Personalized Task Dependency Graphs)来对其进行缓解。

MMoE、AITM这类常用架构,大多默认一个前提:漏斗里任务之间的依赖强度,对所有流量都一样。
论文举了一个例子:
物理上的先后顺序(先点击才可能付费)是必须遵守的,但相邻任务之间的关联强度会随物品而变。把所有样本都塞进同一个固定拓扑,一方面限制了模型的表达能力;另一方面,信息必须一级一级往下传,深层任务拿到的信号已经被中间环节稀释了。
已有工作也关注任务关系,但各有局限:PMTRec个性化的是优化权重,没有刻画任务间的结构差异;MIT的任务关系全局共享;CSRL学习因果结构,但迭代搜索代价达到O(T²)甚至更高,难以满足工业时延要求。

PTDG的核心思路,是把任务依赖看作一张动态、可学习的拓扑,而不是写死的链条。对应论文总结的三点创新:动态拓扑学习,按实例构造依赖强度;可控传播,用带硬因果掩码的GCN建立自适应捷径;自适应优化,用结构化掩码从参数层面缓解梯度冲突,同时降低计算复杂度。

PTDG不直接学一个完整的T×T邻接矩阵(T是任务数),而是从物品embedding(由DeepFM或MLP这类骨干模型提取)生成两组低秩因子,相乘得到物品级的依赖矩阵。低秩分解在这里起结构正则的作用:工业交互数据噪声大,直接学满秩矩阵容易拟合到虚假相关。低秩形式还让依赖图计算和GCN消息传递的开销都随任务数T线性增长。
在此基础上,模型再用用户和场景特征生成一个“个性化偏置向量”,去调制物品级的低秩因子。这样,同一个物品在不同用户、不同场景下,依赖强度也可以不同。
得到动态依赖矩阵后,PTDG先乘上一个硬因果掩码,保证信息只能沿行为顺序传递,防止“付费→点击”这样的语义倒流;然后在任务节点上做GCN传播。
和固定链条相比,关键差别在于:依赖图可以在相关性强的任务之间建立自适应捷径,比如让“点击”的信息直接到达“付费”,不必经过冗余的中间节点。论文认为,这正是缓解信号侵蚀的地方。
任务之间传递信息,也可能加剧梯度冲突。PTDG的做法不是在梯度层面做投影或重加权,而是给每个任务学一个二值掩码,只让它看到共享参数里的一部分:
与PCGrad这类需要O(T²)次梯度投影的方法不同,APM带来的训练开销可以忽略。此外,各任务的损失权重用“任务损失/其指数滑动平均”来归一,把不同任务拉到同一量级。
数据:公开数据集KuaiRand1K(840万样本,6个任务,正样本率0.02%–37.9%),以及一个工业数据集(1320万样本,7个任务)。每个设置用20个随机种子重复,报告均值。对比方法包括MMoE、PLE、STEM、MoCoGrad、PMTRec、MIT。
两个数据集上的平均AUC如下表:

两个数据集上,PTDG的平均AUC都是最高的;论文还报告,PTDG的平均表现比MoCoGrad高1.59%。优势在稀疏的深层任务上尤为明显:
(以上百分比都是论文正文的原始表述,每个都对应特定任务和特定对比方法。)
在点击这类稠密浅层任务上,PTDG与各基线保持相当水平。论文据此指出,个性化图拓扑在缓解深层节点信号侵蚀的同时,不损害头部任务的表现。
消融实验在工业数据集上进行,指标为平均AUC:

每个部件都有明确贡献。按论文正文的表述:去掉个性化依赖图,平均AUC下降1.21%;GCN换成简单拼接下降0.93%,说明图上的消息传递更能保住信号;去掉APM下降0.59%。
低秩维度:q从1调到7,q=2时最好,说明低秩近似用比满秩邻接矩阵更少的参数,有效控制了过拟合。论文给出的经验值是q≈T/3。
PTDG被部署为一个主流应用分发平台的排序模型,该平台日活超过1亿。线上A/B持续两周:基线桶和实验桶各占10%流量,其余80%仍走原线上模型;基线是当时线上的MMoE(也就是离线对比里的同一个MMoE)。论文对每个桶的日级指标做双样本t检验(p<0.05),核心业务指标的提升在统计上显著:
面向工业约束:计算随任务数线性增长,线上端到端延迟相比MMoE基线只增加8 ms,在服务SLA之内;APM掩码训练后固定,不增加推理开销。
收益落在最难的目标上:提升集中在稀疏、高价值的深层转化任务,同时保持稠密任务的精度。
线上验证:在日活超过1亿的平台上取得统计显著的CVR和eCPM提升,论文认为这体现了PTDG作为多任务学习框架在真实推荐系统中带来稳定收益的工业可行性。
最适合的场景:深层转化稀疏、不同物品转化路径差异明显的业务,正是PTDG收益最集中的地方。
在因果约束内个性化:依赖图的个性化始终在硬因果掩码允许的范围内进行,从结构上避免了语义倒流。
更大规模与跨域:论文提到,下一步将扩展到更大的任务集合,以及跨域场景。
论文标题:Personalized Task Dependency Graphs for Mitigating Signal Erosion in Multi-Task Recommendation
收录会议:CIKM 2026(罗马)
论文(arXiv):https://arxiv.org/abs/2609.04862
DOI:10.1145/3799682.3839876
文章来自于"量子位",作者 "PTDG团队"。
【开源免费】graphrag是微软推出的RAG项目,与传统的通过 RAG 方法使用向量相似性作为搜索技术不同,GraphRAG是使用知识图谱在推理复杂信息时大幅提高问答性能。
项目地址:https://github.com/microsoft/graphrag
【开源免费】Dify是最早一批实现RAG,Agent,模型管理等一站式AI开发的工具平台,并且项目方一直持续维护。其中在任务编排方面相对领先对手,可以帮助研发实现像字节扣子那样的功能。
项目地址:https://github.com/langgenius/dify
【开源免费】RAGFlow是和Dify类似的开源项目,该项目在大文件解析方面做的更出色,拓展编排方面相对弱一些。
项目地址:https://github.com/infiniflow/ragflow/tree/main
【开源免费】phidata是一个可以实现将数据转化成向量存储,并通过AI实现RAG功能的项目
项目地址:https://github.com/phidatahq/phidata
【开源免费】TaskingAI 是一个提供RAG,Agent,大模型管理等AI项目开发的工具平台,比LangChain更强大的中间件AI平台工具。
项目地址:https://github.com/TaskingAI/TaskingAI