计算机科学
骨料(复合)
数据挖掘
异构网络
代表(政治)
维数(图论)
理论计算机科学
人工智能
机器学习
无线网络
政治
电信
数学
复合材料
材料科学
法学
纯数学
无线
政治学
作者
Xiangbo Tian,Liqing Qiu,Jianyi Zhang
标识
DOI:10.1016/j.ins.2021.10.018
摘要
With the development of online social networks, user behavior prediction based on the data collected from these social networks has attracted increasing attention. In heterogeneous social networks, a node usually has several heterogeneous attributes to describe itself from different angles. However, most existing methods only utilize an attribute of each node and neglect other heterogeneous attributes. Therefore, this paper proposes a new user heterogeneous information embedding method, called user heterogeneous information embedding (UHIE). This method utilizes the attention mechanism to aggregate the heterogeneous attribute information of each neighbor to obtain their low-dimension representation. Then, the graph neural network is employed to aggregate the multi-relational information from neighbors to obtain the low-dimension representation of nodes. Furthermore, a new soft thresholding method is proposed to eliminate the unimportant information, called multi-head self-attention soft thresholding (MSST), which employs the multi-head self-attention mechanism to calculate an importance threshold for each feature. Based on UHIE and MSST, a new user behavior prediction model is proposed, called Heterogeneous Residual Self-Attention Shrinkage Network (HRSN). This model utilizes UHIE to aggregate heterogeneous information including all heterogeneous attribute information of nodes, and employs MSST to eliminate unimportant information. The experimental results on three real-world datasets show the superiority of the proposed model.
科研通智能强力驱动
Strongly Powered by AbleSci AI