用结构化网络提升排队系统强化学习的效率与泛化能力
A Novel Switch-Type Policy Network for Resource Allocation Problems: Technical Report
- 设计开关型神经网络,融合传统策略的结构模式
- 样本效率更高,在新环境表现显著优于MLP
- 适合需要强泛化的排队网络控制场景
深度强化学习(DRL)在排队网络控制中表现出强大潜力,但普遍使用的多层感知机(MLP)存在样本效率低、易过拟合训练环境的问题,导致在新网络上性能下降。为此,本文提出一种开关型神经网络(STN)架构,通过借鉴传统非学习策略的结构模式,确保相似状态下的动作选择一致性。该设计不仅简化了学习过程,还通过减少过拟合倾向提升了泛化能力。主要贡献包括:(1)提出STN作为MLP的更优替代方案;(2)实验证明STN在多种训练场景下具有更优样本效率;(3)在熟悉环境中性能与MLP相当,在新环境中显著超越。通过嵌入领域知识,STN增强了近端策略优化(PPO)算法的有效性,且不牺牲性能,适用于广泛的排队网络控制问题。
原文摘要 · Abstract (English)
Deep Reinforcement Learning (DRL) has become a powerful tool for developing control policies in queueing networks, but the common use of Multi-layer Perceptron (MLP) neural networks in these applications has significant drawbacks. MLP architectures, while versatile, often suffer from poor sample efficiency and a tendency to overfit training environments, leading to suboptimal performance on new, unseen networks. In response to these issues, we introduce a switch-type neural network (STN) architecture designed to improve the efficiency and generalization of DRL policies in queueing networks. The STN leverages structural patterns from traditional non-learning policies, ensuring consistent action choices across similar states. This design not only streamlines the learning process but also fosters better generalization by reducing the tendency to overfit. Our works presents three key contributions: first, the development of the STN as a more effective alternative to MLPs; second, empirical evidence showing that STNs achieve superior sample efficiency in various training scenarios; and third, experimental results demonstrating that STNs match MLP performance in familiar environments and significantly outperform them in new settings. By embedding domain-specific knowledge, the STN enhances the Proximal Policy Optimization (PPO) algorithm's effectiveness without compromising performance, suggesting its suitability for a wide range of queueing network control problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。