LibMTL:一个用于多任务学习的 Python 库
机器学习
2022-03-29 v1
摘要
本文介绍 LibMTL,一个基于 PyTorch 构建的开源 Python 库,为多任务学习(MTL)提供统一、全面、可复现且可扩展的实现框架。LibMTL 考虑了 MTL 中的不同设定与方法,支持大量 SOTA 的 MTL 方法,包括 12 种损失加权策略、7 种架构以及 84 种不同架构与损失加权方法的组合。此外,LibMTL 的模块化设计使其易用且良好可扩展,因此用户可借助 LibMTL 的支持轻松快速地开发新 MTL 方法、与现有 MTL 方法公平比较,或将 MTL 算法应用于真实世界应用。LibMTL 的源代码与详细文档分别见于 https://github.com/median-research-group/LibMTL 与 https://libmtl.readthedocs.io。
引用
@article{arxiv.2203.14338,
title = {LibMTL: A Python Library for Multi-Task Learning},
author = {Baijiong Lin and Yu Zhang},
journal= {arXiv preprint arXiv:2203.14338},
year = {2022}
}