Listwise learning to rank method combining approximate NDCG ranking indicator with Conditional Generative Adversarial Networks

Jinzhong Li,Huan Zeng,Cunwei Xiao,Chunjuan Ouyang,Hua Liu
DOI: https://doi.org/10.1016/j.patrec.2024.01.015
IF: 4.757
2024-01-28
Pattern Recognition Letters
Abstract:Some previous empirical studies have shown that the performances of the listwise learning to rank approaches are in general better than the pointwise or pairwise learning to rank techniques. The listwise learning to rank methods which directly optimize information retrieval indicators are a type of essential and popular method of learning to rank. However, the existing learning to rank approaches based on Generative Adversarial Networks (GANs) do not utilize a loss function based on information retrieval indicators to optimize the generator and/or discriminator. Thus, an approach of learning to rank that combines approximate Normalized Discounted Cumulative Gain (NDCG) ranking indicators with Conditional Generative Adversarial Networks (CGAN) is proposed in this paper, named NCGAN-LTR. The NCGAN-LTR approach constructs loss functions of the generator and discriminator based on the Plackett-Luce model and an approximate version of the NDCG ranking indicator, which is utilized to train the network parameters of CGAN. The experimental results on four benchmark datasets of learning to rank, i.e., TREC TD2004, OHSUMED, MQ2008, and MSLR-WEB10K demonstrate that our proposed NCGAN-LTR approach has superior performance across almost various ranking indicators of information retrieval compared with the IRGAN-List approach.
computer science, artificial intelligence
What problem does this paper attempt to address?