arrow
Return

Graph neural networks-based preference learning method for object ranking

delete2024-04-01
delete0
PRE
AI
Z
Zhenhua Meng
R
Rongheng Lin *
B
Budan Wu
DOI:10.1016/j.ijar.2024.109131delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
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 existing parametric preference learning methods are difficult to balance the relationship between expressive power and training cost. To deal with this 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 model consisting of multi -layer perceptron (MLP) and GNNs to extract edge features. In the third stage, we build 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 baselines, our method can achieve marked performance gains in terms of object ranking scenario.
Keywords:
Preference learning
Graph neural networks
Pairwise preference
Learning to rank
Object ranking

Journal

International Journal of Approximate Reasoning cover
International Journal of Approximate Reasoning
IF:
3
Papers:
2.9K
Citations:
5.1K

Organization

B
beijing university of posts & telecommunications
Scholars:
1.4W
Papers: 1.2W
Citations: 9