动态调整参数掩码,缓解多任务强化学习中的梯度冲突
Soft Conflict-Resolution Decision Transformer for Offline Multi-Task Reinforcement Learning
- 用费舍尔信息动态调节掩码值,保留关键参数
- 基于四分位距自适应设定任务专属稀疏阈值
- 适合需高效知识共享的复杂多任务场景
多任务强化学习旨在为多种任务学习统一策略,但常因任务间梯度冲突导致性能下降。现有基于掩码的方法使用粗粒度二值掩码抑制冲突参数,但会过度抑制关键参数,阻碍知识迁移。此外,不同任务冲突程度各异,而现有方法采用固定稀疏率,难以适应变化。为此,我们提出SoCo-DT,一种基于参数重要性的软冲突化解法。通过费舍尔信息动态调整掩码值,保留重要参数并抑制冲突项。引入基于四分位距(IQR)的动态稀疏策略,利用训练中冲突与和谐得分分布构建任务专属阈值。进一步结合非对称余弦退火调度,实现稀疏率随训练过程持续演化。在Meta-World基准测试中,SoCo-DT在MT50上比最先进方法提升7.6%,在次优数据集上提升10.5%,显著缓解梯度冲突,提升多任务性能。
原文摘要 · Abstract (English)
Multi-task reinforcement learning (MTRL) seeks to learn a unified policy for diverse tasks, but often suffers from gradient conflicts across tasks. Existing masking-based methods attempt to mitigate such conflicts by assigning task-specific parameter masks. However, our empirical study shows that coarse-grained binary masks have the problem of over-suppressing key conflicting parameters, hindering knowledge sharing across tasks. Moreover, different tasks exhibit varying conflict levels, yet existing methods use a one-size-fits-all fixed sparsity strategy to keep training stability and performance, which proves inadequate. These limitations hinder the model's generalization and learning efficiency. To address these issues, we propose SoCo-DT, a Soft Conflict-resolution method based by parameter importance. By leveraging Fisher information, mask values are dynamically adjusted to retain important parameters while suppressing conflicting ones. In addition, we introduce a dynamic sparsity adjustment strategy based on the Interquartile Range (IQR), which constructs task-specific thresholding schemes using the distribution of conflict and harmony scores during training. To enable adaptive sparsity evolution throughout training, we further incorporate an asymmetric cosine annealing schedule to continuously update the threshold. Experimental results on the Meta-World benchmark show that SoCo-DT outperforms the state-of-the-art method by 7.6% on MT50 and by 10.5% on the suboptimal dataset, demonstrating its effectiveness in mitigating gradient conflicts and improving overall multi-task performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。