arXiv:2603.26249cs.LG2026-03

用知识蒸馏让大模型变小,适配家庭能源控制器

Knowledge Distillation for Efficient Transformer-Based Reinforcement Learning in Hardware-Constrained Energy Management Systems

  • 用大模型教小模型学决策,保持控制效果
  • 模型参数减少96%,推理内存降90%,速度提63%
  • 适合嵌入式设备部署,节能效果不打折

基于Transformer的强化学习在家庭能源管理中表现优异,尤其是决策变压器(Decision Transformer)能从历史数据中学习有效电池调度策略,提升光伏自用率并降低电费。然而,此类模型计算开销大,难以部署在内存与延迟受限的住宅控制器上。本文研究知识蒸馏技术,将高性能决策变压器策略迁移到更紧凑的学生模型中,以适配嵌入式部署。基于Ausgrid数据集,我们在异构多建筑数据上训练教师模型,并通过匹配动作来蒸馏小型学生模型,从而在保持控制质量的同时大幅压缩模型。在多种师生配置下,蒸馏后性能基本保持,甚至有最高1%的提升;参数量最多减少96%,推理内存最多下降90%,推理时间最多缩短63%。即使在相同架构下蒸馏,成本改善也相当可观。结果表明,知识蒸馏使决策变压器在资源受限硬件上的家庭能源管理应用成为可能。

原文摘要 · Abstract (English)

Transformer-based reinforcement learning has emerged as a strong candidate for sequential control in residential energy management. In particular, the Decision Transformer can learn effective battery dispatch policies from historical data, thereby increasing photovoltaic self-consumption and reducing electricity costs. However, transformer models are typically too computationally demanding for deployment on resource-constrained residential controllers, where memory and latency constraints are critical. This paper investigates knowledge distillation to transfer the decision-making behaviour of high-capacity Decision Transformer policies to compact models that are more suitable for embedded deployment. Using the Ausgrid dataset, we train teacher models in an offline sequence-based Decision Transformer framework on heterogeneous multi-building data. We then distil smaller student models by matching the teachers' actions, thereby preserving control quality while reducing model size. Across a broad set of teacher-student configurations, distillation largely preserves control performance and even yields small improvements of up to 1%, while reducing the parameter count by up to 96%, the inference memory by up to 90%, and the inference time by up to 63%. Beyond these compression effects, comparable cost improvements are also observed when distilling into a student model of identical architectural capacity. Overall, our results show that knowledge distillation makes Decision Transformer control more applicable for residential energy management on resource-limited hardware.

知识蒸馏强化学习能源管理边缘部署

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