Search papers, labs, and topics across Lattice.
The paper introduces the Kemeny Transformer, a Transformer-based architecture trained with reinforcement learning to approximate Kemeny-optimal ranking aggregation. This approach tackles the NP-hard problem of minimizing Kemeny distance in consensus ranking, which is crucial for various applications like recommendation systems and search engines. Experiments show the Kemeny Transformer outperforms traditional heuristics and Markov-chain methods while offering faster inference than integer linear programming.
Reinforcement learning unlocks fast, high-quality consensus ranking aggregation, outperforming classical heuristics and ILP solvers for the NP-hard Kemeny optimization problem.
Aggregating a consensus ranking from multiple input rankings is a fundamental problem with applications in recommendation systems, search engines, job recruitment, and elections. Despite decades of research in consensus ranking aggregation, minimizing the Kemeny distance remains computationally intractable. Specifically, determining an optimal aggregation of rankings with respect to the Kemeny distance is an NP-hard problem, limiting its practical application to relatively small-scale instances. We propose the Kemeny Transformer, a novel Transformer-based algorithm trained via reinforcement learning to efficiently approximate the Kemeny optimal ranking. Experimental results demonstrate that our model outperforms classical majority-heuristic and Markov-chain approaches, achieving substantially faster inference than integer linear programming solvers. Our approach thus offers a practical, scalable alternative for real-world ranking-aggregation tasks.