TSPRank: Bridging Pairwise and Listwise Methods with a Bilinear Travelling Salesman Model
Weixian Waylon Li, Yftah Ziser, Yifei Xie, Shay B. Cohen, Tiejun Ma
Abstract
Traditional Learning-To-Rank (LETOR) approaches, including pairwise methods like RankNet and LambdaMART, often fall short by solely focusing on pairwise comparisons, leading to sub-optimal global rankings. Conversely, deep learning based listwise methods, while aiming to optimise entire lists, require complex tuning and yield only marginal improvements over robust pairwise models. To overcome these limitations, we introduce Travelling Salesman Problem Rank (TSPRank), a hybrid pairwise-listwise ranking method. TSPRank reframes the ranking problem as a Travelling Salesman Problem (TSP), a well-known combinatorial optimisation challenge that has been extensively studied for its numerous solution algorithms and applications. This approach enables the modelling of pairwise relationships and leverages combinatorial optimisation to determine the listwise ranking. This approach can be directly integrated as an additional component into embeddings generated by existing backbone models to enhance ranking performance. Our extensive experiments across three backbone models on diverse tasks, including stock ranking, information retrieval, and historical events ordering, demonstrate that TSPRank significantly outperforms both pure pairwise and listwise methods. Our qualitative analysis reveals that TSPRank's main advantage over existing methods is its ability to harness global information better while ranking. TSPRank's robustness and superior performance across different domains highlight its potential as a versatile and effective LETOR solution. The code and preprocessed data are available at https://github.com/waylonli/TSPRank-KDD2025.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1dc2cda8-be8c-4662-a853-4a968bba8600Cited by top-tier papers1
Ask how each one uses itBuilds on10
- Stock Selection via Spatiotemporal Hypergraph Attention Network: A Learning to Rank ApproachRamit Sawhney, Shivam Agarwal, Arnav Wadhwa, Tyler Derr et al.AAAI 2021 · 183 citations
- SetRank: Learning a Permutation-Invariant Ranking Model for Information RetrievalLiang Pang, Jun Xu, Qingyao Ai, Yanyan Lan et al.SIGIR 2020 · 113 citations
- CI-STHPAN: Pre-trained Attention Network for Stock Selection with Channel-Independent Spatio-Temporal HypergraphHongjie Xia, Huijie Ao, Long Li, Yu Liu et al.AAAI 2024 · 48 citations
- Cross-Positional Attention for Debiasing ClicksHonglei Zhuang, Zhen Qin, Xuanhui Wang, Michael Bendersky et al.WWW 2021 · 43 citations
- Are Neural Rankers still Outperformed by Gradient Boosted Decision Trees?Zhen Qin, Le Yan, Honglei Zhuang, Yi Tay et al.ICLR 2021 · 41 citations
Related papers
- NeuroLKH: Combining Deep Learning Model with Lin-Kernighan-Helsgaun Heuristic for Solving the Traveling Salesman ProblemLiang Xin, Wen Song, Zhiguang Cao, Jie ZhangNeurIPS 2021 · 202 citations
- An Alternative Cross Entropy Loss for Learning-to-RankSebastian BruchWWW 2021 · 58 citations
- Which Tricks are Important for Learning to Rank?Ivan Lyzhin, Aleksei Ustimenko, Andrey Gulin, Liudmila ProkhorenkovaICML 2023 · 8 citations
- An Efficient Combinatorial Optimization Model Using Learning-to-Rank DistillationHonguk Woo, Hyunsung Lee, Sangwoo ChoAAAI 2022 · 7 citations
- Fast Attention-based Learning-To-Rank Model for Structured Map SearchChiqun Zhang, Michael R. Evans, Max Lepikhin, Dragomir YankovSIGIR 2021 · 4 citations
