HAMLET: Hierarchical Attention-based Model with Multi-Task Self-Training for User Profiling

Fuxin Ren,Zhongbao Zhang,Yang Yan,Zhi Wang,Sen Su,Philip S. Yu
DOI: https://doi.org/10.1109/bigdata52589.2021.9671313
2021-01-01
Abstract:User profiling is playing an increasingly important role in real-world applications. Previous works have shown that integrating user information from multiple social networks helps to significantly improve the performance of user profiling. However, these studies either ignore the different contributions of various features in different profiling tasks or need to train one model for each task. What’s more, the assumption of the strong relatedness between user profiling tasks limits their application. These phenomena make inferring comprehensive user attributes still an open problem. In this paper, we propose a novel method, called Hierarchical Attention-based Model with sparse-sharing-based muLti-task sElf-Training algorithm (HAMLET), for comprehensive user profiling. More specifically, we first employ a hierarchical attention-based network as our base network to represent users. It assigns various features from different social networks with different weights for different users during the fusing procedure. Then, we propose a multi-task self-training algorithm that takes advantage of both task correlations and self-training to obtain better performance. We conduct extensive experiments on two real-world datasets and verify the superiority of HAMLET for user profiling.
What problem does this paper attempt to address?