动态调整嵌入向量的维数

本文探讨了在深度推荐系统中,如何通过动态搜索嵌入向量大小来提高性能并减少内存需求。针对不同用户和项目的频率,提出了一种策略网络,该网络基于强化学习动态调整嵌入大小,以实现更准确的预测和资源效率。这种方法通过硬选择策略,一次专注于一个嵌入向量大小,避免了过拟合和存储开销。
摘要由CSDN通过智能技术生成

汤吉良老师团队,发表于2020的SIGIR

《Automated Embedding Size Search in Deep Recommender Systems》论文阅读笔记

  1. 背景:
    深度推荐系统,将用户项目表示为低维嵌入向量,然后输入到深度神经网络中进行预测。基于深度学习的推荐系统,能够很好的学习用户,项目的特征表示,而且还能建模用户,项目之间的非线性关系。
  2. 已有方法:
    用户,项目的嵌入向量大小固定统一。软选择嵌入向量的大小。
  3. 存在问题:
    推荐性能并不是最优,并且空间复杂。
    (1) 不同的用户,项目有不同的频率,不同的嵌入大小会影响模型的性能(低维嵌入更适合于频率较低的用户/项目。在数据较少的情况下,低维嵌入可以很好的训练,但是高维嵌入可能会导致过拟合)
    (2) 用户,项目的频率是动态变化的;
    (3) 对所有的用户项目使用固定统一的嵌入大小需要更多的内存。
  4. 本文:
    动态搜索嵌入向量的大小,并引入了一个嵌入大小调整策略网络。硬选择嵌入向量的大小,做出更准确的选择,并且减小了存储空间。
    在这里插入图片描述
    给定第k个用户项目交互对,策略网络首先选择合适嵌入向量大小,然后查找嵌入表
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值