cs.LGMay 13, 2026

Trajectory-Level Data Augmentation for Offline Reinforcement Learning

Authors: Tobias SchmählingMatthias BurkhardtTobias Windisch

Organizations: University of Applied Sciences Kempten, Kempten, Germany.

Abstract

We propose a data augmentation method for offline reinforcement learning, motivated by active positioning problems. Particularly, our approach enables the training of off-policy models from a limited number of suboptimal trajectories. We introduce a trajectory-based augmentation technique that exploits task structure and the geometric relationship between rewards, value functions, and mathematical properties of logging policies. During data collection, our augmentation supports suboptimal logging policies, leading to higher data quality and improved offline reinforcement learning performance. We provide theoretical justification for these strategies and validate them empirically across positioning tasks of varying dimensionality and under partial observability.

Explore similar work

CardsList