zoukankan      html  css  js  c++  java
  • Item-CF和User-CF算法训练过程优化的心得

    〇、背景

    数据集:MovieLens-1M(其中用户数6040,电影数3592,评分数1000000);

    工具:Java、MySQL、Redis;

    主要参考资料:《推荐系统实践》项亮;

    一、Item-CF算法描述

      1.Item-CF算法思想

        为当前用户推荐——与当前用户感兴趣物品相似的 且 当前用户没有接触的物品

      2.实现步骤

        (1)计算物品之间的相似度;

        (2)根据物品相似度和用户的行为历史,作出推荐。

    二、User-CF算法描述

      1.User-CF算法思想

        先找到目标用户的相似用户(即与他有相同兴趣的其他用户),然后把 相似用户喜欢的 且 目标用户未作出行为的 物品推荐给目标用户。

      2.实现步骤

        (1)计算用户兴趣的相似度;

        (2)找到与当前用户兴趣相似的用户集合,从该集合的历史物品列表中作出推荐。

    三、遇到的问题

      基于物品/用户的协同过滤算法是实现推荐系统的基本算法,同时也算是学习推荐系统的入门。两个算法的思想简单明了,所涉及的知识点也比较容易理解,但实现过程中也会有一些坑,比如说——训练过程太耗时。

      以Item-CF为例,在算法实现过程中,我先计算物品之间的相似度,并将结果以set的数据结构暂存到Redis(一个基于内存的键值对数据库),方便使用。

      而Item-CF算法最耗时的地方是 计算用户对某物品兴趣度的过程,公式如下:

      

      其中Puj表示,用户u对物品j的兴趣度。N(u)表示用户u喜欢的物品集合,S(j,K)表示与物品j最相似的K个物品,这里取K=10。而N(u)∩S(j,K)表示与物品j相似,且用户u感兴趣的物品。wji表示物品j和物品i的相似度。rui表示用户u对物品i的兴趣。

       每个用户对每个物品都需要计算,其时间复杂度大概为O(U*I*10),U为用户数,I为物品数,10表示与当前物品最相似的10个物品。

    四、重复步骤的优化

      这里的优化主要指的是,在MovieLens数据集下,计算用户对某电影兴趣度的过程优化。

      1.重复的步骤

      从下面的伪代码可以看出:当user=1 movie=1时,会从redis取与当前电影相似的10部电影;当user=2 movie=1时,又会取一遍!这样算下来,会有六千多次的多余,大大影响了效率。

    1 for(int user=1;user<=6040;user++){
    2      //从数据库中取当前用户的评分历史
    3             
    4      for(int movie=1;movie<=3952;movie++){
    5              //从redis中取与当前电影相似的10部电影
    6      }
    7 
    8  }

      2.优化方法:

      以空间换时间,在算法开始时就把所有电影的相似电影取出来当作Java静态变量待用。

      优点:避免了重复;缺点:占用了Java虚拟机的堆空间,使得垃圾回收更加频繁(当然,可以设置参数,加大堆内存解决)。

    五、数据库查询的优化

      1.数据库的多次查询

      从下面的代码可以看出:6040个用户,每循环一次就要查询一次数据库,即对用户的查询需要六千多次。

      而一次查询一个用户的数据,反复查6040次,和一次直接查询6040个用户,时间差是非常大的。

    1 for(int user=1;user<=6040;user++){
    2    //从数据库中取当前用户的评分历史
    3 
    4 }

      

      2.优化方法

        通过一次查询多个用户数据而非一次查询一个用户,可以大大提高效率。

        本方法其实是使用到了SQL语句中的“in”关键字来一次性查询多个用户数据,例如:Select * from user id in 1,2,3....

        但该注意的地方是,若一次性查询太多个用户的数据,会造成Java虚拟机内存溢出(Out Of Memory异常),所以得在实践中找到最佳的可以一次性查询的数据数量。

    六、计算过程优化

      这里的优化主要指的是,在MovieLens数据集下,计算用户对某电影兴趣度的过程优化。

      1.耗时

      即使通过上面的优化,但由于数据量大,且个人电脑的硬件限制等,耗时还是不可接受的(粗略计算超过12小时)。

      2.解决方法

      利用多线程,步骤如下:

      (1)方法切分

        将整个计算过程细分为若干个独立的计算单元,每个计算单元作一条线程。

        比如:将查询用户的评分历史作一条线程,将查询与本电影相似的电影作一条线程,将计算用户对某电影兴趣度作一条线程。

      (2)线程合作

        多线程的执行具有不可预知性,我们需要制约它们的前后顺序。我这里采用的是“读者-写者”的方法。

        比如:A线程查询用户后将结果传递给B线程;B线程收到A线程的数据后开始执行,并将本线程的结果传给C线程.....

        具体到Java中,可以使用BlockingQueue集合类来实现数据互斥地写入、读出。

      (3)效果

        用本方法优化后,可以达到一个"线程流水线"的效果,大大缩减算法训练时间,如图所示:

        

  • 相关阅读:
    blocking to nonblocking of Python
    hug -- Embrace the APIs of the future
    supplychain on blockchain
    xstate -- JavaScript state machines and statecharts
    计算PI -- 采用刘徽的割圆术方法
    Gunicorn
    AIOHTTP
    APScheduler
    prefect
    FastAPI
  • 原文地址:https://www.cnblogs.com/Drajun/p/12918328.html
Copyright © 2011-2022 走看看