Arsm Gradient Estimator for Supervised Learning to Rank
Arsm Gradient Estimator for Supervised Learning to Rank
复制标题
DOI:
10.1109/icassp40776.2020.9053127
复制
发表时间:
2019-11
期刊:
影响因子:
--
通讯作者:
Siamak Zamani Dadaneh;Shahin Boluki;Mingyuan Zhou;Xiaoning Qian
中科院分区:
文献类型:
--
作者:
Siamak Zamani Dadaneh;Shahin Boluki;Mingyuan Zhou;Xiaoning Qian
We propose a new model for supervised learning to rank. In our model, the relevance labels are assumed to follow a categorical distribution whose probabilities are constructed based on a scoring function. We optimize the training objective with respect to the multivariate categorical variables with an unbiased and low-variance gradient estimator. Learning-to-rank methods can generally be categorized into pointwise, pairwise, and listwise approaches. Although our scoring function is pointwise, the proposed framework permits flexibility over the choice of the loss function. In our new model, the loss function need not be differentiable and can either be pointwise or listwise. Our proposed method achieves better or comparable results on two datasets compared with existing pairwise and listwise methods.