用正交投影压缩表示空间,让强化学习更高效
Learning in Low-Dimensional Subspaces: Orthogonal Bottlenecks for Reinforcement Learning

- 引入固定正交投影,强制编码器特征落在低维子空间
- 瓶颈维度超过任务内在秩后性能不降,部分任务可压缩至极低维
- 无需额外损失或预训练,适配任意RL架构
深度强化学习代理通常依赖高维神经表示,尽管越来越多证据表明任务相关的价值和策略结构本质上是低维的。本文提出一种简单有效的表示级先验:通过固定正交投影将编码器特征约束在低维子空间,无需辅助目标、预训练或修改底层强化学习算法。在线性可实现性假设下,我们证明当瓶颈维度超过特征空间中最优价值函数的内在秩时,瓶颈保持表达能力,并使诱导梯度动态保持不变,仅对应于等价的低维参数化。实验发现,在单任务与多任务基准上,一旦瓶颈维度超过一个较小的任务相关阈值,基线性能即被匹配或超越;许多情况下,价值表示可压缩至极低维度而无性能损失,且最小必要维度主要取决于环境复杂度而非编码器宽度。此外,我们分析了表示几何,发现正交瓶颈能稳定特征范数,并关联更高的有效秩。这些结果支持强化学习中的流形假说的表示空间解释,并将正交瓶颈定位为一种轻量、架构无关的强化学习表示塑造机制。
原文摘要 · Abstract (English)
Deep reinforcement learning (RL) agents commonly rely on high-dimensional neural representations, despite growing evidence that task-relevant value and policy structure may be intrinsically low-dimensional. In this work, we present a simple yet effective representation-level prior that inserts a fixed orthonormal projection to constrain encoder features to a low-dimensional subspace, requiring no auxiliary objectives, pretraining, or changes to the underlying RL algorithm. Under a linear realizability assumption, we prove that when the bottleneck dimension exceeds the intrinsic rank of the optimal value function in feature space, the bottleneck preserves expressivity and leaves the induced gradient dynamics unchanged up to an equivalent low-dimensional parameterization. Empirically, we find that across both single and multi-task benchmarks, baseline performance is either matched or improved once the bottleneck dimension exceeds a small task-dependent threshold; in many cases, value representations can be compressed to extremely low dimensions without loss, and the minimal sufficient dimension depends far more on environment complexity than encoder width. In addition, we analyze representation geometry and find that orthogonal bottlenecks stabilize feature norms and are associated with higher effective rank. Together, these results support a representation-space interpretation of the manifold hypothesis in reinforcement learning and position orthogonal bottlenecks as a lightweight, architecture-agnostic mechanism for shaping RL representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。