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

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Dong, Chaosheng, Xiao, Peiyao, Wang, Yijia, Ji, Kaiyi
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912906863443968
author Dong, Chaosheng
Xiao, Peiyao
Wang, Yijia
Ji, Kaiyi
author_facet Dong, Chaosheng
Xiao, Peiyao
Wang, Yijia
Ji, Kaiyi
contents 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 \href{https://github.com/amazon-science/DeepMTL2R}{https://github.com/amazon-science/DeepMTL2R}.
format Preprint
id arxiv_https___arxiv_org_abs_2602_14519
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle DeepMTL2R: A Library for Deep Multi-task Learning to Rank
Dong, Chaosheng
Xiao, Peiyao
Wang, Yijia
Ji, Kaiyi
Machine Learning
Information Retrieval
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 \href{https://github.com/amazon-science/DeepMTL2R}{https://github.com/amazon-science/DeepMTL2R}.
title DeepMTL2R: A Library for Deep Multi-task Learning to Rank
topic Machine Learning
Information Retrieval
url https://arxiv.org/abs/2602.14519