用点态可用信息识别任务相似性,提升多任务学习效果
Identifying Task Groupings for Multi-Task Learning Using Pointwise V-Usable Information
- 基于点态可用信息(PVI)衡量任务难度,相似难度任务优先分组
- 在15个NLP数据集上,分组后模型参数更少但性能更优且跨领域稳定
- 适合追求高效多任务训练的NLP研究者,尤其在医疗和生物领域
多任务学习的成功高度依赖任务分组方式。盲目或随机分组可能导致负迁移,使联合模型表现劣于单任务模型。尽管已有诸多方法尝试识别任务关联并度量其相似性,但从大量潜在组合中找出最优分组仍是难题。本文提出一种基于点态可用信息(PVI)的任务相关性度量方法,PVI是最近提出的、用于估计给定模型下数据集包含可用信息量的指标。我们假设:若任务的PVI估计值无显著差异,则它们足够相似,可从联合学习中获益。我们在15个NLP数据集(涵盖通用、生物医学与临床领域)上进行了全面实验,将联合学习器与单任务学习器、现有基线方法及最新大模型(Llama 2、GPT-4)对比。结果表明,通过将具有相似PVI估计值的任务分组,联合学习器在参数更少的情况下仍取得有竞争力的表现,并在各领域保持一致性能。
原文摘要 · Abstract (English)
The success of multi-task learning can depend heavily on which tasks are grouped together. Naively grouping all tasks or a random set of tasks can result in negative transfer, with the multi-task models performing worse than single-task models. Though many efforts have been made to identify task groupings and to measure the relatedness among different tasks, it remains a challenging research topic to define a metric to identify the best task grouping out of a pool of many potential task combinations. We propose a metric of task relatedness based on task difficulty measured by pointwise V-usable information (PVI). PVI is a recently proposed metric to estimate how much usable information a dataset contains given a model. We hypothesize that tasks with not statistically different PVI estimates are similar enough to benefit from the joint learning process. We conduct comprehensive experiments to evaluate the feasibility of this metric for task grouping on 15 NLP datasets in the general, biomedical, and clinical domains. We compare the results of the joint learners against single learners, existing baseline methods, and recent large language models, including Llama 2 and GPT-4. The results show that by grouping tasks with similar PVI estimates, the joint learners yielded competitive results with fewer total parameters, with consistent performance across domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。