论文阅读:DAS: Dual-Aligned Semantic IDs Empowered Industrial Recommender System

基本信息 论文标题:DAS: Dual-Aligned Semantic IDs Empowered Industrial Recommender System 作者单位:快手 论文链接:https://arxiv.org/pdf/2508.10584 来源:CIKM 2025 Motivation:论文要解决的问题是什么 Semantic id生产时,要么没有和协同信号对齐(fig2(1)),要么是两阶段对齐方式(fig2(2)): 例如LETTER先生成协同emb,然后和semantic id对齐 或者例如QARM,先协同对齐emb,再生产semantic id 把协同对齐和生产semantic id分成两个阶段,天然有信息损失,不是最优的。本文的目的就是把生产协同emb,以及semantic id的协同对齐放到一个模型中联合训练完成,尽量减少信息损失(fig2(3))。 主模型 主模型如上图所示,中间的ICDM是user和item的双塔模型,用于学习user和item的协同id-based emb;两边分别是生产user和item的semantic id的量化模型。 中间的ICDM就是经典的召回双塔模型,使用点击样本进行训练,唯一不同的是,在user和item塔都有流行度去偏模块,用于学习user和item的无偏emb,后续user和item的semantic id协同对齐用的也是无偏的emb。 两边分别是user和item的semantic id量化模型,两者比较类似,以item为例: 先把item的各种信息,如title、desc、ocr等信息用文本构造成prompt,输入到LLM,借助LLM的summary和reasoning能力,产出item的详细描述 然后把LLM产出的描述再输入到一个预训练的embedding模型PLM,文中用的是bge m3模型,得到item emb 后续就是标准的RQ-VAE过程了 需要注意的是,上述前两步,分别用到了LLM和PLM两个大模型,而且看图上这两个模型都是freeze的,也就是说并不微调这两个大模型。后续协同对齐用的emb是RQ-VAE重构emb的中间层结果,即图中的item quantized emb。 semantic id的协同对齐方面,有三大类对齐任务: U2I对齐:量化user emb和协同item emb对齐、量化item emb和协同user emb对齐 U2U和I2I对齐:量化user emb和协同user emb对齐、量化item emb和协同item emb对齐 U2U和I2I的共现对齐:点击相同item的两个量化user emb对齐、同一个user点击的两个item的量化item emb对齐 由于fig3中的协同模型和semantic id模型是联合训练的,总共有3大类loss: 中间的ICDM的双塔召回模型的loss 两边的产semantic id的loss 三个模块的对齐loss 评论 可借鉴 把semantic id的生产和协同信号对齐统一成一阶段的模式,信息损失更少 中间的ICDM模型生产协同emb时进行了去偏,协同对齐的时候用的是去偏的emb,这是其他论文很少提到的 可改进 太复杂了!3个模块,3大类loss,每类loss又有很多个小loss,总loss数量加起来有十多个。。。 任务太多,各种去偏、对齐loss,真的不会互相影响吗? 中间的ICDM模块有必要吗?我理解ICDM本质是为了训练产出协同emb,但是因为训练样本本身是点击样本,样本本身已经包含了搜推场景的协同信号,也就是ICDM本身没必要存在了,直接用相同的样本训练两边的semantic id量化模型就行了,也能实现在训练semantic id的过程中,完成协同信号的对齐 生产semantic id的emb来自LLM和PLM,但是这两个大模型都是freeze的,如果把这两个模型也sft,效果会不会更好?其实我原本以为的一阶段就是这样的,这也是我在【论文阅读:Empowering Large Language Model for Sequential Recommendation via Multimodal Embeddings and Semantic IDs】中提到的一阶段方法。

October 5, 2025 · 1 min

论文阅读:QARM: Quantitative Alignment Multi-Modal Recommendation at Kuaishou

基本信息 论文标题:QARM: Quantitative Alignment Multi-Modal Recommendation at Kuaishou 作者单位:快手 论文链接:https://arxiv.org/pdf/2411.11739 来源:CIKM 2025 Motivation:论文要解决的问题是什么 多模态emb在搜推场景应用时通常采用如下图的两阶段方式,先预训练多模态emb,然后作为一个冻结特征放到搜推模型中。这种方式存在2个问题: 表征不对齐:多模态emb预训练的任务通常是图片分类或者文本的MLM,和下游搜推任务不对齐 表征不更新:多模态emb在搜推任务中作为冻结特征,没有更新 本文的方法就是想要解决上述2个问题。 对齐搜推任务的多模态emb预训练 为了解决多模态emb表征不对齐的问题,本文提出的多模态emb预训练任务直接对齐搜推场景,使用U2I和I2I召回模型,挖掘出相似item pair,然后通过对比学习微调多模态大模型。 具体来说,通过U2I和I2I模型,能够拿到item emb;然后用每一个target item emb去行为流中检索出最相似的商品,作为trigger item emb。<trigger, target>构成一对正样本,然后进行对比学习训练。 通过召回模型构造的训练样本,和搜推场景的协同信号对齐了,解决了开头提到的第一个问题,即表征不对齐的问题。 Semantic id生产方法 Semantic id的生产方法如上图右半部分所示,有两种方式: VQ:直接圈定一定数量(如N)的item emb作为底池,编号1~N,然后任意来一个item emb,通过对底池emb进行KNN搜索,找出top-k相似商品,假设是(a,b,…,k),则VQ编码的semantic id就是(a,b,…,k)。文中取k=25,感觉挺大的。。。 RQ-Kmeans:对圈定的N个item emb不断进行Kmeans聚类、求残差、残差继续Kmeans聚类的过程。文中取迭代次数为L=6,但是没说每次聚到多少个类。 注意:文中的RQ-Kmeans方法和RQ-VAE还不一样,RQ-Kmeans没有训练过程,也没有重构loss,纯粹是每次进行聚类,然后选聚类中心作为码本的过程。文中也没有对比过为啥不用RQ-VAE。 产出两套semantic id之后,直接在下游排序任务中进行端到端更新,解决开头提到的表征不更新的问题。具体建模方法比较常规,不是本文的重点,略讲。 评论 可借鉴 多模态emb预训练任务是i2i的,直接和下游搜推任务对齐 semantic id有两种产出方式,VQ和RQ-Kmeans,尽可能多地保留原始多模态emb的信息 可改进 多模态emb预训练和下游任务对齐,在2025年不算新鲜事了,常规操作。而且文中i2i的构造过程依赖U2I和I2I召回模型,有外部依赖,不够漂亮 VQ的方法,k=25这也太长了吧,相当于一个小型行为流了,会导致下游任务的特征处理更复杂 为什么用RQ-Kmeans而不是RQ-VAE,没有任何说明与对比 从pretrain emb量化成semantic id的过程中,存在严重的信息丢失,这在Empowering Large Language Model for Sequential Recommendation via Multimodal Embeddings and Semantic IDs论文中有讨论

October 4, 2025 · 1 min

论文阅读:Empowering Large Language Model for Sequential Recommendation via Multimodal Embeddings and Semantic IDs

基本信息 论文标题:Empowering Large Language Model for Sequential Recommendation via Multimodal Embeddings and Semantic IDs 作者单位:香港城市大学&腾讯 论文链接:https://arxiv.org/pdf/2509.02017 来源:CIKM 2025 Motivation:论文要解决的问题是什么 LLM4SR的基本范式如下,即用LLM直接来做搜推的范式(这种方式在学术界常见,但在工业界不常见)。由于LLM的输入词表范围是有限的(通常比较小),因此其token emb dim通常比较大,比如2048或者4096;而搜推场景的item量级很大,而且在不断更新,因此工业界经典的id-based的搜推模型的item emb dim通常比较小,比如64或128。经典的id-based的搜推模型能比较好地学习到搜推场景的协同信号,为了让LLM模型也能感知这种信息,LLM4SR范式通常会先预训练一个id-based的经典搜推模型,然后将其中的item id emb通过下图的Linear Projection的映射层,映射到LLM token emb的空间,让LLM也能感知搜推的协同信号。 上述LLM4SR范式存在两个问题: 维度坍缩:id-based训出来的id emb dim比较小(如64),LLM token emb dim比较大(如4096),在由id emb通过Linear Projection映射到toen emb的过程中,虽然64映射到4096空间了,但扩维后的矩阵存在低秩问题,即还是只利用了4096中的64维的空间。 论文中,作者分两种情况进行了分析,如果Linear Projection只是一个线性层的话,通过公式推导能得出上述结论;如果Linear Projection包含非线性变换,作者通过实验分析也发现了维度坍缩的现象。 灾难遗忘:除了使用id-based模型产出的id emb,LLM4SR也常用多模态模型产出item emb表征,然后转换成semantic id输入到LLM4SR中。在这种情况下,产出的semantic id通过会遗忘多模态item emb的信息,导致下游LLM4SR的效果不佳。 论文中,作者用公式9来衡量semantic id保留pretrain多模态emb的信息量。具体来说,如果行为流中的商品序列是{A,B,C,D},target item是E。使用pretrain多模态emb能计算出E和A~D的相似度,例如相似度<E,A> > <E,B>。如果将pretrain多模态emb转换成semantic id,然后由semantic id恢复出新的A~E的emb之后,再计算E和A~D的相似度,如果仍然有<E,A> > <E,B>,则认为一致(concordant),否则不一致(disconcordant)。这个分析方法挺好的,通过这个指标能估算出转换成semantic id之后,仍然保留原有pretrain多模态emb对搜推场景的序关系的保留程度。 作者发现,转换成semantic id之后,信息只保留了37.14%;进一步,如果semantic id是在下游任务中端到端训练的,则信息只保留了5.5%,也就是说94.5%的pretrain emb的序的信息都丢掉了,也就是灾难遗忘。 Semantic id构建方法 3套emb来源,一套id-based经典搜推模型产出的包含协同信号的emb,另外两套是LLM2CLIP产出的多模态文本和图片emb。作者提到传统CLIP对长文本处理能力较弱,所以升级到LLM2CLIP,能更好地处理长文本。 Semantic id构建方法是经典的RQ-VAE的方法,但有如下两个改进点: 将emb的重构loss由MSE升级成MMD (maximum mean discrepancy),MSE是计算原始emb和重构emb的欧式距离的误差,而MMD是计算两个分布的diff,实验表明能MMD比MSE能保留更多的pretrain多模态emb信息(即上述公式9),保留44.36% 对量化后的emb做了对齐,因为LLM2CLIP本身进行了图文模态的对齐,所以文中只新增了id emb分别和文本、图片模态的对齐 此外,还有一点论文没提但可能和常规RQ-VAE不同之处,就是原始emb在进行RQ-VAE之前,有一个Encoder升维的操作,在重构loss前对应有一个Decoder降维的操作,而semantic id量化恢复emb是Decoder之前的那个。这一升一降,估计也有助于缓解维度坍缩。 ...

October 4, 2025 · 1 min

论文阅读:Inductive Representation Learning on Large Graphs

摘要 大规模图网络的节点嵌入对很多任务有很重要的作用,比如在推荐系统领域、蛋白质网络研究领域等。然而,目前大多数节点嵌入方法要求所有节点都在训练集中,且难以泛化到未见过的新节点上,这些方法称为直推式(transductive)方法。本文提出了一个归纳式(inductive)的节点嵌入方法GraphSAGE,它利用节点属性生成未见过节点的embedding。GraphSAGE并不直接训练节点embedding本身,而是训练生成embedding的函数,这个函数通过采样并聚合节点邻居的特征来生成自身节点的embedding。在三个数据集上的实验结果表明,GraphSAGE的性能显著强于其他方法。 简介 Graph embedding很重要,然而大多数工作只能在一个固定的图上学习节点embedding,无法泛化到训练期间未见过的节点上,是直推式(transductive)方法。 但是,现实世界中的图网络往往是动态变化的,比如社交网络、用户和商品的交互网络、蛋白质相互作用网络等。归纳式(inductive)的方法训练完之后,可以泛化到未见过的节点上,甚至泛化到未见过的图上,比如在蛋白质相互作用研究中,使用某个器官的蛋白质网络训练的模型,可以泛化到另一个器官的蛋白质网络中,只要这两个网络中的蛋白质的属性有相同的格式。 本文的GraphSAGE就是一种归纳式(inductive)的方法,它的特点如下: 利用节点的属性特征(node features)生成节点的embedding,所以学习的是embedding生成函数,而不是embedding本身 因此,只要未见过节点的属性值域与训练集中的属性值域相同,就可以将embedding生成函数泛化到未见过的节点上,从而生成未见过节点的embedding GraphSAGE在生成embedding时,聚合了邻居的属性信息,所以既学到了邻居的拓扑结构,又学到了邻居的属性分布,一举两得 虽然GraphSAGE主要针对属性图,但由于它仍然利用了结构信息,所以也适用于无属性的图 GraphSAGE既可以用在有监督任务中,也可以用在无监督任务中 方法 GraphSAGE的全称是Graph SAmple and aggreGatE,所以其核心步骤就两步,采样(sample)与聚合(aggregate)。 如下图Fig1所示,先形象地理解一下GraphSAGE的过程。例如为了得到图中红色节点(目标节点target node)的embedding,第1步采样时,k=1采样的是其1-hop邻居;第2步采样时,k=2采样的是其2-hop的邻居。采样的过程是从由内到外进行的。 采样完成之后,开始聚合,聚合的过程是从外到内聚合的,即先聚合2-hop邻居到1-hop邻居上,再聚合1-hop邻居到目标节点上。由此我们得到的目标节点的embedding。 最后参数优化与损失函数有关,利用节点embedding,可以构造有监督或者无监督的loss,然后梯度下降进行优化。 接下来我们来详细看一下其伪代码,正文Algorithm 1给出了所有节点的前向过程,但是当图很大的时候,往往采用mini-batch的方式,我们直接看mini-batch的前向算法吧,如上图Algorithm 2所示。 大的流程仍然是采样与聚合,其中第1~7行为采样过程,第8~16行为聚合过程。 采样 以下图为例,假设网络层数\(K=3\),当前batch中只有红色节点。初始的时候令\(B^3\)只包含红色节点;采样的时候是从\(B^3\)到\(B^0\)进行的。\(B^2\)采样的是\(B^3\)的1-hop邻居,同时加上\(B^3\)本身,所以\(B^2\)包括图中的绿色+红色节点。类似的,\(B^1\)采样的是\(B^2\)的1-hop邻居,同时加上\(B^2\)本身,所以\(B^1\)包括图中的蓝色+绿色+红色节点。类似的,\(B^0\)包括图中的黄色+蓝色+绿色+红色节点。 其实在采样的时候,每个节点都只采样它自己的1-hop邻居,但是由于存在第5行求并集的操作,所以对于初始的红色节点来说,最终采样到的\(B^0\)包含了其最多\(K=3\)-hop的邻居。 另外一个值得提醒的是,采样的过程是从\(B^3\)到\(B^0\)降序进行的,主要是为方便后续聚合的时候从从\(B^0\)到\(B^3\)进行。 采样的目的有两个: 不同节点的邻居数目相差很大,如果不进行采样的话,热门节点的数目会非常多,导致训练有偏,而且不同batch的样本量大小也相差很大,不方便预估每个batch的训练时间 采样之后,每个batch训练时只跟当前采样的\(B^0\)里面的节点有关,网络参数更新时也只需要更新与\(B^0\)相关的参数,而不需要更新所有参数,可以大幅缩减训练时间 聚合 聚合操作就是聚合邻居的embedding,来更新自身的embedding。聚合与采样类似,也是分层进行的,只不过方向和采样相反。比如\(K=3\)时,需要聚合3层,每层又需要聚合多次。下图展示了\(k=1,2,3\)时的聚合情况。 以\(k=1\)为例,此时,所有在\(B^1\)里的节点都是目标节点,都需要聚合邻居的信息,包括如下聚合过程: 黄色节点→蓝色节点 蓝色节点→绿色节点 绿色节点→红色节点 上面→表示聚合方向。注意所有→左边的embedding都是\(h^{k-1=0}\)的embedding,即上一个循环时的embedding。比如第2步用的蓝色节点并不是第1步聚合得到的蓝色节点,而是上一个循环得到的蓝色节点(上一个循环为初始\(h^0\))。所以,上述三次聚合互不影响,可以并行进行。 当所有节点聚合完成之后,→右边的embedding变成了\(h^{k=1}\)的embedding,作为下一层\(k=2\)时的左边embedding。 如上图所示,当\(k=2\)时,最外层的黄色节点已经不参与计算了,此时包括如下聚合过程: 蓝色节点→绿色节点 绿色节点→红色节点 虽然绿色节点还是只聚合其直接邻居蓝色节点,但是由于蓝色节点在上一轮中聚合了黄色节点,所以绿色节点在这一轮中能够通过蓝色节点间接聚合到黄色节点,即绿色节点聚合到了其2-hop邻居。类似的,红色节点也聚合到了其2-hop邻居即蓝色节点。 当\(k=3\)时,蓝色节点也已经不参与计算了,此时包括如下聚合过程: 绿色节点→红色节点 根据上面的分析,红色节点能间接聚合到其3-hop邻居,即最远聚合到黄色节点的信息。 三层聚合结束之后,最终我们得到了红色节点的embedding。可以看到,为了得到红色这一个节点的embedding,如果网络层数为3的话,其最终聚合了三层节点的信息。在GraphSAGE中需要设置采样参数,例如fanouts=[20,10,5],就表示第一层每个节点采样20个邻居,第二层每个节点采样10个邻居,第三层每个节点采样5个邻居。这样每个节点最终聚合了20*10*5=1000个邻居节点的信息。可见,邻居聚合的威力很大,只需要少数几层就可以聚合大量邻居节点。GraphSAGE文中说只需要两层,fanouts=[25,10]就取得了很好的效果。 聚合函数 上述操作只是把红色节点的邻居聚合到一起了,相当于收集到了红色节点的邻居,怎样根据邻居embedding来生成自身节点的embedding呢,这就需要聚合函数来完成了。 有关聚合函数的描述,我觉得原文有点描述不清楚,我这里总结一下,可分为四种聚合函数,如下图所示: 所有聚合函数都有两步,第一步是聚合邻居信息,第二步是进行非线性激活,差别在于邻居的定义,以及聚合操作。 Mean aggregator是最简单的聚合操作,即把邻居(不包含v本身)求均值,然后和自身concat起来,最后非线性激活。 GCN aggregator和mean aggregator非常像,它们的区别是,GCN aggregator在聚合邻居的时候,也聚合了它本身,即GCN认为v也是v的邻居之一(相当于有自回路)。但是它在非线性激活的时候,没有和自身上一个状态concat,而这个caoncat操作类似ResNet中的短路原则,可以避免长距离信息丢失的问题。因此,GCN aggregator的网络不能太深,而且往往效果不如Mean aggregator。 Pooling aggregator对所有邻居先过一个MLP(公式中的\(W_{pool}\)和\(b_{pool}\)),然后进行element-wise的max pooling,接着把pooling结果和自身concat,最后非线性激活。作者测试发现这里使用max pooling和mean pooling的效果相当。 ...

May 31, 2022 · 1 min

论文阅读:LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation

摘要 GCN模型是不是越复杂越好呢?这篇文章分析发现,GCN中常用的矩阵变换(feature transformation)和非线性激活函数(nonlinear activation)没有作用,甚至有反作用,据此作者提出了一个非常简单的GCN模型LightGCN,模型参数只有节点的embedding。这么简单的模型在推荐任务上,比大多数复杂模型的性能都要好,而且作者从理论分析了如此设计存在的若干好处。 简介 作者所在团队在2019年发表了一个NGCF的模型,该模型基于user和item的交互关系网络,使用GCN训练得到user和item的embedding,然后使用embedding相似度进行推荐。 简单来说,第k+1层的user和item的embedding使用如下公式计算。其中的W1是直接对embedding进行变换的矩阵,W2是对user和item点乘之后进行变换的矩阵;而σ是非线性激活函数。 以user为例,右边有两项,第一项是对user在第k层的embedding进行矩阵变换;第二项是邻居聚合。其中邻居聚合又有两项,第一项是对item的embedding进行矩阵变换;第二项是对user和item点乘之后进行矩阵变换。 作者发现,对于协同过滤任务来说,由于user和item都只有ID本身,没有很多的属性,所以并不需要复杂的矩阵变换和非线性激活函数。言下之意是,如果节点有丰富的属性信息的话,非线性变换和激活有用?感觉可以这么理解:有些属性重要,有些属性不重要,所以需要非线性激活函数进行识别?如果只有节点ID的话,ID的embedding的所有维度都是重要的,不需要非线性激活,直接线性加权聚合就行了。 然后作者对NGCF模型进行了简单的消融实验,如下表所示,NGCF就是原始的NGCF,NGCF-f、-n、-fn分别表示去掉矩阵变换W1和W2、去掉非线性激活函数σ、同时去掉W1、W2和σ。很意外的是,-f、-n、-fn居然都比原始的NGCF效果好,而且-fn效果最好。说明对于只有user和item顶点,没有属性的网络来说,不用过于复杂的矩阵变换和非线性激活,效果反而更好。 按道理NGCF的参数空间比NGCF-f大,且前者能覆盖后者(只需要把W1和W2设置成单位矩阵),为什么前者的效果反而比后者差呢?作者进一步分析了两者训练时的loss和recall曲线,发现NGCF的参数空间虽然比NGCF-f大,但其收敛后的loss更大,recall更小。也就是说训练效果反而不如NGCF-f。作者认为,加入过多的矩阵变换和非线性变换,导致模型过于复杂,难以训练到较好的效果。据此,作者提出了一个更简单的模型LightGCN,具体看下一节介绍。 方法 既然前面分析说矩阵变换和非线性激活函数会起副作用,LightGCN的方法非常简单,就是把这两个操作去掉。如公式3所示,每个节点的embedding表示直接等于其邻居的embedding的线性加权求和,既没有矩阵变换,也没有非线性激活函数,如此的简单。而且,对比公式3和公式1可知,LightGCN没有显式使用自回路,即计算某个节点的embedding的时候,只用了其邻居的embedding,没有用自己的embedding;而NGCF在公式1中使用了自回路。 其网络结构图如下: 最后,节点的最终embedding等于其各层embedding的加权求和。如公式4所示,权重系数α可以手工指定,也可以使用注意力网络来自动学习。为简便起见,本文直接设置为等权重,所有系数都等于1/(K+1),相当于所有层embedding求平均。 节点最终embedding等于各层embedding的加权求和有如下三个好处: GNN存在over-smoothing的问题,即随着网络层数越深,深层网络的输出结果趋向于相同。即所有节点的最后一层的输出有可能很接近。而如果把所有层加起来的话,能一定程度上缓解这个问题 GNN不同层捕获的语义信息不一样,使用所有层输出能增强表达能力,这个和CNN的道理是类似的。 所有层embedding求和可以捕获自回路的信息。也就是说虽然公式3没有显式使用自回路,但计算顶点最终embedding时(公式4)可以隐含自回路的信息,这个后面会给出证明。 除此之外,由于LightGCN很简单,所以也很好训练,更容易收敛,收敛效果更好。总之,虽然LightGCN很简单,但它很强大,而且有很多好处。 模型分析 接下来,作者分析了为什么LightGCN可以学习到自回路,其证明思路是这样的。另一篇工作SGCN和这篇工作很像,也做了很多简化,且显式添加了自回路。作者通过分析发现LightGCN可以表达SGCN的形式,间接说明LightGCN隐含可以考虑自回路。下面是具体的证明过程。 首先定义user和item的交互矩阵\(\mathbf{R}\in \mathbb{R}^{M\times N}\),其中M和N分别表示user和item的个数。\(\mathbf{R}_{ui}\)为1表示u和i有交互,等于0表示没有交互。则全图的邻接矩阵可以表示为公式6: 公式3的矩阵形式可以表示成公式7,其中矩阵D为度矩阵。对照下原始的GCN公式,其实就是把原始GCN的变换矩阵W和非线性激活函数σ去掉了。 由于LightGCN是将多层embedding加权求和,所以最终结果是公式8: 然后作者对SGCN的公式进行了简单的变换,发现形式上和公式8是一致的,所以LightGCN也能隐含学习到自回路特征。 此外,作者还分析了另一个模型APPNP,APPNP借鉴pagerank的思想,可以缓解GNN过深带来的over-smoothing问题。然后作者如法炮制,对APPNP的公式进行变换,发现也和公式8等价,所以LightGCN也能缓解GNN的over-smoothing问题。其实这个从LightGCN不只使用最后一层,而是使用所有层embedding就能得到这个结论,不需要这么大费周章证明。 实验 最后是实验环节。作者将LightGCN和本文开头提到的NGCF进行了对比,实验结果表明,LightGCN的性能相比NGCF有显著提升,而且比NGCF-fn也高。作者提到,虽然NGCF-fn已经去掉了矩阵变换和非线性激活函数,但NGCF-fn仍然还有自回路、user和item的点积、dropout等等,还是比较复杂,不好训练。而LightGCN非常简单,只有embedding和邻居的线性加权,所以LightGCN还是比NGCF-fn好。真的很神奇啊,照这个说法,难道连dropout也会起副作用? 此外,在消融实验中,作者还对比了LightGCN和LightGCN-single,LightGCN-single是只用LightGCN的最后一层作为节点的embedding。作者发现,当网络层数增大到4层时,LightGCN-single性能显著下降,出现了over-smoothing的问题。而LightGCN由于多层组合的操作,不会有over-smoothing的问题。考虑到性能和收益,LightGCN使用了3层神经网络。 评价 结果有些意外,LightGCN这么简单的模型,效果居然比复杂模型还要好?感觉即使是NGCF,模型也不复杂啊,和CV、NLP那些大模型相比简单多了,怎么就训练不好了呢?难道是GNN特有的现象? 感觉和数据有关,本文测试的数据是只包含user和item顶点,顶点没有属性。如果是属性图的话,也许结论会有变化。 不过至少提供了调参的思路:去掉矩阵变换、去掉非线性激活函数、甚至是去掉dropout。。。另外重要的一点是,不要只用最后一层的embedding,而是组合所有层的embedding进行加权求和。 另外,本文开篇提到的NGCF和本文是同一批作者,自己批判自己一年前发表的工作,不免让人担心这篇工作的可靠性。。。以及当时的NGCF难道没有做本文开篇的消融实验吗?难道不应该吗?

May 8, 2022 · 1 min

DCN论文解读《Deep & Cross Network for Ad Click Predictions》

摘要 特征工程是很多预测模型成功的关键因素,然而特征工程并不简单,且通常需要人工设计或穷举搜索。通用DNN模型能够学习到所有隐式的特征交叉,但并不是所有特征交叉都有用、都能学好。本文提出Deep & Cross Network (DCN),它保留了DNN模型,同时引入了一个新的交叉网络cross network,交叉网络能更加高效地学习特定阶以内的特征交叉。特别地,DCN在每一层都进行显式的、自动的特征交叉,既不需要人工特征工程,也不会增加太多的模型复杂度。实验结果表明,DCN在CTR任务和非CTR任务上都取得了显著的性能优势,且内存消耗最低。 简介 CTR预估对CPC(cost-per-click)广告很重要,而特征交叉对CTR预估很重要。目前的特征交叉依赖于手工或穷举完成,并且很难泛化到没见过的特征交叉上。 本文提出了一个新的神经网络cross network,它能自动地进行显式特征交叉。Cross network包含多层网络,每增加一层网络会产生更高一阶的特征交叉,同时保留上一层的特征交叉结果。所以层数越多,交叉阶数越高,最高交叉阶数取决于网络的层数。除了cross network,我们仍然保留了传统的DNN网络,cross network和DNN网络组成Depp & Cross Network (DCN)。在DCN中,cross network和DNN网络联合训练,其中cross network以较小的参数量捕捉显式的特征交叉,而DNN网络以较多的参数量捕捉非常复杂的、隐式的特征交叉。实验表明DCN具有显著的性能优势。 相关工作 因子分解机相关的工作如FM、FFM,表示能力不足。通用DNN网络,很强大,可以近似任何(arbitrary)函数。然而,在现实问题中,感兴趣的特征交叉往往不是任意的(arbitrary)。在Kaggle比赛中,大多数获胜方案中使用的是手工交叉的特征,它们的阶数比较低,并且是显式的、有效的;而DNN网络学习到的特征是隐式的,且是高度非线性的。因此,有必要设计一个网络,它能自动学习到有限阶的显式的特征交叉,且比通用DNN更加高效。 主要贡献 设计了一个cross network,它能在每一层进行显式的、自动的特征交叉,无需人工特征工程。 Cross network简单高效,随着网络层数的加深,特征交叉的阶数也不断上升,且网络能学习到从低阶到高阶的所有交叉项,所有交叉项的系数也各不相同。 Cross network内存高效,易于实现。 实验结果表明,DCN比DNN的目标损失更低,且参数量少了一个数量级。 DCN网络结构 图1 DCN网络结构 Embedding层 CTR预估任务的输入特征既有稠密特征(例如身高、体重等),也有稀疏特征(例如性别、国籍等)。在DCN网络的输入层,会先将稀疏特征通过embedding矩阵变换为低维稠密向量(图1最低层的右边两个Sparse feature),然后和其他稠密特征拼接在一起作为整体的输入特征\(x_0\)。 Cross Network 图2 Cross layer图解 Cross Network的每一个交叉层如图2所示。核心就是公式(3),即第\(l+1\)层的交叉结果\(x_{l+1}\)等于对第\(l\)层的结果\(x_l\)做一个变换\(f(x_l,w_l,b_l)\)然后加上\(x_l\)。这里加\(x_l\)有点像ResNet的思想,就是直连,保留上一次交叉的结果。 公式(3)的核心操作是\(x_0x_l^T\),通过这个操作,一方面可以产生从1到\(l+1\)阶的所有交叉项;另一方面\(x_0x_l^T\)的维度是d*d(d为\(x_0\)的维度),使得系数\(w_l\)是一个d*1的向量而不是矩阵,即\(w_l\)只需要很少的参数量就能增加一阶的特征交叉。由于每增加一层,只增加两个维度为d的参数向量\(w_l\)和\(b_l\),所以对于\(L_c\)层的cross network,参数量是\(d*L_c*2\),相比于DNN的参数量要少得多。 手工推算一下,可得: $$ x_1=x_0*x_0*w_0+b_0+x_0=x_0^2w_0+b_0+x_0 $$$$ x_2=x_0*x_1*w_1+b_1+x_1=x_0^3w_0w_1+x_0^2(w_0+w_1)+x_0(b_0w_1+1)+b_0+b_1 $$可以看到,随着网络层数的增加,交叉的阶数也不断上升,而且完整包括了从0到\(l+1\)的交叉项,每个交叉项的系数也各不相同。 Deep Network 深度网络就是经典的MLP,公式如下: 由于参数\(W_l\)是一个矩阵,故DNN的参数量远多于cross network。 Combination layer 最后将cross network和deep network的输出拼接起来,过一个logits layer,预测当前item的CTR。本文使用point-wise的方法,即训练样本是每个item是否点击,所以是一个二分类问题,损失函数是交叉熵损失。 实验结果 相比于DNN、FM等方法,DCN的测试误差最低,且参数量、内存开销最小。此外,作者还将DCN用于其他非CTR的分类任务中,也取得了性能优势。 总结 DCN的思路简单、巧妙,通过\(x_0\)和上一层的交叉结果相乘,则所有交叉项的阶数都上升一阶,所以随着层数的增加,交叉阶数也在不断增加。通过控制网络层数可以控制需要交叉的最高阶数。而且\(x_0x_l^T\)的设计,每层增加的参数量也很小,不错。 ...

January 22, 2022 · 1 min

【论文速读】End-to-End Differentiable Learning of Protein Structure

首先放出本文的Hightlights: 蛋白质三级结构预测一般分为两种方法,一种是基于模板的预测方法(Template-Based Modeling, TBM),另一种是从头测序方法(Free Modeling, FM)。TBM方法目前已经能达到比较好的预测精度,但并不是所有蛋白都有同源模板,当模板蛋白和目标蛋白的相似性低于某个阈值时,TBM方法的性能就会比较差。而传统的FM方法,需要过多的人工特征,使得整个流程非常复杂。 作者在一次报告中,将传统的蛋白质结构预测算法(上图)比作10年前的图像识别算法(下图),虽然10年前的图像识别算法也能达到比较好的性能,但需要很多人工设计的特征,比如SIFT特征等,不够简洁漂亮。随着深度学习的兴起,现在图像识别不再需要人工设计特征,只要搭建好神经网络,输入原始图片即可完成识别和分类,性能比之前的人工方法还要好。所以,作者也希望能提出一个简洁、纯深度学习的模型来预测蛋白质的三级结构。 本文的模型(Recurrent Geometric Networks, RGN)从宏观上来说就如博客开篇的图片所示,非常的简洁漂亮,输入是蛋白质的一维序列,经过神经网络,输出是每个氨基酸残基的三个扭转角,然后再通过三维重构,得到蛋白质的笛卡尔坐标。 更具体来说,RGN包括三个部分,分别是模型预测、三维重构和误差反向传播,下面分别介绍这三个部分。 模型预测是上图的左下角部分,即输入是蛋白质序列,输出是每个氨基酸残基的三个扭转角。因为每个氨基酸对应三个扭转角输出,每个氨基酸和其上下文的氨基酸有关联,所以使用双向LSTM最合适不过了。Bi-LSTM没什么好讲的,关键讲讲其模型的输入和输出。输入部分,作者把每个氨基酸编码成一个41维的向量,如上图所示,其中包括20维氨基酸的one-hot向量(因为只有20种氨基酸)、20维PSSM位置向量和1维具体的位置信息。其中的PSSM位置向量可以理解为这个位置上的不同氨基酸的概率分布,由于有20种氨基酸,所以PSSM向量维度也是20。网上没有找到氨基酸的PSSM向量示例,找到一个DNA的,如下图,每个位置上,字母越大表示出现该核苷酸的概率越大,换成氨基酸是类似的道理。所以整个网络的输入,除了PSSM矩阵,没有任何人工设计的特征,已经很优雅了。 https://davetang.org/muse/2013/10/01/position-weight-matrix/ Bi-LSTM的输出是三个扭转角,但并不是三个实数这么简单。作者首先把整个拉氏图平面聚类,比如聚类成m=60个点,然后就把输出离散化成60类的分类问题。分类输出采用Softmax归一化,这样就会得到60类的概率分布,如RGN网络图最右边的子图所示。60类的概率分布再通过加权平均的方式得到最终的三个扭转角的实数值。我很好奇为什么需要经过一个离散再加权平均的方法,Bi-LSTM直接回归输出三个扭转角的实数不是更省事吗? 预测得到三个扭转角之后,进入RGN的第二个阶段,就是三维重构,在RGN网络图的左上角。三维重构说起来也简单,就是根据每个氨基酸残基的三个扭转角,重构出蛋白质的三维结构。由于常规的蛋白质三维结构坐标系是笛卡尔坐标系(直角坐标系),所以需要把扭转角坐标转换为笛卡尔坐标,以便于求解误差。这个部分作者没有细说,因为是另一篇论文:Parallelized Natural Extension Reference Frame: Parallelized Conversion from Internal to Cartesian Coordinates。 最后就是怎样求解误差以及误差反向传播了。这个也比较有意思,想想看,对于一条长为L的蛋白质序列,给定预测的三维结构和真实的三维结构,怎样计算它们之间的误差。不能直接对应坐标相减,因为有可能两个坐标系的坐标原点不一样。作者的方法是这样的:对于预测结构,求每两个氨基酸的距离差,就是\(\tilde{d}_{j,k}\);对于真实结构,也做类似的操作。这样做的好处是抹掉了坐标原点的影响,用两点之间的相对距离来表示三维结构。然后,对真实的\(\tilde{d}_{j,k}^{(exp)}\)和预测的\(\tilde{d}_{j,k}^{(pred)}\)再求二范数\(||D||_2\),最后除以长度进行归一化,就得到了误差dRMSD。 上述dRMSD误差相比于之前领域内常用的TMscore,好处就是可微分,可自动求导,可梯度下降了;另外,如上所述,dRMSD不要求预测结构和真实结构进行对齐;但是有一点是dRMSD对size敏感,而且不能识别镜面对称这种错误结构,比如左手和右手的结构是镜面对称的,如果真实结构是左手,但模型预测成了右手,dRMSD是检测不到这种错误的。 模型介绍完毕,训练和测试数据集来自CASP竞赛。作者把每一届CASP比赛的数据集作为测试集,从CASP7~CASP12;每一届比赛之前公布的所有PDB数据集(seq, structure)作为对应测试集的训练集。其中,CASP11分出一部分作为验证集,用来优化网络超参数。 测试结果如Table 1所示,可以看到,在没有模板的FM类别中,本文的RGN预测误差是最小的;在有模板的TBM类别中,RGN的性能几乎垫底,当然这里参与评测的都是当届比赛中Top-5的模型,所以RGN和这5个模型比是垫底,但差距是很小的。 另外,作者提到,在TBM类别的数据集中,CASP的参赛模型比较依赖模板的质量。具体来说,对于真实的结构,如果模板结构和真实结构误差很小(y轴),则模型的预测结构和真实结构的误差也很小(x轴),这两个变量成一定的线性相关关系(第一行)。而对于本文的RGN,则没有这种相关关系,说明RGN一视同仁,不会受模板质量的影响,因为RGN是纯深度学习的模型,根本就没有用到模板。 在预测速度上,RGN虽然需要训练几周甚至上月的时间,但预测速度是毫秒级别的,是评测的几个模型中最快的。快速的RGN能使一些新的应用成为可能,比如药物发现、蛋白质设计等。 最后,总结一下本文的主要工作、创新点和局限性: 三个特点: torsional angles,局部信息 geometric units ,全局信息 dRMSD,局部+全局 创新点: 简洁,Model replaces structure prediction pipelines with one mathematical function 另辟蹊径,纯deeplearning,不依赖structural templates、co-evolutionary information、energy model等,本文预测融合了本文方法和领域知识的新模型有望解决蛋白质结构预测问题 局限性: 依赖PSSM矩阵 本文作者来自哈佛医学院系统药理学实验室,文章只有作者一个人,很了不起了。哈佛医学院的另一个教授评价作者:“AlQuraishi 研究的特点在于,一名埋头在哈佛医学院和波士顿生物医学社区丰富研究生态系统中的研究人员,居然能够在计算机科学最热门的领域里抗衡谷歌等巨头。——Peter Sorger”,太棒了,我也想做这样的研究。 ...

August 30, 2019 · 1 min