Improving statistical relational learning with graph embeddings for socio-economic data retrieval

Abstract

Abstract Social media data is useful for personalized search engines, recommender systems, and targeted online marketing. Sometimes values of attributes are missing due to security reasons or problematic data collection process. In this case, the information about connections between vertices become more important since it explicitly allows for using the structure of a social graph for inferring missing attributes. One of the general and effective approaches of inferring missing attributes on graph structures are statistical relational learning. For machine learning tasks, the graph embeddings represent topological properties better, but they are not aimed at the attributes prediction. In this study, we introduce a method combining graph embeddings and the statistical relational learning. We consider different their combinations, as there are possible different hidden connections between social ties and considered attributes. We compare the performance using real data from the social network with different missing attributes and assortative patterns: gender, age, and economic status. As a result, the inclusion of graph embeddings in statistical relational learning improves accuracy and significantly decreases the number of iterations.