基于用戶特征遷移的協(xié)同過濾推薦

柯良文,王 靖

(華僑大學(xué)計(jì)算機(jī)科學(xué)與技術(shù)學(xué)院,福建廈門361021)

·先進(jìn)計(jì)算與數(shù)據(jù)處理·

基于用戶特征遷移的協(xié)同過濾推薦

柯良文,王 靖

(華僑大學(xué)計(jì)算機(jī)科學(xué)與技術(shù)學(xué)院,福建廈門361021)

為提高推薦系統(tǒng)在數(shù)據(jù)稀疏情況下的推薦質(zhì)量,提出一種基于用戶特征遷移的協(xié)同過濾推薦模型。利用矩陣分解技術(shù)提取輔助領(lǐng)域的用戶特征,通過建立正則項(xiàng)約束的矩陣分解模型,將輔助領(lǐng)域的用戶特征遷移到目標(biāo)領(lǐng)域中,協(xié)助目標(biāo)領(lǐng)域用戶特征的學(xué)習(xí),最終生成目標(biāo)領(lǐng)域的用戶推薦。設(shè)計(jì)快速收斂的Wiberg算法得到模型的最優(yōu)解,并對實(shí)際應(yīng)用中的可行性進(jìn)行分析。通過對2個(gè)公開數(shù)據(jù)集的實(shí)驗(yàn)結(jié)果表明,該模型能夠?qū)崿F(xiàn)輔助領(lǐng)域用戶特征的遷移,有效提高目標(biāo)領(lǐng)域的推薦質(zhì)量。

數(shù)據(jù)稀疏;用戶特征遷移;協(xié)同過濾;矩陣分解;Wiberg算法

1 概述

隨著互聯(lián)網(wǎng)的不斷普及和應(yīng)用,網(wǎng)絡(luò)上的信息量正呈指數(shù)式的增長。用戶一方面可以方便獲取到豐富信息,另一方面則要面臨過量信息伴隨著的信息過載問題[1]:無法從海量的信息中獲取到對自己有用的部分。個(gè)性化推薦系統(tǒng)根據(jù)用戶的喜愛和偏好,從海量的信息中發(fā)現(xiàn)用戶感興趣和有價(jià)值的信息,并將其推薦給用戶。目前,推薦系統(tǒng)已應(yīng)用到多種領(lǐng)域,如電子商務(wù)網(wǎng)站的 Amazon,eBay等,電影網(wǎng)站的MovieLens,Reel.com等,新聞網(wǎng)站的GroupLens等。

協(xié)同過濾是推薦系統(tǒng)應(yīng)用最為成功的技術(shù)之一,其基本假設(shè)是:如果用戶X和用戶Y對于n個(gè)項(xiàng)目有相似的評分或購買行為,那么他們對其他項(xiàng)目也會有類似的評價(jià)。根據(jù)這一假設(shè),協(xié)同過濾技術(shù)首先要度量目標(biāo)用戶和其他用戶的相似度,然后選取與目標(biāo)用戶相似度最高的前k個(gè)用戶作為最近鄰居,最后通過這些最近鄰居的興趣偏好為目標(biāo)用戶作出推薦。然而在實(shí)際的應(yīng)用中,用戶評分的項(xiàng)目在整個(gè)項(xiàng)目集中往往只占其中一小部分,導(dǎo)致整個(gè)評分矩陣十分稀疏,從而在計(jì)算用戶間的相似度時(shí)不夠準(zhǔn)確,降低了推薦系統(tǒng)的推薦質(zhì)量。

針對協(xié)同過濾出現(xiàn)的稀疏性問題,研究者提出了多種不同解決辦法,如文獻(xiàn)[2]利用奇異值分解(Singular Value Decomposition,SVD)技術(shù)移除沒有代表性或者無關(guān)緊要的用戶和項(xiàng)目,以達(dá)到用戶-項(xiàng)目評分矩陣降維的目的;文獻(xiàn)[3]提出了一種先聚類再經(jīng)過非負(fù)矩陣分解的兩階段聯(lián)合聚類協(xié)同過濾算法,通過縮小原始評分矩陣規(guī)模以提高非負(fù)矩陣分解算法面對稀疏矩陣預(yù)測上的準(zhǔn)確度;文獻(xiàn)[4]提出了一種變權(quán)重相似度計(jì)算和自適應(yīng)局部融合參數(shù)的協(xié)同過濾算法,來緩解數(shù)據(jù)稀疏和數(shù)據(jù)集異構(gòu)的問題;文獻(xiàn)[5]提出一種基于奇異值分解和增強(qiáng)Pearson相關(guān)系數(shù)的特征遞增算法(HybridSVD)來克服數(shù)據(jù)稀疏性的問題。然而,這些方法普遍存在一個(gè)問題:只局限在單個(gè)領(lǐng)域的學(xué)習(xí)任務(wù)中,而沒有利用其他領(lǐng)域的知識來改善推薦系統(tǒng)的質(zhì)量。近年來,多領(lǐng)域?qū)W習(xí)成為推薦系統(tǒng)的一個(gè)新的研究方向[6-7],特別是將遷移學(xué)習(xí)的方法應(yīng)用到協(xié)同過濾算法中已受到學(xué)者們的關(guān)注。關(guān)于遷移學(xué)習(xí)在協(xié)同過濾技術(shù)應(yīng)用的研究現(xiàn)狀,本文將做詳細(xì)介紹。

針對數(shù)據(jù)稀疏的問題,本文在矩陣分解模型的基礎(chǔ)上,基于遷移學(xué)習(xí)的思想,提出一種用戶特征遷移的協(xié)同過濾推薦模型(TUF)。通過學(xué)習(xí)輔助領(lǐng)域的用戶特征,并將其遷移到目標(biāo)領(lǐng)域的學(xué)習(xí)任務(wù)中,幫助提高目標(biāo)領(lǐng)域的推薦精度。同時(shí),采用Wiberg算法對目標(biāo)模型進(jìn)行快速求解以獲得最優(yōu)解。

2 相關(guān)研究

2.1 問題定義

推薦系統(tǒng)中普遍采用一個(gè)m×n的矩陣R表示所有用戶對項(xiàng)目的評分,R的行列數(shù)m和n分別表示用戶數(shù)和項(xiàng)目數(shù)。R的元素值ri,j∈{「a,b」U}代表用戶對項(xiàng)目的評分,其中,a和b分別代表評分值的上下限;?表示該項(xiàng)未評分。通過一個(gè)與評分矩陣R大小相同的標(biāo)記矩陣W來表示某一項(xiàng)是否被評分,W的元素值wi,j∈{0,1},其中0代表該項(xiàng)未評分,1代表該項(xiàng)已評分。協(xié)同過濾的目標(biāo)就是通過評分矩陣R已知的評分項(xiàng)去預(yù)測未知項(xiàng)的評分值。

2.2 基于矩陣分解的協(xié)同過濾

基于矩陣分解的協(xié)同過濾技術(shù)認(rèn)為用戶-項(xiàng)目評分矩陣R是一個(gè)近似逼近低秩的矩陣[8],采用矩陣Z表示無缺失值的評分矩陣,并且用d表示Z的秩,其中,d＜＜min(m,n),那么Z可以分解成如下形式:

分解后的矩陣U和矩陣V可以分別表示用戶特征矩陣和項(xiàng)目特征矩陣。實(shí)際情況下,用戶-項(xiàng)目評分矩陣R是帶有噪聲的矩陣,假設(shè)噪聲符合等方向的高斯分布,則對于U和V的最優(yōu)估計(jì)可以通過下面的的損失函數(shù)來確定:

其中,||·||F表示Frobenius范數(shù)。

在實(shí)際應(yīng)用中,由于R是一個(gè)極度稀疏的矩陣(缺失項(xiàng)用零填充),因此需要對這些零元進(jìn)行單獨(dú)處理。通過引入加權(quán)的矩陣分解方法可以在計(jì)算損失函數(shù)時(shí)剔除掉這些零元,而只考慮非零項(xiàng)。修改后的矩陣分解模型可以表示成以下形式:

其中,Γ表示 Hadamard積(Hadamard積定義為: (W☉R)i,j=Wi,jRi,j)。為避免過度擬合,在模型式(3)加入正則項(xiàng),則進(jìn)一步修改后的矩陣分解模型如下所示:

對于模型式(4),pu和pv為正則項(xiàng)的控制參數(shù),用來協(xié)調(diào)實(shí)際用戶評分矩陣和矩陣分解模型學(xué)習(xí)后的填充矩陣之間的訓(xùn)練集誤差。

2.3 應(yīng)用遷移學(xué)習(xí)的協(xié)同過濾推薦算法

遷移學(xué)習(xí)(又稱多任務(wù)學(xué)習(xí))是一種新的機(jī)器學(xué)習(xí)框架,它不同于傳統(tǒng)的監(jiān)督學(xué)習(xí)、非監(jiān)督學(xué)習(xí)和半監(jiān)督學(xué)習(xí)。為了描述這種方法,本文引用文獻(xiàn)[9]給出的關(guān)于遷移學(xué)習(xí)的一般性定義:給定一個(gè)輔助領(lǐng)域DS及它的學(xué)習(xí)任務(wù)TS和一個(gè)目標(biāo)領(lǐng)域DT及其學(xué)習(xí)任務(wù)TT,遷移學(xué)習(xí)的目標(biāo)是利用DS和TS的知識來幫助提高DT的目標(biāo)預(yù)測函數(shù)fT(·),其中DS≠DT,或者TS≠TT。

根據(jù)不同的遷移知識,遷移學(xué)習(xí)方法可以分成3種:基于模型的遷移,基于特征的遷移,基于樣本的遷移。目前,研究者針對不同的遷移學(xué)習(xí)方法提出了各種相應(yīng)的協(xié)同過濾推薦算法,其中基于模型的遷移方法有:評分矩陣生成模型(Rating Matrix Generative Model,RMG-M)[10]等;基于特征的遷移方法有聯(lián)合矩陣分解(Collective Matrix Factorization,CMF)[11]、坐標(biāo)系統(tǒng)遷移(Coordinate System Transfer,CST)[12]等;基于樣本的遷移方法有:集成分解遷移(Transfer by Integrative Factorization,TIF)[13]。從本質(zhì)上看,本文提出的TUF模型屬于基于特征的遷移方法。

3 基于用戶特征遷移的協(xié)同過濾推薦模型

在現(xiàn)實(shí)世界中,不同的電子商務(wù)網(wǎng)站擁有共同的用戶群,而這些用戶在不同的電子商務(wù)網(wǎng)站上對商品的評價(jià)有相似的評價(jià)模式,例如用戶X在某電影網(wǎng)站對科幻類的電影有較高的評價(jià),則該用戶在某書籍網(wǎng)站上對與科幻相關(guān)的書籍應(yīng)該也有較高的評價(jià)。通過挖掘較成熟領(lǐng)域(輔助領(lǐng)域)用戶的潛在評價(jià)模式可以用來幫助新領(lǐng)域(目標(biāo)領(lǐng)域)的相關(guān)學(xué)習(xí)任務(wù)。為此,本文提出了一種基于用戶特征遷移的協(xié)同過濾推薦模型,基本流程如圖1所示。

圖1 TUF模型的基本流程

為了將遷移學(xué)習(xí)應(yīng)用到協(xié)同過濾算法中,首先需要考慮遷移什么樣的知識,即如何從輔助領(lǐng)域中學(xué)習(xí)用戶特征UL。本文將采用核范數(shù)正則化用于從輔助領(lǐng)域的評分矩陣中學(xué)習(xí)用戶的特征。記RA為輔助領(lǐng)域的用戶-項(xiàng)目評分矩陣,WA為輔助領(lǐng)域的標(biāo)記矩陣,則核范數(shù)正則化最小二乘模型的構(gòu)造如下:

其中,||Z||?表示Z的核范數(shù),即矩陣Z的所有奇異值之和。文獻(xiàn)[14]給出了模型(5)的求解算法SOFT-IMPUTE。令Z=USVT為矩陣Z的奇異值分解,Sd為Z的前d個(gè)最大奇異值構(gòu)造的奇異值矩陣,Ud為前d個(gè)最大奇異值對應(yīng)的左奇異向量。顯然,表示了從輔助領(lǐng)域的評分矩陣中提取的d個(gè)用戶特征。本文采用模型(5)提取輔助領(lǐng)域的用戶特征基于下面3個(gè)原因:

(1)模型(5)通過多次的SVD分解進(jìn)行迭代求解,能更準(zhǔn)確地提取稀疏矩陣的特征信息;

(2)由于每一層迭代的SVD分解只需要計(jì)算前k個(gè)最大的奇異值(k＜＜m,n),模型(5)的求解算法可以應(yīng)用于大規(guī)模數(shù)據(jù)處理。這將有利于有較豐富數(shù)據(jù)的輔助領(lǐng)域(例如,項(xiàng)目數(shù)比較多)的任務(wù)學(xué)習(xí);

(3)模型(5)不涉及到矩陣的維度,可以減少用戶的參數(shù)設(shè)置。

下一步將決定如何遷移知識,即如何將從輔助領(lǐng)域中學(xué)習(xí)到的用戶特征UL,用于幫助學(xué)習(xí)目標(biāo)領(lǐng)域的用戶特征U,進(jìn)一步的再用于幫助目標(biāo)領(lǐng)域的用戶-項(xiàng)目評分預(yù)測。從理論上看,當(dāng)輔助領(lǐng)域和目標(biāo)領(lǐng)域的用戶、項(xiàng)目完全一致時(shí),2個(gè)領(lǐng)域的用戶特征應(yīng)具有一致性,即UL=U。然而在實(shí)際應(yīng)用中,由于輔助領(lǐng)域和目標(biāo)領(lǐng)域的差異,2個(gè)領(lǐng)域的用戶特征雖然相似,卻并不會完全相同。因此,本文通過引入正則項(xiàng)來確保UL和U的相似性。將正則項(xiàng)引入模型(3),并引入正則項(xiàng)來避免過擬合,本文構(gòu)造基于用戶特征遷移的協(xié)同過濾推薦模型如下:

其中,pu和pv是控制參數(shù)。pu越大時(shí),表示目標(biāo)領(lǐng)域的用戶特征矩陣U越接近于輔助領(lǐng)域的用戶特征矩陣UL。顯然,當(dāng)pu趨近于無窮大時(shí),U將完全等同于UL;當(dāng)pu趨近于0時(shí),模型對于目標(biāo)領(lǐng)域的學(xué)習(xí)任務(wù)將沒有利用輔助領(lǐng)域的知識。

與文獻(xiàn)[12]提出的CST模型相比較,本文的TUF模型沒有對用戶特征矩陣和項(xiàng)目特征進(jìn)行正交性約束,其主要原因有:(1)用戶特征或項(xiàng)目特征不一定表現(xiàn)出正交性;(2)從數(shù)學(xué)角度看,2個(gè)正交矩陣之間的距離表現(xiàn)為其張成的子空間距離,因而用F范數(shù)來度量2個(gè)正交矩陣的距離并不合理。由于沒有正交性的約束,CST模型的求解方法將不適用于TUF模型,因此在本文第4節(jié)中將給出TUF模型的具體求解算法。

4 基于Wiberg算法的TUF模型求解

本文提出的TUF模型核心在于求解目標(biāo)領(lǐng)域的用戶特征矩陣U和項(xiàng)目特征矩陣V,使模型達(dá)到或逼近最優(yōu)解。文獻(xiàn)[15]提出一種數(shù)值算法來解決帶缺失值的矩陣分解模型。此后,研究者提出了多種迭代算法來解決這類最優(yōu)化問題,例如文獻(xiàn)[8]提出一種Damped-Newton算法來求解式(4)的矩陣分解模型。在這些算法中,Wiberg算法[16]由于對初始值不敏感,并能以快速的迭代速度在全局范圍內(nèi)到達(dá)收斂,具有較好的數(shù)值表現(xiàn)效果。近年來,文獻(xiàn)[17]進(jìn)一步對Wiberg算法進(jìn)行研究。然而,由于文獻(xiàn)[17]提出的算法主要在于解決沒有帶正則項(xiàng)的式(3)缺失矩陣分解模型,并不適合本文提出的TUF模型。因此,本文將重新討論TUF模型求解方法的具體方案。根據(jù)文獻(xiàn)[17]利用Wiberg算法求解缺失矩陣分解模型的基本思想,本文首先對TUF模型進(jìn)行適當(dāng)?shù)淖冃汀?/p>

令u,ul∈Cmd,v,ul∈Cnd分別是由矩陣U,Ul∈Cm×d,Vl∈Cn×d的各行向量進(jìn)行向量化得到的新向量,如,v=vec(V)=記s為目標(biāo)領(lǐng)域用戶-項(xiàng)目評分矩陣R所有已評分項(xiàng)的數(shù)目,r=[ri,j]∈Cs是一個(gè)只包含已評分項(xiàng)的向量,其中,ri,j表示R的第i行第j列元素。為了消去模型(5)中標(biāo)記矩陣W,構(gòu)造一個(gè)由v1,v2,…,vn構(gòu)成的s×md的矩陣P和一個(gè)由u1,u2,…,um構(gòu)成的s×nd的矩陣Q,即:

則模型(6)可以重新構(gòu)造為:

注意到上面的形式只是為了描述方便而表示的矩陣基本結(jié)構(gòu),由于要處理缺失的項(xiàng),矩陣P和Q將只保留與已知項(xiàng)相對應(yīng)的行,例如,如果ri,j是缺失值,則要移除P和Q的((i-1)×n+j)行。因此,P和Q的行向量數(shù)等于已知評分項(xiàng)的總個(gè)數(shù)s。

4.1 Wiberg算法的公式推導(dǎo)

為了求解模型(7)的最優(yōu)解,傳統(tǒng)的 Gauss-Newton方法定義向量x=[uT,vT]T,并通過尋找dΓ/dx=0迭代更新解向量x。在每次迭代中, Gauss-Newton方法對變量u,v進(jìn)行同時(shí)更新。與傳統(tǒng)的Gauss-Newton方法不同,Wiberg算法分別對變量u,v進(jìn)行更新[16]。對給定v,可以通過計(jì)算 ?Γ/?u=(PTP+puI)u-(PTr+puul)=0對u進(jìn)行更新,即有:

在對v進(jìn)行更新時(shí),Wiberg算法將參數(shù)u看作是v的函數(shù),即u=(v),則TUF模型中的最優(yōu)化問題便可以轉(zhuǎn)化成只有一個(gè)參數(shù)變量v的損失函數(shù)φ(v):

其中,f≡f(v)≡g((v),v)=(v)-r;gu≡gu(v)=(v)-ul;gv≡gv(v)=v。記v+φv為更新后的v變量,對φ(v)二階泰勒展開,可得:

構(gòu)造φv極小化上式中的泰勒展開式,即可獲得v的更新量

以及:

進(jìn)一步,分別對f(v)和gu(v)進(jìn)行求導(dǎo)。因?yàn)?由復(fù)合函數(shù)求導(dǎo)法則:

整理上式,即有:

將式(13)代入式(12),可以得到:

將式(14)分別代入式(10)和式(11),即可獲得Hessian矩陣以及

4.2 算法描述

根據(jù)上節(jié)的相關(guān)公式推導(dǎo),對于TUF模型的求解方法,有具體如下的Wiberg算法:

算法Wiberg算法

輸入輔助領(lǐng)域的評分矩陣RA和目標(biāo)領(lǐng)域的評分矩陣R,正則項(xiàng)參數(shù)pu和pv。

輸出目標(biāo)領(lǐng)域的預(yù)測評分矩陣Z。

Step1通過SOFT-IMPUTE算法對輔助領(lǐng)域的最優(yōu)化模(5)進(jìn)行填充,得到RA的近似低秩逼近矩陣ZA。

Step2對ZA進(jìn)行奇異值分解,即有ZA=USVT,得到輔助領(lǐng)域的用戶特征

Step3隨機(jī)初始化v,并通過UL和R構(gòu)造出ul和r。

Step4由v構(gòu)造P矩陣,并根據(jù)式(8)對u進(jìn)行更新。

Step5如果收斂則跳轉(zhuǎn)至 Step8,否則轉(zhuǎn)至Step6。

Step6由u構(gòu)造Q矩陣,計(jì)算和,并根據(jù)式(9)式求解Δv,更新v←v+Δv。

Step7如果收斂則跳轉(zhuǎn)至Step8,否則跳轉(zhuǎn)至Step4。

Step8通過u和v構(gòu)造出矩陣U和V,并計(jì)算Z=UVT。

在Wiberg算法中,計(jì)算量主要集中在式(8)和式(9)。在實(shí)際應(yīng)用中,為了減小時(shí)間和空間的復(fù)雜度,算法并不需要直接生成矩陣的逆,而是通過求解線性方程組以獲取u和v更新向量。例如,式(8)、式(9)等價(jià)于求解線性方程組:

5 實(shí)驗(yàn)與結(jié)果分析

5.1 實(shí)驗(yàn)數(shù)據(jù)集

為了驗(yàn)證本文模型的有效性,采用互聯(lián)網(wǎng)上2個(gè)公開的數(shù)據(jù)集對算法進(jìn)行測試和驗(yàn)證。

(1)MovieLens數(shù)據(jù)集

數(shù)據(jù)來源:http://www.grouplens.org/node/ 73。該數(shù)據(jù)集中的評分?jǐn)?shù)據(jù)包含943個(gè)獨(dú)立用戶對1682部電影(共包含19類電影)進(jìn)行10萬次評分的數(shù)據(jù)(評分1～5)。隨機(jī)選擇500個(gè)用戶對所有電影的評分?jǐn)?shù)據(jù),要求每個(gè)用戶至少評價(jià)過30部以上的電影。在該數(shù)據(jù)集上進(jìn)行5組實(shí)驗(yàn),每組實(shí)驗(yàn)選取一種電影類別的評分?jǐn)?shù)據(jù)作為目標(biāo)領(lǐng)域的數(shù)據(jù)集,而其余電影類別的評分?jǐn)?shù)據(jù)則作為輔助領(lǐng)域的數(shù)據(jù)集。對目標(biāo)領(lǐng)域的數(shù)據(jù)集,按照4∶1的比例將其進(jìn)一步劃分為訓(xùn)練集和測試集。數(shù)據(jù)集的具體描述見表1。這里稀疏度定義為已評分?jǐn)?shù)據(jù)占整個(gè)領(lǐng)域數(shù)據(jù)集的比例。

表1 MovieLens評分?jǐn)?shù)據(jù)集描述

(2)EachMovie數(shù)據(jù)集

數(shù)據(jù)來源:http://research.compaq.com/SRC/ eachmovie。該數(shù)據(jù)集約有7.2萬個(gè)用戶對1 628部電影進(jìn)行2.8百萬次評分的數(shù)據(jù)(評分1～6),為了評分?jǐn)?shù)據(jù)的統(tǒng)一,實(shí)驗(yàn)時(shí)將原來的評分6由評分5來代替。在該數(shù)據(jù)集上隨機(jī)抽取500個(gè)用戶,并對電影項(xiàng)目進(jìn)行隨機(jī)劃分:1 000部電影作為輔助領(lǐng)域的項(xiàng)目,500部電影作為目標(biāo)領(lǐng)域的項(xiàng)目,每個(gè)用戶在輔助領(lǐng)域和目標(biāo)領(lǐng)域里都至少評價(jià)過25部以上的電影。進(jìn)一步地,劃分目標(biāo)領(lǐng)域的訓(xùn)練集和測試集,其中訓(xùn)練集根據(jù)目標(biāo)領(lǐng)域的用戶評價(jià)數(shù)劃分為4組,每組用戶的評價(jià)數(shù)依次為10,15,20,25,對應(yīng)的其余評分?jǐn)?shù)據(jù)作為測試集。

5.2 評價(jià)指標(biāo)

本文采用平均絕對誤差(Mean Absolute Error, MAE)作為評價(jià)評分預(yù)測準(zhǔn)確性的標(biāo)準(zhǔn)。平均絕對誤差通過計(jì)算預(yù)測的用戶評分和實(shí)際的用戶評分之間的偏差來度量預(yù)測的準(zhǔn)確性,具體計(jì)算方法如下:

其中,N為測試集的用戶評分?jǐn)?shù);pi,j表示用戶i對項(xiàng)目j的預(yù)測評分值;ri,j表示實(shí)際評分值。MAE越小,推薦質(zhì)量越高。每組實(shí)驗(yàn)均進(jìn)行5次隨機(jī)實(shí)驗(yàn)(即對訓(xùn)練集進(jìn)行隨機(jī)劃分),并取平均值作為最終的評價(jià)結(jié)果。

5.3 對比算法和參數(shù)設(shè)置

為了驗(yàn)證TUF模型能否有效提高推薦系統(tǒng)的推薦質(zhì)量,選擇3種非遷移學(xué)習(xí)算法:PCC、Soft-Impute(SD)和TMF,2種遷移學(xué)習(xí)算法:CMF和Soft-Impute(MD)作為比較方法。其中,PCC是基于Pearson相關(guān)相似性(Pearson Corelation Coefficients, PCC)的最近鄰協(xié)同過濾算法[19];Soft-Impute(SD)是單個(gè)領(lǐng)域的 Soft-Impute算法(Soft-Impute on Single Domain),即只對目標(biāo)領(lǐng)域評分矩陣進(jìn)行預(yù)測填充缺失項(xiàng);TMF是傳統(tǒng)的矩陣分解方法(Traditional Matrix Factorization)[8],即上文的模型(4);CMF為文獻(xiàn)[11]提出的基于多領(lǐng)域數(shù)據(jù)的聯(lián)合矩陣分解模型;Soft-Impute(MD)為多個(gè)領(lǐng)域的Soft-Impute算法(Soft-Impute on Multiple Domains),即對目標(biāo)領(lǐng)域和輔助領(lǐng)域構(gòu)成的評分矩陣進(jìn)行預(yù)測填充缺失項(xiàng)。而其他的一些遷移學(xué)習(xí)算法,如CST模型[12]同時(shí)應(yīng)用到了用戶特征和項(xiàng)目特征,TCF模型[20]對輔助領(lǐng)域和目標(biāo)領(lǐng)域要求有共同的用戶集和項(xiàng)目集。本文實(shí)驗(yàn)的數(shù)據(jù)集并不符合這些要求,因此對于這些算法本文不進(jìn)行實(shí)驗(yàn)比較。

在參數(shù)設(shè)置上,對有利用到用戶特征數(shù)的模型(TMF,CMF和TUF)本文嘗試選擇不同的用戶特征數(shù):d∈{3,4,…,10}。在PCC算法中,選擇的最近鄰居數(shù)為{5～120}。對于TMF模型,正則項(xiàng)的參數(shù)設(shè)置如下:pu=pv={0.1,0.5,1,5,10}。對于CMF模型,選擇’Identity’作為目標(biāo)領(lǐng)域和輔助領(lǐng)域的預(yù)測函數(shù)(prediction link),其他參數(shù)的設(shè)置如下:α=0.5,pu=pz={0.1,0.5,1,5,10},pv={0.1,0.5,1, 5,10,20}。在2種不同領(lǐng)域的Soft-Impute算法中,對參數(shù)λ的設(shè)置范圍為{1,2,…,20}。對于本文的TUF模型,正則項(xiàng)的參數(shù)設(shè)置如下:pu={1,5,10, 50},pv={1,2,5,10},并且pu≥pv。

5.4 實(shí)驗(yàn)結(jié)果

每個(gè)算法的實(shí)驗(yàn)結(jié)果為在上一小節(jié)的參數(shù)設(shè)定范圍內(nèi)取得的最優(yōu)效果。表 2和表 3分別為MovieLens和EachMovie目標(biāo)領(lǐng)域測試集上的實(shí)驗(yàn)結(jié)果。

表2 6種算法在不同電影類別下的MAE指標(biāo)比較

表3 6種算法在不同數(shù)據(jù)稀疏度下的MAE指標(biāo)比較

從表2和表3可以看出,幾乎所有遷移學(xué)習(xí)方法的結(jié)果均好于非遷移學(xué)習(xí)方法的結(jié)果。這種對比在Soft-Impute(SD)和Soft-Impute(MD),TMF和TUF上體現(xiàn)的尤為明顯。值得一提的是,Soft-Impute(SD)和TMF可以認(rèn)為是Soft-Impute(MD)和TUF在無遷移學(xué)習(xí)下的特例。這表明了遷移學(xué)習(xí)方法利用了輔助領(lǐng)域的信息,能有效提高目標(biāo)領(lǐng)域的推薦質(zhì)量。而在所有的遷移學(xué)習(xí)方法中,TUF模型均取得了最好的推薦結(jié)果。這表明了和CMF, Soft-Impute(MD)方法相比,TUF能更為有效的利用輔助領(lǐng)域的用戶特征信息。

此外,表3的結(jié)果還表明,稀疏度越低,TUF取得的優(yōu)勢越明顯。例如在表3的結(jié)果中,與CMF模型相比較,稀疏度為5%的測試結(jié)果MAE值減少了0.005,而稀疏度為2%的測試結(jié)果MAE值減少了0.037。這說明對評分?jǐn)?shù)據(jù)極其稀疏的情形,TUF模型體現(xiàn)了更好的適應(yīng)性,能有效緩解數(shù)據(jù)稀疏的問題。

6 結(jié)束語

本文在矩陣分解和遷移學(xué)習(xí)方法的基礎(chǔ)上,提出了一種用戶特征遷移的協(xié)同過濾推薦模型(TUF),以緩解數(shù)據(jù)稀疏的問題。為提取輔助領(lǐng)域的用戶特征信息,本文并沒有簡單地利用SVD方法進(jìn)行獲取,而是先通過Soft-Impute算法對輔助領(lǐng)域的缺失評分矩陣進(jìn)行填充,然后對其奇異值分解獲取更為準(zhǔn)確的用戶特征。通過TUF模型對輔助領(lǐng)域的用戶特征進(jìn)行遷移,幫助目標(biāo)領(lǐng)域的用戶對未評分項(xiàng)目的預(yù)測。此外,本文采用能夠快速達(dá)到收斂的Wiberg算法對模型進(jìn)行迭代求解以獲得最優(yōu)解。實(shí)驗(yàn)結(jié)果表明,引入特征遷移的矩陣分解模型相比較于傳統(tǒng)的矩陣分解模型和其他遷移學(xué)習(xí)算法,能有效緩解評分矩陣數(shù)據(jù)極端稀疏情況,顯著提高推薦系統(tǒng)的推薦質(zhì)量。

本文模型雖然只利用輔助領(lǐng)域的用戶特征信息,但也適用于輔助領(lǐng)域和目標(biāo)領(lǐng)域有共同項(xiàng)目集時(shí)對項(xiàng)目特征的遷移,而如何改進(jìn)模型使其能夠同時(shí)對用戶特征和項(xiàng)目特征進(jìn)行遷移是下一步的研究方向。

[1] 許海玲,吳 瀟,李曉東,等.互聯(lián)網(wǎng)推薦系統(tǒng)比較研究[J].軟件學(xué)報(bào),2009,20(2):350-362.

[2] Billsus D,Pazzani M J. Learning Collaborative Information Filters[C]//Proceedings of ICML’98. Madison,USA:[s.n.],1998:54-48.

[3] 吳 湖,王永吉,王 哲,等.兩階段聯(lián)合聚類協(xié)同過濾算法[J].軟件學(xué)報(bào),2010,21(5):1042-1054.

[4] 程小林,熊 焰,劉青文,等.一種基于自適應(yīng)局部融合參數(shù)的協(xié)同過濾方法[J].計(jì)算機(jī)工程,2014, 40(1):39-44.

[5] 曾小波,魏祖寬,金在弘.協(xié)同過濾系統(tǒng)的矩陣稀疏性問題的研究[J].計(jì)算機(jī)應(yīng)用,2010,30(4):1079-1082.

[6] Li Bin.Cross-domain Collaborative Filtering:A Brief Survey[C]//Proceedings of the 23rd International Conference on Tools with Artificial Intelligence. [S.l.]:IEEE Press,2011:1085-1086.

[7] Ning X,Karypis G.Multi-task Learning for Recommender System[C]//Proceedings of the 2nd Asian Conference on Machine Learning.Tokyo,Japan:[s.n.],2010:269-284.

[8] Buchanan A M,Fitzgibbon A W.Damped Newton Algorithms for Matrix Factorization with Missing Data[C]// Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.[S.l.]:IEEE Press,2005:316-322.

[9] Sinno J P,Yang Qiang.A Survey on Transfer Learning[J].IEEE Transactions on Knowledge and Data Engineering,2010,22(10):1345-1359.

[10] Li Bin,Yang Qiang,Xue Xiangyang.Transfer Learning for Collaborative Filtering via a Rating-matrix Generative Model[C]//Proceedings of the 26th Annual International Conference on Machine Learning.Quebec, Canada:[s.n.],2009:617-624.

[11] Singh A P,Gordon G J.RelationalLearning Via Collective Matrix Factorization[C]//Proceedings of the 14th ACM SIGKDD InternationalConference on Knowledge Discovery and Data Mining.[S.l.]:ACM Press,2008:650-658.

[12] Pan Weike,Evan W X,Lin N,et al.Transfer Learning in Collaborative Filtering for Sparsity Reduction[C]// Proceedings of the 24th AAAI Conference on Artificial Intelligence.[S.l.]:AAAI Press,2010:230-235.

[13] Pan Weike,Evan W X,Yang Qiang.Transfer Learning in Collaborative Filtering with Uncertain Ratings[C]// Proceedings of the 26th AAAI Conference on Artificial Intelligence.Toronto,Canada:AAAIPress,2012: 662-668.

[14] Mazumder R,Hastie T,Tibshirani R.Spectral Regularization Algorithms for Learning Large Incomplete Matrices[J]. Journal of Machine Learning Research,2010,(11):2287-2322.

[15] Golub G H,van Loan C F.Matrix Computations[M]. Baltimore,USA:Johns Hopkins University Press,1996.

[16] Wiberg T.Computation of Principal Components When Data are Missing[C]//Proceedings of the 2nd Symposium on Computational Statistics.Berlin,Germany: [s.n.],1976:229-236.

[17] Okatani T,Deguchi K.On the Wiberg Algorithm for Matrix Factorization in the Presence of Missing Components[J].International Journal of Computer Vision, 2007,72(3):329-337.

[18] 李 改,李 磊.基于矩陣分解的協(xié)同過濾算法[J].計(jì)算機(jī)工程與應(yīng)用,2011,47(30):4-7.

[19] Resnick P,Iacovou N,Suchak M,et al.GroupLens:An Open Architecture for Collaborative Filtering of Netnews[C]//Proceedings of ACM Conference on Computer Supported Cooperative Work.North Carolina,USA: ACM Press,1994:175-186.

[20] Pan W,Liu N N,Xiang E W,et al.Transfer Learning to Predict Missing Ratings via Heterogeneous User Feedbacks[C]//Proceedings of the 22th International Joint Conference on Artificial Intelligence.[S.l.]:AAAI Press,2011:2318-2323.

編輯 金胡考

Collaborative Filtering Recommendation Based on User Feature Transfer

KE Liangwen,WANG Jing
(School of Computer Science and Technology,Huaqiao University,Xiamen 361021,China)

In order to improve the recommendation quality of recommender system with data sparsity,this paper proposes a user collaborative filtering recommendation model based on feature transfer.Firstly,matrix factorization technology is applied to collect the user feature from the auxiliary domain.Secondly,it constructs a matrix factorization model with the constraint of regularization term,which is used to transfer the user feature learned from the auxiliary domain to the target domain,so as to help the learning of user feature in the target domain.Finally,user recommendation is made for the target domain.A fast convergence Wiberg algorithm is also designed for the model to get the optimal solution,whose feasibility is also discussed for practical application.Through the experiment on two real world data sets, the model can effectively transfer the user feature of source domain,and improve the quality of recommender system for target domain.

data sparsity;user feature transfer;collaborative filtering;matrix factorization;Wiberg algorithm

1000-3428(2015)01-0037-07

A

TP311

10.3969/j.issn.1000-3428.2015.01.007

國家自然科學(xué)基金資助項(xiàng)目(61370006);福建省高等學(xué)校杰出青年科研人才培育計(jì)劃基金資助項(xiàng)目(11FJPY01);福建省高等學(xué)校新世紀(jì)優(yōu)秀人才支持計(jì)劃基金資助項(xiàng)目(2012-FJ-NCET-ZR01)。

柯良文(1988-),男,碩士研究生,主研方向:數(shù)據(jù)挖掘,個(gè)性化推薦算法;王 靖,副教授、博士。

2014-02-19

2014-03-15 E-mail:lwke1213@163.com

中文引用格式:柯良文,王 靖.基于用戶特征遷移的協(xié)同過濾推薦[J].計(jì)算機(jī)工程,2015,41(1):37-43.

英文引用格式:Ke Liangwen,Wang Jing.Collaborative Filtering Recommendation Based on User Feature Transfer[J]. Computer Engineering,2015,41(1):37-43.