Spectral Representation-based Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gao, Chenxiao, Sun, Haotian, Li, Na, Schuurmans, Dale, Dai, Bo
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914284859031552
author Gao, Chenxiao
Sun, Haotian
Li, Na
Schuurmans, Dale
Dai, Bo
author_facet Gao, Chenxiao
Sun, Haotian
Li, Na
Schuurmans, Dale
Dai, Bo
contents In real-world applications with large state and action spaces, reinforcement learning (RL) typically employs function approximations to represent core components like the policies, value functions, and dynamics models. Although powerful approximations such as neural networks offer great expressiveness, they often present theoretical ambiguities, suffer from optimization instability and exploration difficulty, and incur substantial computational costs in practice. In this paper, we introduce the perspective of spectral representations as a solution to address these difficulties in RL. Stemming from the spectral decomposition of the transition operator, this framework yields an effective abstraction of the system dynamics for subsequent policy optimization while also providing a clear theoretical characterization. We reveal how to construct spectral representations for transition operators that possess latent variable structures or energy-based structures, which implies different learning methods to extract spectral representations from data. Notably, each of these learning methods realizes an effective RL algorithm under this framework. We also provably extend this spectral view to partially observable MDPs. Finally, we validate these algorithms on over 20 challenging tasks from the DeepMind Control Suite, where they achieve performances comparable or superior to current state-of-the-art model-free and model-based baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2512_15036
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Spectral Representation-based Reinforcement Learning
Gao, Chenxiao
Sun, Haotian
Li, Na
Schuurmans, Dale
Dai, Bo
Machine Learning
Artificial Intelligence
In real-world applications with large state and action spaces, reinforcement learning (RL) typically employs function approximations to represent core components like the policies, value functions, and dynamics models. Although powerful approximations such as neural networks offer great expressiveness, they often present theoretical ambiguities, suffer from optimization instability and exploration difficulty, and incur substantial computational costs in practice. In this paper, we introduce the perspective of spectral representations as a solution to address these difficulties in RL. Stemming from the spectral decomposition of the transition operator, this framework yields an effective abstraction of the system dynamics for subsequent policy optimization while also providing a clear theoretical characterization. We reveal how to construct spectral representations for transition operators that possess latent variable structures or energy-based structures, which implies different learning methods to extract spectral representations from data. Notably, each of these learning methods realizes an effective RL algorithm under this framework. We also provably extend this spectral view to partially observable MDPs. Finally, we validate these algorithms on over 20 challenging tasks from the DeepMind Control Suite, where they achieve performances comparable or superior to current state-of-the-art model-free and model-based baselines.
title Spectral Representation-based Reinforcement Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2512.15036