We present a novel method, Contextual goal-Oriented Data Augmentation (CODA), which uses commonly available unlabeled trajectories and context-goal pairs to solve Contextual Goal-Oriented (CGO) problems. By carefully constructing an action-augmented MDP that is equivalent to the original MDP, CODA creates a fully labeled transition dataset under training contexts without additional approximation error. We conduct a novel theoretical analysis to demonstrate CODA's capability to solve CGO problems in the offline data setup. Empirical results also showcase the effectiveness of CODA, which outperforms other baseline methods across various context-goal relationships of CGO problem. This approach offers a promising direction to solving CGO problems using offline datasets.
@article{arxiv.2408.07753,
title = {How to Solve Contextual Goal-Oriented Problems with Offline Datasets?},
author = {Ying Fan and Jingling Li and Adith Swaminathan and Aditya Modi and Ching-An Cheng},
journal= {arXiv preprint arXiv:2408.07753},
year = {2025}
}