🤖 AI Summary
To address the challenge of learning long-range dependencies in time-series modeling—typically requiring substantial data—this paper proposes Differential Contrastive Predictive Coding (DCPC). DCPC constructs cross-trajectory contrastive tasks by concatenating heterogeneous time-series segments, thereby reducing data requirements for long-horizon dependency modeling. It introduces a novel paradigm that jointly optimizes contrastive representation learning and temporal difference estimation, and is the first to directly deploy the learned representations in off-policy, goal-conditioned reinforcement learning. Integrating contrastive learning, successor representations, and goal-conditioned RL, DCPC achieves up to double the success rate over baselines across multiple prediction and control tasks. In tabular environments, it improves sample efficiency by 20× over standard successor representations and by 1500× over Monte Carlo-based contrastive predictive coding.
📝 Abstract
Predicting and reasoning about the future lie at the heart of many time-series questions. For example, goal-conditioned reinforcement learning can be viewed as learning representations to predict which states are likely to be visited in the future. While prior methods have used contrastive predictive coding to model time series data, learning representations that encode long-term dependencies usually requires large amounts of data. In this paper, we introduce a temporal difference version of contrastive predictive coding that stitches together pieces of different time series data to decrease the amount of data required to learn predictions of future events. We apply this representation learning method to derive an off-policy algorithm for goal-conditioned RL. Experiments demonstrate that, compared with prior RL methods, ours achieves $2 imes$ median improvement in success rates and can better cope with stochastic environments. In tabular settings, we show that our method is about $20 imes$ more sample efficient than the successor representation and $1500 imes$ more sample efficient than the standard (Monte Carlo) version of contrastive predictive coding.