DeepMTL2R: A Library for Deep Multi-task Learning to Rank
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
| MTL Method | PF Finding | Scalarization or Gradient | Suggested Evaluation metric | Reference |
| Linear Scalarization (LS) | Y | Scalarization | HVI ( Auger et al., 2012 ) | - |
| ScaleInvariantLinearScalarization | Y | Scalarization | HVI | ( Navon et al., 2022 ) |
| WeightedChebyshev (WC) | Y | Scalarization | HVI | ( Hotegni et al., 2024 ) |
| Soft WeightedChebyshev | Y | Scalarization | HVI | ( Lin et al., 2024 ) |
| EPO | Y | Gradient | HVI | ( Mahapatra and Rajan, 2021 ) |
| WC_MGDA | Y | Gradient | HVI | ( Momma et al., 2022 ) |