English

Meta-RL with Shared Representations Enables Fast Adaptation in Energy Systems

Machine Learning 2026-03-10 v1

Abstract

Meta-Reinforcement Learning addresses the critical limitations of conventional Reinforcement Learning in multi-task and non-stationary environments by enabling fast policy adaptation and improved generalization. We introduce a novel Meta-RL framework that integrates a bi-level optimization scheme with a hybrid actor-critic architecture specially designed to enhance sample efficiency and inter-task adaptability. To improve knowledge transfer, we meta-learn a shared state feature extractor jointly optimized across actor and critic networks, providing efficient representation learning and limiting overfitting to individual tasks or dominant profiles. Additionally, we propose a parameter-sharing mechanism between the outer- and inner-loop actor networks, to reduce redundant learning and accelerate adaptation during task revisitation. The approach is validated on a real-world Building Energy Management Systems dataset covering nearly a decade of temporal and structural variability, for which we propose a task preparation method to promote generalization. Experiments demonstrate effective task adaptation and better performance compared to conventional RL and Meta-RL methods.

Keywords

Cite

@article{arxiv.2603.08418,
  title  = {Meta-RL with Shared Representations Enables Fast Adaptation in Energy Systems},
  author = {Théo Zangato and Aomar Osmani and Pegah Alizadeh},
  journal= {arXiv preprint arXiv:2603.08418},
  year   = {2026}
}

Comments

accepted at PAKDD 2026, Hong Kong

R2 v1 2026-07-01T11:10:23.930Z