Saved in:
Bibliographic Details
Main Authors: Liang, Hao, Shi, Shuqing, Zhang, Yudi, Huang, Biwei, Du, Yali
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2510.21427
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914111886983168
author Liang, Hao
Shi, Shuqing
Zhang, Yudi
Huang, Biwei
Du, Yali
author_facet Liang, Hao
Shi, Shuqing
Zhang, Yudi
Huang, Biwei
Du, Yali
contents Large-scale networked systems, such as traffic, power, and wireless grids, challenge reinforcement-learning agents with both scale and environment shifts. To address these challenges, we propose GSAC (Generalizable and Scalable Actor-Critic), a framework that couples causal representation learning with meta actor-critic learning to achieve both scalability and domain generalization. Each agent first learns a sparse local causal mask that provably identifies the minimal neighborhood variables influencing its dynamics, yielding exponentially tight approximately compact representations (ACRs) of state and domain factors. These ACRs bound the error of truncating value functions to $κ$-hop neighborhoods, enabling efficient learning on graphs. A meta actor-critic then trains a shared policy across multiple source domains while conditioning on the compact domain factors; at test time, a few trajectories suffice to estimate the new domain factor and deploy the adapted policy. We establish finite-sample guarantees on causal recovery, actor-critic convergence, and adaptation gap, and show that GSAC adapts rapidly and significantly outperforms learning-from-scratch and conventional adaptation baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2510_21427
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Causality Meets Locality: Provably Generalizable and Scalable Policy Learning for Networked Systems
Liang, Hao
Shi, Shuqing
Zhang, Yudi
Huang, Biwei
Du, Yali
Machine Learning
Large-scale networked systems, such as traffic, power, and wireless grids, challenge reinforcement-learning agents with both scale and environment shifts. To address these challenges, we propose GSAC (Generalizable and Scalable Actor-Critic), a framework that couples causal representation learning with meta actor-critic learning to achieve both scalability and domain generalization. Each agent first learns a sparse local causal mask that provably identifies the minimal neighborhood variables influencing its dynamics, yielding exponentially tight approximately compact representations (ACRs) of state and domain factors. These ACRs bound the error of truncating value functions to $κ$-hop neighborhoods, enabling efficient learning on graphs. A meta actor-critic then trains a shared policy across multiple source domains while conditioning on the compact domain factors; at test time, a few trajectories suffice to estimate the new domain factor and deploy the adapted policy. We establish finite-sample guarantees on causal recovery, actor-critic convergence, and adaptation gap, and show that GSAC adapts rapidly and significantly outperforms learning-from-scratch and conventional adaptation baselines.
title Causality Meets Locality: Provably Generalizable and Scalable Policy Learning for Networked Systems
topic Machine Learning
url https://arxiv.org/abs/2510.21427