用推理式变压器提升无线网络强化学习的评判模型性能
ReaCritic: Reasoning Transformer-based DRL Critic-model Scaling For Wireless Networks
- 引入双向推理结构,增强批评者对复杂状态的判断能力
- 在多种无线网络场景中实现更快收敛和更高最终性能
- 适合需要动态决策的智能无线管理任务
异构网络(HetNets)因用户需求多样和无线环境时变,给智能管理带来巨大挑战,显著增加决策复杂度,限制了现有深度强化学习(DRL)方法的适应性。在价值型或基于演员-批评者的DRL算法中,批评者模块通过估计价值函数指导策略学习,但传统批评者多采用浅层架构,直接将观测映射为标量估计,难以应对多任务复杂性。借鉴大语言模型推理时延展推理步骤可提升决策质量的发现,本文提出ReaCritic——一种基于推理式变压器的批评者模型扩展方案,通过并行状态-动作输入的横向推理与深层变压器堆叠的纵向推理,赋予批评者类推理能力。该方法兼容多种价值型及演员-批评者框架,在多种HetNet设置和标准OpenAI Gym控制任务中均表现出更优的泛化能力与性能。实验表明,ReaCritic显著提升收敛速度与最终表现,代码已开源。
原文摘要 · Abstract (English)
Heterogeneous Networks (HetNets) pose critical challenges for intelligent management due to the diverse user requirements and time-varying wireless conditions. These factors introduce significant decision complexity, which limits the adaptability of existing Deep Reinforcement Learning (DRL) methods. In many DRL algorithms, especially those involving value-based or actor-critic structures, the critic component plays a key role in guiding policy learning by estimating value functions. However, conventional critic models often use shallow architectures that map observations directly to scalar estimates, limiting their ability to handle multi-task complexity. In contrast, recent progress in inference-time scaling of Large Language Models (LLMs) has shown that generating intermediate reasoning steps can significantly improve decision quality. Motivated by this, we propose ReaCritic, a reasoning transformer-based critic-model scaling scheme that brings reasoning-like ability into DRL. ReaCritic performs horizontal reasoning over parallel state-action inputs and vertical reasoning through deep transformer stacks. It is compatible with a broad range of value-based and actor-critic DRL algorithms and enhances generalization in dynamic wireless environments. Extensive experiments demonstrate that ReaCritic improves convergence speed and final performance across various HetNet settings and standard OpenAI Gym control tasks. The code of ReaCritic is available at https://github.com/NICE-HKU/ReaCritic.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。