Generative Temporal Difference Learning for Infinite-Horizon Prediction
Not on the map. Why?
Abstract (arXiv)
We introduce the γ-model, a predictive model of environment dynamics with an infinite probabilistic horizon. Replacing standard single-step models with γ-models leads to generalizations of the procedures central to model-based control, including the model rollout and model-based value estimation. The γ-model, trained with a generative reinterpretation of temporal difference learning, is a natural continuous analogue of the successor representation and a hybrid between model-free and model-based mechanisms. Like a value function, it contains information about the long-term future; like a standard predictive model, it is independent of task reward. We instantiate the γ-model as both a generative adversarial network and normalizing flow, discuss how its training reflects an inescapable tradeoff between training-time and testing-time compounding errors, and empirically investigate its utility for prediction and control.
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). First posted on arXiv 27 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 4; 2022: 5; 2023: 9; 2024: 12; 2025: 6; 2026: 7.
43 papers.
Sort:
By area:
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2023Contrastive Difference Predictive Codingcited by 35
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learningcited by 31
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2024Learning Latent Dynamic Robust Representations for World Modelscited by 20
- 2025Temporal Difference Flowscited by 20
- 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RLcited by 20
- 2021Understanding the World Through Actioncited by 17
- 2023RePo: Resilient Model-Based Reinforcement Learning by Regularizing Posterior Predictabilitycited by 17
- 2022Contrastive Value Learning: Implicit Models for Simple Offline RLcited by 15
31 more
- 2024A Distributional Analogue to the Successor Representationcited by 15
- 2024Prioritized Generative Replaycited by 13
- 2025Value Flowscited by 12
- 2023Bootstrapped Representations in Reinforcement Learningcited by 11
- 2023Self-Supervised Reinforcement Learning that Transfers using Random Featurescited by 10
- 2023Value function estimation using conditional diffusion models for controlcited by 10
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learningcited by 9
- 2024Distributional Successor Features Enable Zero-Shot Policy Optimizationcited by 8
- 2023Accelerating exploration and representation learning with offline pre-trainingcited by 7
- 2025Intention-Conditioned Flow Occupancy Modelscited by 7
- 2025Towards Robust Zero-Shot Reinforcement Learningcited by 7
- 2021Successor Feature Representationscited by 6
- 2022Mingling Foresight with Imagination: Model-Based Cooperative Multi-Agent Reinforcement Learningcited by 6
- 2023Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learningcited by 5
- 2021Taylor Expansion of Discount Factorscited by 4
- 2024Video Occupancy Modelscited by 4
- 2024Bellman Diffusion Modelscited by 4
- 2026Compositional Planning with Jumpy World Modelscited by 4
- 2024State Chrono Representation for Enhancing Generalization in Reinforcement Learningcited by 3
- 2024Off-Policy Maximum Entropy RL with Future State and Action Visitation Measurescited by 3
- 2024Associative memory inspires improvements for in-context learning using a novel attention residual stream architecturecited by 3
- 2024Exploration by Learning Diverse Skills through Successor State Measurescited by 2
- 2023Deep Generative Models for Decision-Making and Controlcited by 1
- 2026Can We Really Learn One Representation to Optimize All Rewards?cited by 1
- 2026Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learningcited by 1
- 2026Offline Reinforcement Learning with Universal Horizon Modelscited by 1
- 2026Temporal Difference Learning for Diffusion Modelscited by 1
- 2024Diminishing Return of Value Expansion Methodscited by 0
- 2025A Unified Framework for Zero-Shot Reinforcement Learningcited by 0
- 2026Maximum-Entropy Exploration with Future State-Action Visitation Measurescited by 0
- 2026Switching Successor Measures for Hierarchical Zero-shot Reinforcement Learningcited by 0
- 2026Temporal Difference Learning for Diffusion Models13 Jun 2026
- 2026Offline Reinforcement Learning with Universal Horizon Models15 May 2026
- 2026Switching Successor Measures for Hierarchical Zero-shot Reinforcement Learning13 May 2026
- 2026Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learning22 Apr 2026
- 2026Maximum-Entropy Exploration with Future State-Action Visitation Measures19 Mar 2026
- 2026Compositional Planning with Jumpy World Models23 Feb 2026
- 2026Can We Really Learn One Representation to Optimize All Rewards?11 Feb 2026
- 2025A Unified Framework for Zero-Shot Reinforcement Learning23 Oct 2025
- 2025Towards Robust Zero-Shot Reinforcement Learning17 Oct 2025
- 2025Value Flows9 Oct 2025
- 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RL8 Sep 2025
- 2025Intention-Conditioned Flow Occupancy Models10 Jun 2025
31 more
- 2025Temporal Difference Flows12 Mar 2025
- 2024Diminishing Return of Value Expansion Methods29 Dec 2024
- 2024Associative memory inspires improvements for in-context learning using a novel attention residual stream architecture19 Dec 2024
- 2024Off-Policy Maximum Entropy RL with Future State and Action Visitation Measures9 Dec 2024
- 2024State Chrono Representation for Enhancing Generalization in Reinforcement Learning9 Nov 2024
- 2024Prioritized Generative Replay23 Oct 2024
- 2024Bellman Diffusion Models16 Jul 2024
- 2024Video Occupancy Models25 Jun 2024
- 2024Exploration by Learning Diverse Skills through Successor State Measures14 Jun 2024
- 2024Learning Latent Dynamic Robust Representations for World Models10 May 2024
- 2024Distributional Successor Features Enable Zero-Shot Policy Optimization10 Mar 2024
- 2024A Distributional Analogue to the Successor Representation13 Feb 2024
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learning5 Feb 2024
- 2023Contrastive Difference Predictive Coding31 Oct 2023
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learning10 Oct 2023
- 2023RePo: Resilient Model-Based Reinforcement Learning by Regularizing Posterior Predictability31 Aug 2023
- 2023Bootstrapped Representations in Reinforcement Learning16 Jun 2023
- 2023Deep Generative Models for Decision-Making and Control15 Jun 2023
- 2023Value function estimation using conditional diffusion models for control9 Jun 2023
- 2023Self-Supervised Reinforcement Learning that Transfers using Random Features26 May 2023
- 2023Accelerating exploration and representation learning with offline pre-training31 Mar 2023
- 2023Diminishing Return of Value Expansion Methods in Model-Based Reinforcement Learning7 Mar 2023
- 2022Contrastive Value Learning: Implicit Models for Simple Offline RL3 Nov 2022
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objective18 Sep 2022
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learning15 Jun 2022
- 2022Planning with Diffusion for Flexible Behavior Synthesis20 May 2022
- 2022Mingling Foresight with Imagination: Model-Based Cooperative Multi-Agent Reinforcement Learning20 Apr 2022
- 2021Successor Feature Representations29 Oct 2021
- 2021Understanding the World Through Action24 Oct 2021
- 2021Taylor Expansion of Discount Factors11 Jun 2021
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problem3 Jun 2021
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learningcited by 31
- 2024Learning Latent Dynamic Robust Representations for World Modelscited by 20
- 2026Compositional Planning with Jumpy World Modelscited by 4
- 2026Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learningcited by 1
- 2026Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learning22 Apr 2026
- 2026Compositional Planning with Jumpy World Models23 Feb 2026
- 2024Learning Latent Dynamic Robust Representations for World Models10 May 2024
- 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learning5 Feb 2024
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problem3 Jun 2021
- 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RLcited by 20
- 2023Value function estimation using conditional diffusion models for controlcited by 10
- 2024Bellman Diffusion Modelscited by 4
- 2026Temporal Difference Learning for Diffusion Modelscited by 1
- 2026Temporal Difference Learning for Diffusion Models13 Jun 2026
- 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RL8 Sep 2025
- 2024Bellman Diffusion Models16 Jul 2024
- 2023Value function estimation using conditional diffusion models for control9 Jun 2023
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2023Contrastive Difference Predictive Codingcited by 35
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2025Temporal Difference Flowscited by 20
- 2021Understanding the World Through Actioncited by 17
- 2023RePo: Resilient Model-Based Reinforcement Learning by Regularizing Posterior Predictabilitycited by 17
- 2024A Distributional Analogue to the Successor Representationcited by 15
- 2024Prioritized Generative Replaycited by 13
- 2025Value Flowscited by 12
- 2023Bootstrapped Representations in Reinforcement Learningcited by 11
- 2023Self-Supervised Reinforcement Learning that Transfers using Random Featurescited by 10
12 most cited of 32 All 32 (JSON)
- 2026Switching Successor Measures for Hierarchical Zero-shot Reinforcement Learning13 May 2026
- 2026Maximum-Entropy Exploration with Future State-Action Visitation Measures19 Mar 2026
- 2026Can We Really Learn One Representation to Optimize All Rewards?11 Feb 2026
- 2025A Unified Framework for Zero-Shot Reinforcement Learning23 Oct 2025
- 2025Towards Robust Zero-Shot Reinforcement Learning17 Oct 2025
- 2025Value Flows9 Oct 2025
- 2025Intention-Conditioned Flow Occupancy Models10 Jun 2025
- 2025Temporal Difference Flows12 Mar 2025
- 2024Diminishing Return of Value Expansion Methods29 Dec 2024
- 2024Associative memory inspires improvements for in-context learning using a novel attention residual stream architecture19 Dec 2024
- 2024Off-Policy Maximum Entropy RL with Future State and Action Visitation Measures9 Dec 2024
- 2024State Chrono Representation for Enhancing Generalization in Reinforcement Learning9 Nov 2024
12 newest of 32 All 32 (JSON)