Read_me

This is a pytorch implementation of submission: Multi-Relational Graph Attention Network for Social Relationship Inference from Human Mobility Data

IJCAI'24

abstract

Inferring social relationships from human mobility data holds significant value in real-life spatio-temporal applications, inspiring the development of a series of graph-based methods for deriving such relationships. However, despite their noted effectiveness, we argue that previous methods either rely solely on direct relations between users, neglecting valuable user mobility patterns, or have not fully harnessed the indirect interactions, thereby struggling to capture users' mobility preferences. To address these issues, in this work, we propose the Multi-Relational Graph Attention Network MRGAN, a novel graph attention network, which is able to explicitly model indirect relations and effectively capture their different impact. Specifically, we first extract a multi-relational graph from heterogeneous mobility graph to explicitly model the direct and indirect relations, %as different mobility patterns and then utilize influence attention and cross-relation attention to further capture the different influence between users, and different importance of relations for each user. Comprehensive experiments on three real-world mobile datasets demonstrate that the proposed model significantly outperforms state-of-the-art models in predicting social relationships between users.

Requirements

• Python 3.8.6
• torch 2.1.2
• pandas 2.3.1
• keras 2.11.0
• networkx
• numpy
• scipy
• scikit-learn
• dgl 1.1.2

Run the code

Python train_MRGAN.py to run the code

Contact

Please email [email protected] or [email protected] if you have any questions