arrow
返回

Spectral Ranking Regression

delete2022-07-30
delete0
PRE
AI
J
Jennifer Dy
D
Deniz Erdoğmuş
S
Susan Ostmo
J
J. Peter Campbell
M
Michael F. Chiang
S
Stratis Ioannidis
DOI:10.1145/3530693delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
We study the problem of ranking regression, in which a dataset of rankings is used to learn Plackett-Luce scores as functions of sample features. We propose a novel spectral algorithm to accelerate learning in ranking regression. Our main technical contribution is to show that the Plackett-Luce negative log-likelihood augmented with a proximal penalty has stationary points that satisfy the balance equations of a Markov Chain. This allows us to tackle the ranking regression problem via an efficient spectral algorithm by using the Alternating Directions Method of Multipliers (ADMM). ADMM separates the learning of scores and model parameters, and in turn, enables us to devise fast spectral algorithms for ranking regression via both shallow and deep neural network (DNN) models. For shallow models, our algorithms are up to 579 times faster than the Newton's method. For DNN models, we extend the standard ADMM via a Kullback-Leibler proximal penalty and show that this is still amenable to fast inference via a spectral approach. Compared to a state-of-the-art siamese network, our resulting algorithms are up to 175 times faster and attain better predictions by up to 26% Top-1 Accuracy and 6% Kendall-Tau correlation over five real-life ranking datasets.
Keyword:
Plackett-Luce
ranking
spectral methods
Markov Chain
ADMM

期刊

ACM Transactions on Knowledge Discovery from Data 封面图
ACM Transactions on Knowledge Discovery from Data
IF:
4.8
论文数:
1.3K
被引数:
4.4K

机构

N
national institutes of health (nih) - usa
学者数:
10.3W
论文数: 8.2W
被引数: 111
N
Northeastern University
学者数:
2.5W
论文数: 1.6W
被引数: 3.0W
O
Oregon Health & Science University
学者数:
2.5W
论文数: 2.1W
被引数: 39
学者 查看更多机构
引用论文

引用论文

Dandruff and Seborrhea
err1938-04-01
err0
errOAAI
errGeorge M. MacKee; George M. Lewis
err分享
err收藏
err分享
err收藏
err分享
err收藏
Visualising basins of attraction for the cross-entropy and the squared error neural network loss functions
err2020-08-01
err56
errOAAI
errBosman, Anna Sergeevna; Engelbrecht, Andries; Helbig, Marde
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
学者 查看更多内容