arXiv · 2002.07651
Listwise Learning to Rank with Deep Q-Networks
Abstract
Learning to Rank is the problem involved with ranking a sequence of documents based on their relevance to a given query. Deep Q-Learning has been shown to be a useful method for training an agent in sequential decision making. In this paper, we show that DeepQRank, our deep q-learning to rank agent, demonstrates performance that can be considered state-of-the-art. Though less computationally efficient than a supervised learning approach such as linear regression, our agent has fewer limitations in terms of which format of data it can use for training and evaluation. We run our algorithm against Microsoft's LETOR listwise dataset and achieve an NDCG@1 (ranking accuracy in the range [0,1]) of 0.5075, narrowly beating out the leading supervised learning model, SVMRank (0.4958).
Explore related subjects
Keep this discovery
Abhishek Sharma. 2020-02-13. Listwise Learning to Rank with Deep Q-Networks. https://arxiv.org/abs/2002.07651
Cite the original work for its findings. Save a collection to share your selection of sources.