[發(fā)明專利]用于用戶聚類的方法、電子設(shè)備和存儲介質(zhì)

有效

• 申請?zhí)枺?/td>: 202111072920.0
• 申請日: 2021-09-14
• 公開（公告）號: CN113792784B
• 公開（公告）日: 2022-06-21
• 發(fā)明（設(shè)計）人: 顧茂杰;彭飛;唐文斌
• 申請（專利權(quán)）人: 上海任意門科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06F16/901;G06F16/906;G06Q50/00
• 代理公司: 北京市金杜律師事務(wù)所 11256
• 代理人: 陳振

1.一種用于用戶聚類的方法，包括：

獲取與多個用戶相關(guān)聯(lián)的多個用戶標(biāo)識和多項用戶屬性信息以及指示所述多個用戶之間的多種社交行為的行為信息；

基于所述多個用戶標(biāo)識、所述多項用戶屬性信息和所述行為信息，生成有向圖，所述有向圖中的多個節(jié)點表示所述多個用戶，所述有向圖中任意第一節(jié)點到任意第二節(jié)點之間的至少一條有向邊表示所述第一節(jié)點表示的用戶對所述第二節(jié)點表示的用戶存在所述多種社交行為中的至少一種社交行為；

基于所述有向圖，訓(xùn)練圖自編碼器模型，以生成與所述多個用戶相關(guān)聯(lián)的多個嵌入特征表示；以及

基于所述多個嵌入特征表示，經(jīng)由聚類模型，生成所述多個用戶的聚類結(jié)果；

其中訓(xùn)練所述圖自編碼器模型包括：

基于所述有向圖，生成與所述多個節(jié)點之間的節(jié)點對集相關(guān)聯(lián)的領(lǐng)接狀態(tài)集，所述領(lǐng)接狀態(tài)集包括與所述節(jié)點對集中的每一節(jié)點對相關(guān)聯(lián)的領(lǐng)接狀態(tài)，所述領(lǐng)接狀態(tài)包括領(lǐng)接或不領(lǐng)接；

基于所述多個用戶標(biāo)識和所述多個用戶屬性信息，生成與所述多個節(jié)點相關(guān)聯(lián)的多個特征表示；

S1：基于與所述多個節(jié)點所對應(yīng)的多個出邊鄰居集相關(guān)聯(lián)的多個特征表示集和與所述節(jié)點對集相關(guān)聯(lián)的注意力權(quán)重集，生成與所述多個節(jié)點相關(guān)聯(lián)的多個嵌入特征表示；

S2：基于與所述節(jié)點對集相關(guān)聯(lián)的嵌入特征表示對集之間的內(nèi)積集，生成與所述節(jié)點對集相關(guān)聯(lián)的預(yù)測領(lǐng)接概率集；

S3：基于所述領(lǐng)接狀態(tài)集、所述預(yù)測領(lǐng)接概率集和預(yù)定損失函數(shù)，對所述注意力權(quán)重集進(jìn)行更新；以及

如果確定與所述注意力權(quán)重集相關(guān)聯(lián)的更新值集不滿足預(yù)設(shè)收斂條件，則重復(fù)步驟S1-S3。

2.根據(jù)權(quán)利要求1所述的方法，其中所述注意力權(quán)重集包括與所述多種社交行為相關(guān)聯(lián)的多個注意力權(quán)重子集，以及生成與所述多個節(jié)點相關(guān)聯(lián)的多個嵌入特征表示包括對于所述多個節(jié)點中的每個節(jié)點執(zhí)行以下步驟：

從所述節(jié)點所對應(yīng)的出邊鄰居集中確定與所述多種社交行為相關(guān)聯(lián)的多個出邊鄰居子集；

基于所述多個注意力權(quán)重子集和與所述多個出邊鄰居子集相關(guān)聯(lián)的多個特征表示子集，生成與所述節(jié)點相關(guān)聯(lián)的多個中間特征表示，所述多個中間特征表示與所述多種社交行為相對應(yīng)；以及

基于所述多個中間特征表示，生成與所述節(jié)點相關(guān)聯(lián)的嵌入特征表示。

3.根據(jù)權(quán)利要求2所述的方法，其中生成與所述節(jié)點相關(guān)聯(lián)的多個中間特征表示包括對于所述多個出邊鄰居子集中的每個出邊鄰居子集執(zhí)行以下步驟：

從所述多個注意力權(quán)重子集確定與所述出邊鄰居子集所關(guān)聯(lián)的社交行為的類型相關(guān)聯(lián)的注意力權(quán)重子集；

從所確定的注意力權(quán)重子集確定與所述節(jié)點和所述出邊鄰居子集之間的多個節(jié)點對相關(guān)聯(lián)的多個注意力權(quán)重；以及

基于所確定的多個注意力權(quán)重和與所述出邊鄰居子集相關(guān)聯(lián)的特征表示子集，生成與所述節(jié)點相關(guān)聯(lián)的中間特征表示。

4.根據(jù)權(quán)利要求2所述的方法，其中生成與所述節(jié)點相關(guān)聯(lián)的嵌入特征表示包括：

對所述多個中間特征表示進(jìn)行求和或者拼接，以生成與所述節(jié)點相關(guān)聯(lián)的嵌入特征表示。

5.根據(jù)權(quán)利要求1所述的方法，其中生成所述多個用戶的聚類結(jié)果包括：

基于所述多個嵌入特征表示和所述聚類模型，生成多個聚類中心特征表示；

基于第一預(yù)定分布函數(shù)，生成所述多個嵌入特征表示相對于多個聚類中心特征表示的第一分布概率集；

基于所述多個第一分布概率和第二預(yù)定分布函數(shù)，生成所述多個嵌入特征表示相對于多個聚類中心特征表示的第二分布概率集；以及

基于所述第一分布概率集、所述第二分布概率集和預(yù)定目標(biāo)函數(shù)，通過梯度下降更新所述多個聚類中心特征表示。

6.根據(jù)權(quán)利要求5所述的方法，其中生成所述多個嵌入特征表示相對于多個聚類中心特征表示的第一分布概率集包括對于所述多個嵌入特征表示中的每個嵌入特征表示執(zhí)行以下步驟：

確定所述嵌入特征表示與所述多個聚類中心特征表示之間的多個距離；以及

對于所述多個聚類中心特征表示中的每個聚類中心特征表示，基于所述嵌入特征表示與所述聚類中心特征表示之間的距離、所述多個距離和所述第一預(yù)定分布函數(shù)，生成所述嵌入特征表示相對于所述聚類中心特征表示的第一分布概率。