Maximum Total Correlation Reinforcement Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: You, Bang, Liu, Puze, Liu, Huaping, Peters, Jan, Arenz, Oleg
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913853179166720
author You, Bang
Liu, Puze
Liu, Huaping
Peters, Jan
Arenz, Oleg
author_facet You, Bang
Liu, Puze
Liu, Huaping
Peters, Jan
Arenz, Oleg
contents Simplicity is a powerful inductive bias. In reinforcement learning, regularization is used for simpler policies, data augmentation for simpler representations, and sparse reward functions for simpler objectives, all that, with the underlying motivation to increase generalizability and robustness by focusing on the essentials. Supplementary to these techniques, we investigate how to promote simple behavior throughout the episode. To that end, we introduce a modification of the reinforcement learning problem that additionally maximizes the total correlation within the induced trajectories. We propose a practical algorithm that optimizes all models, including policy and state representation, based on a lower-bound approximation. In simulated robot environments, our method naturally generates policies that induce periodic and compressible trajectories, and that exhibit superior robustness to noise and changes in dynamics compared to baseline methods, while also improving performance in the original tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2505_16734
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Maximum Total Correlation Reinforcement Learning
You, Bang
Liu, Puze
Liu, Huaping
Peters, Jan
Arenz, Oleg
Machine Learning
Simplicity is a powerful inductive bias. In reinforcement learning, regularization is used for simpler policies, data augmentation for simpler representations, and sparse reward functions for simpler objectives, all that, with the underlying motivation to increase generalizability and robustness by focusing on the essentials. Supplementary to these techniques, we investigate how to promote simple behavior throughout the episode. To that end, we introduce a modification of the reinforcement learning problem that additionally maximizes the total correlation within the induced trajectories. We propose a practical algorithm that optimizes all models, including policy and state representation, based on a lower-bound approximation. In simulated robot environments, our method naturally generates policies that induce periodic and compressible trajectories, and that exhibit superior robustness to noise and changes in dynamics compared to baseline methods, while also improving performance in the original tasks.
title Maximum Total Correlation Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2505.16734