Listwise approach based on the cross-correntropy for learning to rank

  • Mintao Wu
  • , Jihua Zhu
  • , Jun Wang
  • , Shanmin Pang
  • , Yaochen Li

Research output: Contribution to journalArticlepeer-review

2 Scopus citations

Abstract

The problem of learning to rank is addressed and a novel listwise approach by taking document retrieval as an example is proposed. It first introduces the concept of cross-correntropy into learning to rank and then proposes the listwise loss function based on the cross-correntropy between the ranking list given by the label and the one predicted by training model. The use of the cross-correntropy loss leads to the development of the listwise approach called ListCCE, which employs the gradient descent algorithm to train a neural network model. Experimental results tested on publicly available data sets show that the proposed approach performs better than some existing approaches.

Original languageEnglish
Pages (from-to)878-880
Number of pages3
JournalElectronics Letters
Volume54
Issue number14
DOIs
StatePublished - 12 Jul 2018

Fingerprint

Dive into the research topics of 'Listwise approach based on the cross-correntropy for learning to rank'. Together they form a unique fingerprint.

Cite this