arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.08056cs.LG
收藏

Pareto 低秩适配器:带偏好的高效多任务学习

Pareto Low-Rank Adapters: Efficient Multi-Task Learning with Preferences

EPFL(洛桑联邦理工学院);University of Geneva(日内瓦大学);Meta FAIR

Nikolaos Dimitriadis, Pascal Frossard, Francois Fleuret

AI总结:

本文提出PaLoRA,通过低秩适配器和确定性偏好采样调度,实现参数高效的帕累托前沿学习,在多任务学习中优于现有方法并大幅降低内存开销。

摘要:

机器学习中的多任务权衡可以通过帕累托前沿学习(PFL)方法来解决,这些方法使用单一模型参数化帕累托前沿(PF)。与传统的多任务学习(MTL)在训练前决定单一权衡不同,PFL允许在推理期间选择所需的操作点。然而,最近的PFL方法存在可扩展性有限、收敛速度慢和内存需求过大的问题,同时在偏好到目标空间的映射上表现出不一致性。我们引入了PaLoRA,一种新颖的参数高效方法,通过两种方式解决这些局限性。首先,我们为任何神经网络架构增加任务特定的低秩适配器,并在其凸包中连续参数化帕累托前沿。我们的方法引导原始模型和适配器分别学习通用特征和任务特定特征。其次,我们提出了一种偏好向量的确定性采样调度,以加强这种分工,从而实现更快的收敛,并在整个训练过程中加强从偏好到目标空间映射的有效性。我们的实验表明,PaLoRA在各种数据集上优于最先进的MTL和PFL基线,可扩展到大型网络,在场景理解基准测试中,与竞争的PFL基线相比,内存开销减少了23.8-31.7倍。

英文摘要:

Multi-task trade-offs in machine learning can be addressed via Pareto Front Learning (PFL) methods that parameterize the Pareto Front (PF) with a single model. PFL permits to select the desired operational point during inference, contrary to traditional Multi-Task Learning (MTL) that optimizes for a single trade-off decided prior to training. However, recent PFL methodologies suffer from limited scalability, slow convergence, and excessive memory requirements, while exhibiting inconsistent mappings from preference to objective space. We introduce PaLoRA, a novel parameter-efficient method that addresses these limitations in two ways. First, we augment any neural network architecture with task-specific low-rank adapters and continuously parameterize the PF in their convex hull. Our approach steers the original model and the adapters towards learning general and task-specific features, respectively. Second, we propose a deterministic sampling schedule of preference vectors that reinforces this division of labor, enabling faster convergence and strengthening the validity of the mapping from preference to objective space throughout training. Our experiments show that PaLoRA outperforms state-of-the-art MTL and PFL baselines across various datasets, scales to large networks, reducing the memory overhead $23.8-31.7$ times compared with competing PFL baselines in scene understanding benchmarks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑