Graph neural networks-based preference learning method for object ranking
Zhenhua Meng,Rongheng Lin,Budan Wu
DOI: https://doi.org/10.1016/j.ijar.2024.109131
IF: 4.452
2024-01-28
International Journal of Approximate Reasoning
Abstract:Preference learning refers to the task of predicting the ranking of a collection of alternatives based on observed or revealed preference information. Object ranking is a critical problem within the domain of preference learning, which can be described as learning a ranking function based on training data in a ranked form. Some of the existing preference learning methods are parametric in nature, and such methods are faster to train but less expressive. To deal with the above issue, we introduce the concept of graph neural networks (GNNs) into preference learning, and propose a GNNs-based preference learning method. Our method is composed of three stages to cope with preference learning, i.e. preference relation graph construction, preference relation prediction, and object preference ranking. In the first stage, we map preference information onto the graph structure, and construct a directed graph with objects as nodes and preference relations between objects as edges. In the second stage, we formulate relation prediction as an edge classification problem on the graph, design a multi-layer perceptron (MLP) model to extract edge features, and mine preference information with the help of GNNs. In the third stage, we construct a comparator neural network structure, which takes pairwise preference information as input and the score of each object as output, and the ranking of object scores is the objects' preference order. Experiments on preference learning datasets demonstrate that, compared to existing parametric preference learning methods, our method can achieve marked performance gains in terms of object ranking scenario.
computer science, artificial intelligence