cs.LGFeb 16, 2026

DeepMTL2R: A Library for Deep Multi-task Learning to Rank

Authors: Chaosheng Dong, Peiyao Xiao, Yijia Wang, Kaiyi Ji

Organizations: Amazon USA

Abstract

This paper presents DeepMTL2R, an open-source deep learning framework for Multi-task Learning to Rank (MTL2R), where multiple relevance criteria must be optimized simultaneously. DeepMTL2R integrates heterogeneous relevance signals into a unified, context-aware model by leveraging the self-attention mechanism of transformer architectures, enabling effective learning across diverse and potentially conflicting objectives. The framework includes 21 state-of-the-art multi-task learning algorithms and supports multi-objective optimization to identify Pareto-optimal ranking models. By capturing complex dependencies and long-range interactions among items and labels, DeepMTL2R provides a scalable and expressive solution for modern ranking systems and facilitates controlled comparisons across MTL strategies. We demonstrate its effectiveness on a publicly available dataset, report competitive performance, and visualize the resulting trade-offs among objectives. DeepMTL2R is available at https://github.com/amazon-science/DeepMTL2R.

Figures & tables

Explore similar work

CardsList
  1. Lightweight Ranking Heads: Accelerating Multi-Task Experimentation in Production Recommender Systems

    Sep 21, 2026Sanjay Surendranath Girija, Aniruddh Nath, Li Wei +6Learning to RankMulti-Task Learning

  2. DenoiseRank: Learning to Rank by Diffusion Models

    Feb 17, 2026Ying Wang, Preslav Nakov, Shangsong LiangLearning to Rank

  3. EvoRank: LLM-Guided Evolution of Multi-Objective Learning-to-Rank Pipelines

    Aug 30, 2026Rayhan Patel, Shabaz PatelLearning to RankAutomated Algorithm Discovery