Issue No. 07 - July (2008 vol. 30)

ISSN: 0162-8828

pp: 1158-1170

ABSTRACT

We consider the problem of learning the ranking function that maximizes a generalization of the Wilcoxon-Mann-Whitney statistic on the training data. Relying on an $\epsilon$-accurate approximation for the error-function, we reduce the computational complexity of each iteration of a conjugate gradient algorithm for learning ranking functions from $\mathcal{O}(m^2)$, to $\mathcal{O}(m)$, where $m$ is the number of training samples. Experiments on public benchmarks for ordinal regression and collaborative filtering indicate that the proposed algorithm is as accurate as the best available methods in terms of ranking accuracy, when the algorithms are trained on the same data. However, since it is several orders of magnitude faster than the current state-of-the-art approaches, it is able to leverage much larger training datasets.

INDEX TERMS

Machine learning, Algorithms

CITATION

Balaji Krishnapuram, Ramani Duraiswami, Vikas C. Raykar, "A Fast Algorithm for Learning a Ranking Function from Large-Scale Data Sets",

*IEEE Transactions on Pattern Analysis & Machine Intelligence*, vol. 30, no. , pp. 1158-1170, July 2008, doi:10.1109/TPAMI.2007.70776