arXiv:2509.24947cs.LGcs.AI2025-09被引 1

让深度Q网络的特征更独立,提升线性模型迁移效率

Learning Distinguishable Representations in Deep Q-Networks for Linear Transfer

  • 引入正则项降低状态特征间的正相关性
  • 在标准基准和MinAtar游戏上提升迁移性能
  • 适合希望减少训练开销的RL研究者

深度强化学习在解决复杂序列决策问题上取得成功,但训练需大量超参数调优和高计算成本。迁移学习通过复用先前任务的知识缓解此问题。本文研究发现,标准深度RL模型学习的内部表示高度相关,限制了其与线性函数近似器结合的效果。为此,提出一种新型深度Q学习方法,在目标函数中加入正则项以减少状态表示间的正相关性。利用低相关性特征,显著提升线性函数近似器在迁移学习中的表现。在标准RL基准和MinAtar游戏上的实验与消融研究验证了该方法的有效性,可有效提升迁移性能并降低计算开销。

原文摘要 · Abstract (English)

Deep Reinforcement Learning (RL) has demonstrated success in solving complex sequential decision-making problems by integrating neural networks with the RL framework. However, training deep RL models poses several challenges, such as the need for extensive hyperparameter tuning and high computational costs. Transfer learning has emerged as a promising strategy to address these challenges by enabling the reuse of knowledge from previously learned tasks for new, related tasks. This avoids the need for retraining models entirely from scratch. A commonly used approach for transfer learning in RL is to leverage the internal representations learned by the neural network during training. Specifically, the activations from the last hidden layer can be viewed as refined state representations that encapsulate the essential features of the input. In this work, we investigate whether these representations can be used as input for training simpler models, such as linear function approximators, on new tasks. We observe that the representations learned by standard deep RL models can be highly correlated, which limits their effectiveness when used with linear function approximation. To mitigate this problem, we propose a novel deep Q-learning approach that introduces a regularization term to reduce positive correlations between feature representation of states. By leveraging these reduced correlated features, we enable more effective use of linear function approximation in transfer learning. Through experiments and ablation studies on standard RL benchmarks and MinAtar games, we demonstrate the efficacy of our approach in improving transfer learning performance and thereby reducing computational overhead.

强化学习迁移学习深度Q网络特征解耦

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。