摘要

针对XdeepFM模型将嵌入特征直接送入特征交互网络后使网络获取到大量冗余特征,同时模型二阶特征交叉能力不足的问题,采用图神经网络和注意力机制优化了模型的特征提取能力,并添加因子分解机算法提高了模型的二阶交叉能力,提出了GA-XdeepFM-FM新模型,并在公开数据集Criteo和Avazu上进行实验。实验结果表明:相比较XdeepFM算法以及经典算法,模型的AUC和Logloss皆有显著提高,证明了新模型在特征提取、特征交叉和预测广告点击率方面的有效性,为相关应用提供了更为有效的解决方案。

全文