Redefining edge representations for enhanced information propagation on GNNs
摘要
Graph Neural Networks (GNNs) have achieved great success in learning node representations from graph-structured data. Most GNNs adopt a message-passing paradigm, where nodes act as information carriers and edges serve as communication channels. However, existing models typically represent edges using scalar weights, which only modulate the overall strength of messages and fail to distinguish the importance of individual feature dimensions. This coarse-grained formulation limits the expressiveness of information propagation. To address this issue, we propose to represent edges as vectors, enabling fine-grained, dimension-specific control over the message-passing process. We refer to these edge vectors as relation representations, as they capture the semantic relations between connected nodes. Building on this idea, we present a novel model, Relation-Enhanced Feature Preference Network (ReFPN), which integrates relation representation learning into the message-passing framework. ReFPN dynamically adjusts the contribution of each feature dimension based on the learned relation representations, thereby enhancing the effectiveness of information aggregation. Extensive experiments on five benchmark datasets demonstrate that ReFPN consistently outperforms strong baselines in both node classification and clustering tasks, validating the superiority of fine-grained relation-aware propagation.