用可学习的分段线性网络提升强化学习在资源受限环境下的效率与稳定性。
Agile Reinforcement Learning through Separable Neural Architecture and Applications
- 引入SPAN架构,通过可学习预处理层改进KAN结构,兼顾参数效率与计算开销。
- 在多种任务中实现30%-50%样本效率提升,成功率提高1.3-9倍,收敛更可靠。
- 适合对训练成本敏感、需快速稳定部署的边缘设备或真实工程控制场景。
深度强化学习(RL)正广泛应用于资源受限环境,但主流函数逼近器多层感知机(MLPs)因对价值函数平滑结构的归纳偏置不足,导致参数效率低下,影响采样效率与策略学习速度。尽管存在模型压缩技术,但均为事后操作,无法提升学习效率。基于样条的架构如Kolmogorov-Arnold网络(KANs)虽具参数效率优势,却普遍存在显著计算开销。本文提出针对RL的SPAN(SPline-based Adaptive Networks),在KHRONOS框架基础上引入可学习预处理层。SPAN在离散(PPO)与高维连续(SAC)控制任务、离线设置(Minari/D4RL)及真实数据中心暖通空调(HVAC)控制应用中均进行了评估。相比MLP基线,SPAN在基准测试中实现30%-50%的样本效率提升,成功率达1.3-9倍;尽管每步推理延迟增加1.2-1.8倍,但因收敛可靠性更高,总训练成本降低1.3-6.3倍。在HVAC应用中,SPAN在12个月中的9个月降低能耗,同时使热舒适度违规减少1.1-3.4倍,展现出良好的实际工程泛化能力。此外,SPAN具备更强的即时性能和对超参数变化的鲁棒性,是资源受限环境下高效策略学习的可行高性能替代方案。
原文摘要 · Abstract (English)
Deep reinforcement learning (RL) is increasingly deployed in resource-constrained environments, yet go-to function approximators - multilayer perceptrons (MLPs) - are often parameter-inefficient due to an imperfect inductive bias for the smooth structure of many value functions. This mismatch can also hinder sample efficiency and slow policy learning in this capacity-limited regime. Although model compression techniques exist, they operate post-hoc and do not improve learning efficiency. Spline-based architectures such as Kolmogorov-Arnold Networks (KANs) have been shown to offer parameter efficiency but are widely reported to exhibit significant computational overhead, especially at scale. In seeking to address these limitations, this work introduces SPAN (SPline-based Adaptive Networks) for RL. SPAN adapts the KHRONOS framework with a learnable preprocessing layer. SPAN is evaluated across discrete (PPO) and high-dimensional continuous (SAC) control tasks, offline settings (Minari/D4RL) and a real-world datacenter HVAC control application. SPAN achieves a 30-50% improvement in sample efficiency and 1.3-9 times higher success rates across benchmarks compared to MLP baselines. Despite incurring a per-step evaluation overhead of 1.2-1.8x, SPAN's superior convergence reliability yields an expected total training cost 1.3-6.3x lower than MLP baselines when accounting for convergence failures. In the HVAC application, SPAN reduces energy consumption in 9 of 12 months relative to MLP while simultaneously achieving a 1.1-3.4x reduction in thermal comfort violations across the evaluation year, demonstrating generalization to real-world engineering control. Furthermore, SPAN demonstrates superior anytime performance and robustness to hyperparameter variations, suggesting it as a viable, high-performance alternative for learning efficient policies in resource-limited settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。