GRALP: A Generative Representation Framework for Action Refinement and Latent Planning in Offline Robotic Control
Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Robotics. Pages 7610-7618.
https://doi.org/10.24963/ijcai.2026/846
Keywords
- AIR: Robot control, planning, and execution with guarantees
- Robot control, planning, and execution with guarantees: Safe and robust control under uncertainty
- AIR: Generative AI, robotic foundation models, and reinforcement learning
Abstract
Offline robotic control requires long-horizon reasoning from fixed datasets while avoiding unsafe extrapolation beyond demonstrated behavior.
We propose GRALP, a principled framework that resolves this tension by jointly enforcing support preservation and controllability at the level of temporal abstraction. GRALP adopts a deliberate architectural separation: diffusion is used exclusively as a deterministic action decoder for executing fixed latent skills, while planning and value estimation operate entirely in latent space under conservative constraints. This design enables stable value learning, controllable skill composition, and efficient planning without trajectory-level diffusion sampling at inference. Across unified D4RL benchmarks, GRALP achieves the highest average performance on Navigation, Sequential (Kitchen), and Adroit domains while remaining competitive on locomotion tasks. On contact-rich RoboSuite manipulation with human demonstrations (Lift and Pick-and-Place), GRALP achieves consistently high success rates (over 94%). These results indicate that reliable long-horizon offline control emerges when expressivity is confined to execution and decision-making operates over support-aligned latent abstractions.
- [Ajay et al., 2022] Anurag Ajay, Yilun Du, Abhi Gupta, Joshua Tenenbaum, Tommi Jaakkola, and Pulkit Agrawal. Is conditional generative modeling all you need for decision-making? In The Eleventh International Con- ference on Learning Representations (ICLR), 2022.
- [Chen et al., 2021] Lili Chen, Kevin Lu, Aravind Ra- jeswaran, Kimin Lee, Aditya Grover, Misha Laskin, Pieter Abbeel, Aravind Srinivas, and Igor Mordatch. Decision transformer: Reinforcement learning via sequence model- ing. Advances in neural information processing systems (NeurIPS), 34:15084-15097, 2021.
- [Chen et al., 2023] Huayu Chen, Cheng Lu, Chengyang Ying, Hang Su, and Jun Zhu. Offline reinforcement learn- ing via high-fidelity generative behavior modeling. The Eleventh International Conference on Learning Represen- tations (ICLR), 2023.
- [Fujimoto and Gu, 2021] Scott Fujimoto and Shixiang Shane Gu. A minimalist approach to offline reinforcement learn- ing. Advances in neural information processing systems (NeurIPS), 34:20132-20145, 2021.
- [Fujimoto et al., 2019] Scott Fujimoto, David Meger, and Doina Precup. Off-policy deep reinforcement learning without exploration. In International conference on ma- chine learning (ICML), pages 2052-2062. PMLR, 2019.
- [Hansen et al., 2023] Philippe Hansen, Ilya Kostrikov, Michael Janner, Jakub Grudzien Kuba, and Sergey Levine. Idql: Implicit q-learning as an actor-critic method with diffusion policies. International Conference on Learning Representations (ICLR), 2023.
- [Janner et al., 2021] Michael Janner, Qiyang Li, and Sergey Levine. Offline reinforcement learning as one big sequence modeling problem. Advances in neural information pro- cessing systems (NeurIPS), 34:1273-1286, 2021.
- [Janner et al., 2022] Michael Janner, Yilun Du, Joshua B Tenenbaum, and Sergey Levine. Planning with diffusion for flexible behavior synthesis. In International Conference on Machine Learning (ICML), 2022.
- [Janner et al., 2023] Michael Janner, Justin Fu, Marvin Zhang, and Sergey Levine. Diffusion policies as an ex- pressive policy class for offline reinforcement learning. In International Conference on Learning Representations (ICLR), 2023.
- [Jiang et al., 2023] Zhengyao Jiang, Tianjun Zhang, Michael Janner, Yueying Li, Tim RocktΓ€schel, Edward Grefenstette, and Yuandong Tian. Efficient planning in a compact latent action space. International Conference on Learning Rep- resentations (ICLR), 2023.
- [Kidambi et al., 2020] Rahul Kidambi, Aravind Rajeswaran, Praneeth Netrapalli, and Thorsten Joachims. Morel: Model-based offline reinforcement learning. Advances in neural information processing systems (NeurIPS), 33:21810-21823, 2020.
- [Kim et al., 2024] Donghoon Kim, Minjong Yoo, and Honguk Woo. Offline policy learning via skill-step ab- straction for long-horizon goal-conditioned tasks. In Pro- ceedings of the Thirty-Third International Joint Conference on Artificial Intelligence, (IJCAI-24), pages 4282-4290, 8 2024. Main Track. IJCAI
- [Kostrikov et al., 2021] Ilya Kostrikov, Ashvin Nair, and Sergey Levine. Offline reinforcement learning with im- plicit q-learning. ArXiv, abs/2110.06169, 2021.
- [Kumar et al., 2020] Aviral Kumar, Aurick Zhou, George Tucker, and Sergey Levine. Conservative q-learning for offline reinforcement learning. Advances in neural in- formation processing systems (NeurIPS), 33:1179-1191, 2020.
- [Li et al., 2022] Jinning Li, Chen Tang, Masayoshi Tomizuka, and Wei Zhan. Hierarchical planning through goal-conditioned offline reinforcement learn- ing. IEEE Robotics and Automation Letters (RA-L), 7(4):10216-10223, 2022.
- [Li et al., 2023] Wenhao Li, Xiangfeng Wang, Bo Jin, and Hongyuan Zha. Hierarchical diffusion for offline decision making. In International Conference on Machine Learning (ICML), pages 20035-20064. PMLR, 2023.
- [Liu et al., 2022] Minghuan Liu, Menghui Zhu, and Weinan Zhang. Goal-conditioned reinforcement learning: Prob- lems and solutions. In Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence, (IJCAI-22), pages 5502-5511, 7 2022. Survey Track. IJCAI
- [Lu et al., 2023] Cheng Lu, Huayu Chen, Jianfei Chen, Hang Su, Chongxuan Li, and Jun Zhu. Contrastive energy predic- tion for exact energy-guided diffusion sampling in offline reinforcement learning. In International Conference on Machine Learning (ICML), pages 22825-22855. PMLR, 2023.
- [Mandlekar et al., 2020] Ajay Mandlekar, Fabio Ramos, By- ron Boots, Silvio Savarese, Li Fei-Fei, Animesh Garg, and Dieter Fox. Iris: Implicit reinforcement without interaction at scale for learning control from offline robot manipulation data. In 2020 IEEE International Conference on Robotics and Automation (ICRA), pages 4414-4420. IEEE, 2020.
- [Rosete et al., 2023] Erick Rosete, Oier Mees, Gabriel Kalweit, Joschka Boedecker, and Wolfram Burgard. Latent plans for task-agnostic offline reinforcement learning. In Conference on Robot Learning (CoRL), pages 1838-1849.
- PMLR, 2023.
- [Song et al., 2020] Jiaming Song, Chenlin Meng, and Ste- fano Ermon. Denoising diffusion implicit models. arXiv preprint arXiv:2010.02502, 2020.
- [Venkatraman et al., 2024] Siddarth Venkatraman, Shivesh Khaitan, Ravi Tej Akella, John Dolan, Jeff Schneider, and Glen Berseth. Reasoning with latent diffusion in offline re- inforcement learning. International Conference on Learn- ing Representations (ICLR), 2024.
- [Wang et al., 2023] Zhendong Wang, Jonathan J Hunt, and Mingyuan Zhou. Diffusion policies as an expressive policy class for offline reinforcement learning. In International Conference on Learning Representations (ICLR), 2023.
- [Zaidi et al., 2026] Syed Ahsan Masud Zaidi, Lior Shamir, William Hsu, Scott Dietrich, and Talha Zaidi. Graze: Grounded refinement and motion-aware zero-shot event localization. arXiv preprint arXiv:2604.01383, 2026.
- [Zhou et al., 2020] Guojing Zhou, Hamoon Azizsoltani, Markel Sanz Ausin, Tiffany Barnes, and Min Chi. Hier- archical reinforcement learning for pedagogical policy in- duction (extended abstract). In Proceedings of the Twenty- Ninth International Joint Conference on Artificial Intelli- gence, (IJCAI-20), pages 4691-4695, 7 2020. Sister Con- ferences Best Papers. IJCAI
References were extracted automatically from the PDF and may contain errors.
Full text extracted automatically from the PDF; figures, tables and equations may differ. The PDF is the version of record.
1 Introduction
Learning robotic control policies from fixed datasets is increasingly important in settings where online exploration is unsafe, costly, or physically irreversible, such as contact-rich manipulation, constrained navigation, and long-horizon task execution. In offline settings, agents must compose multi-step behaviors solely from logged demonstrations while avoiding actions that deviate from the physically valid and safe behavior distribution. This creates a central challenge: long-horizon reasoning is essential for task success, yet extrapolation beyond demonstrated support can induce severe robotic failures, including unstable contacts, unsafe transitions, and geometric constraint violations. These constraints reflect open-world deployment settings, where embodied AI systems must operate under limited supervision, perceive contact-rich physical interactions [Zaidi et al., 2026], and compose demonstrated behaviors in novel configurations while strictly avoiding unsafe extrapolation.
Prior offline reinforcement learning (RL) methods address distributional shift through conservative value estimation or behavior regularization (e.g., BCQ [Fujimoto et al., 2019], CQL [Kumar et al., 2020]), but often restrict expressivity, limiting multi-modal robotic behaviors. Implicit Q-Learning (IQL) [Kostrikov et al., 2021; Hansen et al., 2023] relaxes explicit conservatism but typically relies on unimodal policy representations. Sequence-modeling approaches [Chen et al., 2021; Janner et al., 2021] recast control as conditional prediction but lack explicit support constraints. Diffusion-based methods [Janner et al., 2022; Ajay et al., 2022; Janner et al., 2023; Lu et al., 2023] capture rich, multi-modal action distributions yet face fundamental limitations in long-horizon robotic control. Actionlevel diffusion policies accumulate error over long horizons, while trajectory-level diffusion planners incur substantial inference cost [Janner et al., 2022; Ajay et al., 2022]. Hierarchical and latent diffusion methods [Li et al., 2023; Venkatraman et al., 2024] further couple planning decisions to stochastic denoising, which can destabilize value estimation and compound uncertainty across robotic tasks.
We argue the core difficulty in offline robotic planning is not lacking expressive models, but the absence of a unified principle simultaneously enforcing support preservation and controllability at temporal abstraction levels. Effective latent planning requires: (i) decoded behaviors remaining near demonstrations (safe extrapolation), and (ii) latent changes producing meaningful action changes (controllable planning). Violating the former causes unsafe extrapolation; violating the latter yields posterior collapse where planning loses control authority. Existing methods typically emphasize one requirement at the expense of the other.
We propose GRALP, a Generative Representation framework for Action refinement and Latent Planning, which resolves this tension through a key architectural principle: diffusion is used exclusively as a deterministic action decoder, never for latent planning. GRALP constructs a shared latent skill space trained on offline demonstrations, guided by a local principle: decoded action deviation is bounded by latent deviation from dataset support and decoder sensitivity. This principle is instantiated through conservative latent value learning (constraining planned latents to demonstrated regions) and geometric regularization (preserving decoder sensitivity to latent perturbations). This separation enables: (i) efficient discrete planning over latents via a Transformer, (ii) consistent latent-to-action mappings for stable value estimation, and (iii) conservative value estimation directly over planning decisions, achieving controllable and support-preserving temporal abstraction without trajectory-level diffusion overhead.

Our contributions are:
- We formulate a principled design objective for longhorizon offline robotic planning that balances support preservation and controllability at the level of temporal abstraction, providing a unified perspective for understanding common failure modes in existing approaches.
- We introduce GRALP, a unified latent-space planning framework that operationalizes this objective through a deliberate architectural separation: diffusion is used exclusively as an action-space decoder for skill execution, while planning and value estimation operate directly in latent space, enabling efficient long-horizon composition without diffusion-based planning at inference time.
- With extensive evaluation on D4RL benchmarks and RoboSuite manipulation with human demonstrations, we show that GRALP achieves highest average performance on Navigation, Sequential, and Adroit tasks, surpassing strongest baselines by 8.64%, 3.77%, and 3.38% respectively, while generalizing to contact-rich RoboSuite manipulation and leading across multiple task categories among compared baselines.
2 Preliminaries
2.1 Offline Latent-Space Control Setting
We consider an offline reinforcement learning setting modeled as a Markov decision process (S, A, π, π, πΎ), where the agent is given access only to a fixed dataset D = {(π π‘ , ππ‘ , ππ‘ , π π‘+1)} collected by an unknown behavior policy. The goal is to learn a policy that maximizes the discounted return (π½(π) = Eπ οΏ½οΏ½βπ‘=0 πΎ π‘ π(π π‘ , ππ‘ ) οΏ½ ) without additional environment interaction. In robotic manipulation and navigation, such datasets typically consist of short, locally coherent motion fragments generated by low-level controllers. To expose this structure, trajectories are decomposed into fixed-length segments ππ‘ = (π π‘ , ππ‘ , ..., π π‘+π», ππ‘+π»), which serve as basic units for constructing temporally extended latent skills in GRALP.
2.2 DiffusionModeling of Trajectory Segments
Diffusion models represent complex continuous distributions by progressively corrupting data with noise and learning a reverse denoising process. For a trajectory segment representation π₯0, the forward diffusion step produces a noisy sample π₯π via π₯π = β Β―πΌπ π₯0 + β1 - Β―πΌπ π, π βΌ N(0, πΌ), where Β―πΌπ controls the signal-to-noise ratio at diffusion step π.
GRALP employs Denoising Diffusion Implicit Models (DDIM) [Song et al., 2020] for reconstruction, using a deterministic reverse mapping (π₯π-1 = ππ (π₯π, π)) parameterized by a noise prediction network. Since the diffusion decoder is trained on empirical trajectory segments, reconstructed action sequences remain close to demonstrated behavior.
3 Methodology
We present GRALP, a hierarchical offline reinforcement learning framework derived from a single design objective: enabling long-horizon planning while preserving support and controllability in latent space. GRALP decomposes control into two levels: a low-level generative decoder that maps latent skills to short, dataset-consistent motion fragments, and a high-level planner that sequences skills directly in latent space (Fig. 1). Consistent with this design, diffusion is used exclusively as a deterministic action-space decoder, enabling efficient discrete planning while preserving expressive, supportaligned action generation.

3.1 Design Principle: Support Preservation vs. Controllability
Effective latent planning in offline robotics must balance two competing requirements: planned latents should remain close to demonstrated behaviors to avoid unsafe extrapolation, while decoded actions must remain responsive to latent variations to enable meaningful control.
Let ππ : ZβA π» denote thedecodermappinga latent skill to an π»-step action sequence at state π , and let ZD (π ) denote dataset-supported latents near π (encoded from segments with start states in a local neighborhood).
Assumption 1 (Local Decoder Smoothness). For each state π , the decoder ππ is πΏπ -Lipschitz continuous in a neighborhood of dataset-supported latents.
Proposition 1 (Support-Controllability Bound). For any planned latent π§ and state π ,
where πΏπ (π§, π ) = minπ§DβZD (π ) β₯π§ - π§Dβ₯2 measures deviation from dataset-supported skills, and π§ β D is nearest such latent.
Proof sketch. ByAssumption 1, β₯ ππ (π§)- ππ (π§ β² )β₯2 β€ πΏπ β₯π§π§ β² β₯2 locally. Choosing π§ β² = π§ β D yields β₯ ππ (π§) - ππ (π§ β D )β₯2 β€ πΏπ β₯π§ - π§ β D β₯2 = πΏπ πΏπ (π§, π ). The result follows directly from local Lipschitz continuity of ππ . β‘
This bound exposes a fundamental trade-off: constraining πΏπ (π§, π ) preserves dataset support, while maintaining nondegenerate πΏπ ensures latent variations induce meaningful action changes. GRALP explicitly enforces both through conservative latent planning and geometric regularization.
Architectural Implication. The bound motivates a clear separation of roles: diffusion is confined to deterministic action execution, yielding a well-defined regulated decoder ππ (π§), while long-horizon decision making operates entirely in latent space under conservative value learning.
3.2 Diffusion Action Decoder for Latent Skills
Trajectory data are segmented into fixed-length fragments π = (π π‘ , ππ‘ , . . . , π π‘+π», ππ‘+π»), each encoded into a latent skill π§ β Z using a Conditional Diffusion Autoencoder (CDAE). The encoder ππ (π§|π) produces a Gaussian posterior, while a stateconditional prior ππ(π§|π ) regularizes the latent space and provides proposals for planning.
Action sequences are reconstructed using a diffusion decoder trained on empirical segments. Forward noising follows

with a learned reverse process
At inference, deterministic DDIM sampling ensures consistent latent-to-action mappings, essential for reproducible planning and stable value-guided refinement. By confining diffusion to short-horizon action reconstruction within fixedlatent skills, GRALP avoids trajectory-level diffusion at inference, decoupling expressive execution from long-horizon planning.
To prevent the decoder from ignoring latent inputs, latent and state information are injected via Feature-wise Linear Modulation (FiLM) conditioning:
The diffusion model is trained with the standard noiseprediction objective
encouraging decoded behaviors to remain close to the empirical data manifold. As the decoder is trained exclusively on trajectory segments fromD, latent skills selected by the planner decode into short motion fragments that remain aligned with demonstrated behaviors.
3.3 Geometric Regularization for Latent Controllability
Latent planning is effective only if variations in the latent code induce meaningful changes in decoded actions. In expressive decoders such as diffusion models, this is not guaranteed: the decoder may ignore the latent variable and reconstruct actions directly from the state input, a failure mode known as posterior collapse, which renders high-level planning ineffective.
GRALP enforces latent controllability via two complementary mechanisms. First, stochastic state dropout is applied in the decoder's FiLM conditioning, randomly masking state inputs with probability πdrop to force reliance on latent skills. Second, a geometric regularizer that penalizes collapsed latent directions. Let πlow(π | π , π§) denote the action sequence decoded by the diffusion model; we encourage local sensitivity to latent perturbations by penalizing small Jacobian norms:

Algorithm 1 GRALP Training and Planning Pipeline
Input: Offline dataset D, skill horizon π», latent dim ππ§ Output: Trained skill model (ππ, ππ, ππ), latent critics (π1, π2,π), planner ππ
- 1: Stage 1: Diffusion skill model training (Sec. 3.2-3.3)
- 2: Initialize encoder ππ(π§|π), prior ππ, decoder ππ (π|π , π§)
- 3: for π‘ = 1 to π1 do
- 4: Sample trajectory segment π βΌ D
- 5: Sample latent skill π§ βΌ ππ(π§ | π)
- 6: Sample diffusion timestep and apply state dropout
- 7: Update (π, π, π) using loss Eq. 5 and Eq. 7
- 8: end for
- 9: Freeze ππ and collect Encoded Dataset π·π§
- 10: Stage 2: Conservative latent planning (Sec. 3.4)
- 11: Initialize critics π1,π2,π and planner ππ
- 12: for π‘ = 1 to π2 do
- 13: Sample latent transition batch from dataset π·π§
- 14: Update π via expectile regression (Eq. 11)
- 15: Update π1,2 using TD targets (Eq. 8) &CQL (Eq. 10)
- 16: Update planner ππ via AWR (Eqs. 11-12)
- 17: Polyak update target networks
- 18: end for
The log-barrier form provides:
(i) Collapse prevention: unbounded penalty as β₯βπ§ ππ (π§)β₯2 β 0 (ii) Stability: the penalty saturates since π ππ [- log(π + π)] = - 1 π+π β 0 as π ββ.
By preserving non-degenerate decoder sensitivity, Z-force directly controls the Lipschitz term πΏπ in Proposition 1.
Together, these terms maintain an expressive and responsive latent space, ensuring high-level planning retains authority over action generation (Fig. 3).
3.4 Conservative Planning in Latent Space
Once a latent skill space is established, planning reduces to selecting a sequence of latent codes whose decoded motion fragments remain feasible and supported by the offline dataset. However, even in latent space, unconstrained value learning can assign overly optimistic values to skills that are rarely or never produced by the encoder, leading to extrapolation error.
GRALP defines a latent Markov decision process where each latent skill π§ executes a decoded fragment of length π», yielding return π π§ and successor state π β² . Critic and value functions are trained offline using dataset-aligned transitions π·π§ (π , π§, π π§, π β² ). The diffusion decoder is never used to generate training transitions and is invoked only for action execution at evaluation time, ensuring value estimation and planning remain fully latent-space operations. To ensure temporal consistency with skill execution, we use horizon Bellman targets
where π π‘:π‘+π»-1 = οΏ½ π»-1 β=0 πΎ β ππ‘+β is the cumulative return accrued during execution of a latent skill.
The latent action-value function is defined as
where π(π ) predicts the value of future latent decisions. Two Q-networks and a value network share a common representation for stability. The horizon π» trades off temporal abstraction and critic variance; moderate values support reliable long-horizon reasoning while avoiding unstable targets.
To suppress overestimation for unsupported skills, we adapt Conservative Q-Learning to the latent domain. This penalty explicitly restricts latent value overestimation to regions supported by the offline dataset, enforcing small πΏπ (π§, π ) in Proposition 1 and preventing extrapolative skill selection. The critic minimizes a Bellman error while penalizing high values assigned to latent proposals drawn from a distribution π(π§) that targets regions near the dataset support. Specifically, π(π§) includes Gaussian perturbations around encoder latents, local variations of dataset skills, and samples from proposal head trained to match the planner's predictions. The resulting objective enforces a conservative lower bound:
discouraging extrapolative skills while preserving relative ordering among feasible ones. This conservative value learning constrains the support-deviation term πΏs in our design principle (Fig. 4). High-level planning is performed by a causal Transformer that conditions on recent states, previously executed skills, and an optional return-to-go signal. The planner is trained via Advantage-Weighted-Regression (AWR), i.e.,
| Non-Diffusion Diffusion-based | |||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Model-Based | Diffusion as | Planner Diffusion | Policy | OURS | |||||||||||||
| Task | Name | BC | CQL IQL | DT | MOReL | TT | TAP | LDCQ | Diffuser | DD | HDMI | QGPO | SfBC | IDQL | DQL | DQL@1 | (Mean Β± Std) |
| Hopper-M | 58.5 | Model-Free 66.3 | 67.6 | 95.4 | 61.1 | 63.4 | 66.2 | 58.5 | 76.4 | 98.0 | 57.1 | 65.4 | as 82.4 | 64.1 | 96.26 Β± 0.4 | ||
| Hopper-M-R | 52.9 18.1 | 95.0 | 94.7 | 82.7 | 93.6 | 91.5 | 87.3 | 86.2 | 79.3 100.0 | 99.6 | 96.9 | 86.2 | 92.1 | 100.7 | 63.1 | 88.41 Β± 1.8 | |
| Hopper-M-E | 52.5 | 105.4 | 91.5 | 107.6 | 108.7 | 110.0 | 105.5 | 111.3 | 96.8 107.2 | 111.8 | 108.0 | 108.6 | 108.6 | 107.2 | 100.6 | 111.90 Β± 2.3 | |
| HCheetah-M | 42.6 44.0 | 47.4 | 42.6 | 42.1 | 46.9 | 45.0 | 42.8 | 44.2 49.1 | 113.5 48.0 | 54.1 | 45.9 | 51.0 | 50.6 | 47.8 | 51.23 Β± 0.3 | ||
| Locomotion | HCheetah-M-R | 36.6 | 45.5 44.2 | 36.6 | 40.2 | 41.9 | 40.8 | 41.8 | 42.2 | 44.9 | 47.6 | 37.1 | 45.9 | 47.5 | 44.0 | 41.92 Β± 0.42 | |
| HCheetah-M-E | 55.2 | 91.6 86.7 | 86.8 | 53.3 | 95.0 | 91.8 | 90.2 | 79.8 | 39.3 90.6 | 93.5 | 91.6 | 95.9 | 96.1 | 94.8 | 90.13 Β± 0.9 | ||
| Walker2d-M | 75.3 | 78.3 | 74.0 | 77.8 | 79.0 | 64.9 | 69.4 | 79.7 82.5 | 92.1 79.9 | 86.0 | 53.1 | 82.5 | 85.1 | 82.0 | 78.42 Β± 0.8 | ||
| Walker2d-M-R | 26.0 | 72.5 77.2 73.9 | 66.6 | 49.8 | 82.6 | 66.8 | 68.5 | 75.0 | 84.4 | 65.1 | 85.1 | 94.3 | 75.6 | 83.94 Β± 1.8 | |||
| Walker2d-M-E | 107.5 | 108.8 109.6 | 108.1 | 95.6 | 101.9 | 107.4 | 109.3 | 61.2 108.4 | 108.8 | 80.7 107.9 | 110.7 | 109.6 | 108.6 | 109.7 | 108.9 | 111.61 Β± 0.7 | |
| Average | 51.87 | 77.19 76.96 | 74.70 | 72.94 | 78.80 | 74.77 | 76.18 | 75.30 | 82.60 | 86.58 | 72.70 | 81.68 | 86.30 | 75.60 | 83.76 | ||
| 5.7 | 47.4 | 27.3 | 0 | - | - | 134.2 | 81.60 | - | 73.9 | 57.9 | - | - | 143.14 Β± 4.0 | ||||
| Maze2d-U Maze2d-M | 3.8 30.0 | 5.0 34.9 | 32.1 | 1.3 | - | - | 125.3 | 113.9 121.5 | 116.2 122.3 | 120.1 121.8 | 73.8 | 89.5 | - | - | 145.08 Β± | ||
| Maze2d-L | 5.0 | 12.5 58.6 | 18.1 | 0 | - | - | 150.1 | 123.0 | 125.9 | 128.6 | - - | 90.0 | - | - | 4.9 149.92 Β± 2.9 | ||
| AntMaze-U-D | 45.6 80.3 | 62.2 | 53.0 | 0 | - | 69.8 | 81.4 | 72.2 | 49.2 | 73.7 | 74.4 | 74.4 85.3 | 80.2 | 66.2 | 56.4 | 90.75 Β± 3.2 | |
| Navigation | AntMaze-M-D | 0.0 53.7 | 70.0 | 0.0 | - | 0.6 | 85 | 68.9 | 45.5 | 24.6 - | 83.6 | 82.0 | 84.8 | 78.6 | 14.8 | 87.75 Β± 4.2 | |
| AntMaze-L-D | 0.0 | 14.9 47.5 | 0.0 | - | 0 | - | 57.7 | 22.0 | 7.5 71.5 | 64.8 | 45.5 | 67.9 | 56.6 | 1.6 | 52.35 Β± 2.5 | ||
| Average | 14.06 | 53.43 | 22.78 | 0.32 | - | - | 102.94 | 74.28 | - | - | 72.48 | 78.38 | - | - | 111.83 | ||
| Kitchen-P | 28.68 | 46.3 | 83.01 | - | 47.9 | 66.7 | 60.5 | - | 67.96 Β± 3.14 | ||||||||
| Sequential | Kitchen-Mxd | 38.0 51.5 | 49.8 51.0 51.0 | 0.0 | - | - | - | 67.8 | 55.7 52.5 | 56.5 51.8 | 66.0 | 45.4 | 66.5 | 62.6 | - | 66.70 Β± 2.8 | |
| Kitchen-Cmp | 65.0 | 43.8 62.5 | 2.5 15.0 | - - | - - | - - | 62.3 62.5 | - | - | - - | 45.5 62.8 | 77.9 | 62.5 | 84.0 | - | 81.06 Β± 4.3 | |
| Average | 51.5 | 48.2 53.26 | 5.83 | - | - | - | 64.4 | - | - - | 58.1 | 57.06 | 65.23 | 69.03 | - | 71.63 | ||
| Pen-C | 37.0 | 39.2 | 23.5 | - | 11.4 | 57.4 | 47.7 | 61.7 | 47.7 | 82.3 | 28.3 | 49.3 | 81.96 Β± 2.5 | ||||
| Pen-E | 85.1 107.0 | 37.3 - | 110.0 | - | 72.0 | 127.4 | 121.2 | 99.7 | 107.6 | 48.3 109.5 | 54.2 119.1 | - - | 137.8 | 133.5 | 112.6 | 146.20 Β± 3.9 | |
| Door-E | 0.4 1.6 | -0.1 | - | -0.1 | 11.7 | 1.1 | 0.1 | 9.0 | 9.3 | 11.2 | - | 4.4 | -0.3 | 10.6 | 2.23 Β± 1.2 | ||
| Door-C | 0.0 34.9 | 101.5 - | 95.5 | - | 94.1 | 104.8 | 105.1 | 103 | 94.0 | 85.9 | 98.8 | - | 105.0 | 104.8 | 93.7 | 108.03 Β± 0.6 | |
| Relocate-C | -0.1 | -0.3 | - | -0.1 | -0.2 | - | 0.27 Β± 0.06 | ||||||||||
| -0.3 | -0.1 | 15.0 | -0.2 | -0.2 | 0.0 | 0.15 | -0.1 91.3 | 102.5 | - | 0.0 | 0.1 | -0.2 | 109.93 Β± 2.1 | ||||
| Relocate-E Hammer-C | 101.3 0.8 | 95.0 - 2.1 | - | 10.3 | 105.8 | 104.0 | 102.2 | 87.5 | 1.0 | 1.1 | - | 107.0 3.5 | 108.5 0.2 | 95.2 1.1 | |||
| Hammer-E | 1.6 | 0.2 | - | 0.5 | 1.2 | 2.8 | 1.2 | 0.9 | 128.3 | 4.26 Β± 1.3 | |||||||
| Adroit | 125.6 | 86.7 - | 89.7 | - | 15.5 | 127.6 | 45.8 | 103.1 | 106.7 | 111.8 | 123.2 | - 127.6 | 114.8 | 132.35 Β± 1.16 | |||
| Average | 48.05 | 53.96 55.66 | 41.69 | - | 25.45 | 66.96 | 53.44 | 58.86 | 56.70 57.13 | 63.74 | - | 70.95 | 62.925 | 59.6375 | 73.15 | ||
| Pen-H | 34.4 | 37.5 71.3 | 6.3 | - | 36.4 | 76.5 | 74.1 | - | 64.1 | 66.2 | 73.9 | - | - | 72.8 | 66.0 | 80.92 Β± 3.5 | |
| Door-H | 0.5 | 4.3 | 3.9 | - | 11.8 | - | 8.0 | 7.23 Β± | |||||||||
| Relocate-H | 9.9 0.2 | 0.2 | 0.1 | 8.8 | - | 6.9 | 7.1 | 8.5 | - | - | 0.2 | 2.3 Β± | |||||
| Hammer-H | 0.0 1.2 | 0.1 4.4 1.4 | 0.5 | - - | 0.0 0.8 | 0.2 1.4 | 0.3 1.5 | - - | 0.2 1.0 1.2 | 0.1 | 0.2 1.4 | - - | - - | - - | 1.3 | 0.24 0.9 3.55 Β± 1.1 | |
| 19.28 | 21.93 | 18.05 | 21 | - | - | 18.86 | 22.98 | ||||||||||
| Average | 9.03 | 13.00 | 2.72 | - | 9.33 | 21.73 | - | 18.65 | - | ||||||||
emphasizes high-value skills while remaining anchored to conservative value estimates. The corresponding weight
scales the likelihood objective with KL regularization toward the behavior distribution. At evaluation time, GRALP refines planner proposals using the Cross-Entropy Method (CEM, 3 iterations) directly in latent space. Starting from the Transformer's Gaussian proposal, CEM iteratively samples latent candidates, scores each with the conservative critic π(π , π§), and fits a new Gaussian to the top elite samples. The highestvalue latent is then decoded via DDIM. Algorithmic pseudocode for all training stages is provided in Algorithm 1.
4 Experimental Evaluation
We evaluate GRALP on the D4RL benchmark, spanning locomotion, navigation, sequential manipulation, and dexterous robotics control. These domains pose qualitatively different challenges: MuJoCo locomotion emphasizes short-horizon smooth control; Maze2D and AntMaze require composing locally demonstrated motions into goal-directed trajectories under sparse rewards; Kitchen tasks demand multi-stage sequential execution; and Adroit environments introduce highdimensional, contact-rich manipulation (Fig. 5). Categorywise results in Table 1 assess whether GRALP's strengths emerge where long-horizon reasoning and trajectory composition are central. We also evaluate on RoboSuite pick-andplace tasks to examine generalization beyond D4RL.

4.1 Experimental Protocol
GRALP is trained in two stages: diffusion-based action decoder learning, followed by latent critic and causal Transformer planner optimization. The encoder uses 2-layer GRU (hidden size 256), critics are 3-layer MLPs (width 256), and the planner is an 8-layer Transformer with 8 attention heads. Based on preliminary horizon evaluation, we fix the skill horizon to π» β {8, 12} depending on the domain and report results averaged over 10 random seeds with 25 evaluation rollouts per seed. We compare against representative model-free, modelbased, diffusion-planner, and diffusion-policy baselines (Table 1). Following QGPO [Lu et al., 2023], we additionally report D-QL@1, which removes the resampling trick to reflect raw diffusion decision quality. Baseline results are taken from the corresponding publications under matching protocols.
4.2 D4RL Benchmark Results
Table 1 reports unified D4RL results. GRALP attains the highest average performance on Navigation, Sequential, and Adroit tasks, while remaining competitive on Locomotion; in contrast, other baselines excel in at most one or two domains.
Navigation
GRALP achieves the highest average performance on navigation benchmarks (Maze2D and AntMaze). These environments require composing locally demonstrated motion fragments into coherent, maze-spanning trajectories absent from the dataset. GRALP outperforms both diffusion-based planners and policies, particularly on umaze and medium variants that exhibit severe distributional shift, consistent with conservative latent value learning that restricts value extrapolation to unsupported regions.
Sequential Manipulation
On the Kitchen suite, which requires correct temporal ordering and subtask composition, GRALP achieves the highest category-wise average performance. Performance gains are largest on Partial and Mixed variants, where sparse feedback demands composing reusable motion fragments, indicating that latent skills capture reusable structure without taskspecific heuristics.

Adroit Manipulation
GRALP also achieves the strongest overall average on the Adroit benchmark across cloned, expert, and human demonstration settings. These tasks involve high-dimensional, contact-rich dynamics that require fine-grained control while benefiting from temporal abstraction. The results indicate that GRALP preserves local control while enabling effective skilllevel planning, avoiding degradation observed in hierarchical approaches to dexterous manipulation.
Locomotion
On MuJoCo locomotion, GRALP remains competitive with leading offline RL methods but does not uniformly outperform approaches optimizing actions directly. This is expected, as these domains emphasize smooth, reactive control over short horizons, where long-horizon planning provides limited advantage. Importantly, GRALP incurs no performance penalty despite its hierarchical structure.
4.3 Generalization to Contact-RichManipulation
Beyond D4RL, we evaluate GRALP on RoboSuite manipulation tasks, including Lift and Pick-and-Place, using offline human demonstrations collected via teleoperation. RoboSuite employs a 7-DoF Panda manipulator with substantially different kinematics and action spaces than Adroit's Shadow Hand and navigation agents. Lift evaluates single-stage grasp acquisition under precise contact constraints, while Pick-and-Place additionally requires sequencing grasp, transport, and placement. Across 10 seeds with 25 rollouts per seed, GRALP achieves a success rate of 96.6% Β± 2.1% on Lift and 94.3% Β± 3.4% on Pick-and-Place. Together with results on navigation and dexterous manipulation benchmarks, these experiments span diverse embodiments and contact regimes, demonstrating that GRALP's latent skill abstraction generalizes across morphologies while preserving support-aligned execution.
4.4 Ablation Studies
Table 2 reports ablations on four representative tasks spanning sparse-reward navigation, compositional manipulation, and dexterous control. Each removal isolates the role of a component in the support-controllability design principle.
Latent Controllability (πΏs)
Removing Z-force substantially degrades performance across all tasks, with largest drops in AntMaze and Kitchen, indicating reduced sensitivity of decoded actions to latent perturbations. State dropout exhibits a complementary effect, with larger degradation in high-dimensional manipulation tasks where reliance on latent structure is critical. Removing both regularizers causes severe collapse, confirming that both mechanisms jointly preserve decoder responsiveness and prevent posterior collapse.
| Ablation | AntMaze-M-D | Kitchen-Mxd | Pen-C | Maze2d-L |
|---|---|---|---|---|
| GRALP (Full) | 87.75 | 66.14 | 81.96 | 149.9 |
| W/O Z-force | 63.82 | 40.75 | 68.38 | 115.46 |
| W/O State Dropout | 65.19 | 32.46 | 71.22 | 133.36 |
| W/O Both | 40.48 | 29.48 | 51.02 | 95.85 |
| W/O CQL | 37.67 | 35.65 | 44.72 | 73.64 |
| Critic Ignore | 61.19 | 43.28 | 48.33 | 135.49 |
| π»=1 | 34.28 | 30.03 | 42.71 | 118.02 |
Support Preservation (πΏs)
Removing conservative regularization (W/O CQL) produces the largest drop across tasks, particularly in sparse-reward navigation and long-horizon manipulation. Without conservatism, the latent critic overestimates unsupported skills, causing the planner to select infeasible latent actions, which directly validates the necessity of constraining latent deviation from dataset support.
Critic-Guided Refinement
Without refinement (Critic Ignore), performance degrades most on sparse-reward tasks such as AntMaze, while denser environments are less affected, suggesting latent-space local search is particularly useful when credit assignment is difficult and value uncertainty is high.
Temporal Abstraction
Setting π»=1 collapses GRALP to action-level planning and results in consistent performance degradation, demonstrating that fixed-length latent skills are essential for effective longhorizon reasoning. As shown in Fig. 7, performance consistently peaks for intermediate horizons (π»β[8, 12]). Short horizons provide insufficient abstraction, while longer horizons introduce excessive critic variance, revealing a clear abstraction-stability trade-off shared across navigation, manipulation, and dexterous control domains.
4.5 MechanismValidation
We complement quantitative benchmarks and ablations with targeted diagnostics probing the support-controllability principle. Each experiment isolates a specific failure mode predicted by the design inequality in Section 3.
Conservative Critic Behavior
The ring-bandit experiment in Fig. 4 validates conservative value learning in latent space. The critic assigns low values to latents outside dataset support while preserving relative ordering among feasible skills. When combined with constrained CEM-based planning, latent proposals refine to the highestvalue region, confirming that conservative regularization constrains the support-deviation term πΏs without collapsing skill diversity.

Latent Controllability via Z-Force
Fig. 3 illustrates the effect of geometric regularization on decoder sensitivity. Without Z-force, latent perturbations produce minimal changes in decoded actions, indicating collapse. With Z-force, perturbations induce structured action responses, confirming non-degenerate local sensitivity πΏs.
Emergence of Structured Latent Skills
PCA visualizations in Fig. 2 show how latent structure evolves with skill horizon. At π»=1, representations are diffuse and overlapping, while increasing horizons yield progressively more organized and separable clusters around π»=12, aligning with performance trends in Fig. 7.
Long-Horizon Composition
Open-world navigation often requires composing locally demonstrated motion fragments into globally coherent plans that were never observed end-to-end in the dataset. Figure 8 visualizes GRALP stitching fragmented AntMaze data into maze-spanning rollouts. Despite fixed-interval skill transitions, GRALP navigates narrow passages and multi-turn corridors without trajectory-level diffusion planning, confirming that latent skill sequencing enables stable long-horizon behavior while preserving data support.
Together, these diagnostics link GRALP's gains to conservative support preservation, maintained decoder sensitivity, and reliable temporal skill composition.
5 RelatedWork
Offline reinforcement learning must contend with distributional shift from finite datasets, where optimistic value estimates can induce out-of-support actions. Prior work addresses this via behavior regularization and conservative value learning (e.g., BCQ [Fujimoto et al., 2019], TD3+BC [Fujimoto and Gu, 2021], CQL [Kumar et al., 2020], IQL [Kostrikov et al., 2021]), or by recasting control as conditional sequence modeling (e.g., Decision Transformer [Chen et al., 2021]), which avoids explicit bootstrapping but offers limited guarantees for compositional long-horizon planning. Hierarchical and latent-skill approaches reduce temporal complexity by planning over abstractions such as subgoals or latent skills [Mandlekar et al., 2020; Rosete et al., 2023; Kim et al., 2024; Li et al., 2022; Zhou et al., 2020], while model-based methods enable longer-range reasoning through learned dynamics or rollout-based planning (e.g., MOReL [Kidambi et al., 2020], Trajectory Transformer [Janner et al., 2021], TAP [Jiang et al., 2023]). Diffusion models have been applied either as trajectory planners generating full action sequences at inference [Janner et al., 2022; Ajay et al., 2022; Li et al., 2023; Venkatraman et al., 2024] or as expressive stochastic policies [Wang et al., 2023; Chen et al., 2023; Lu et al., 2023; Hansen et al., 2023], while goal-conditioned reinforcement learning frames long-horizon control via conditioning on desired outcomes but suffers from sparse rewards and generalization challenges [Liu et al., 2022]. In contrast, GRALP confines diffusion to deterministic action execution within latent skills, while long-horizon decision making operates in latent space via conservative value learning and skill-level planning, avoiding trajectory-level diffusion at inference.

6 Conclusion
We presented GRALP, a principled framework for offline robotic planning that preserves dataset support while enabling controllable long-horizon decision making through a clear separation of roles: diffusion is used solely for deterministic action execution within latent skills, while planning and value estimation operate directly in latent space under conservative constraints. Across long-horizon benchmarks, GRALP achieves the highest average performance on Navigation, Sequential Manipulation, and Adroit tasks while remaining competitive on locomotion. RoboSuite experiments further demonstrate generalization to contact-rich manipulation from human demonstrations without trajectory-level diffusion planning. While our evaluation focuses on high-fidelity simulation, GRALP is designed with physical deployment constraints in mind: conservative latent planning mitigates unsafe extrapolation, morphology-agnostic skill abstraction promotes cross-platform generalization, and deterministic execution improves reproducibility. Extending to physical platforms with sim-to-real techniques is an important next step.
Reproducibility: Hyperparameters, protocols and implementation are available at github.com/talhazaidi13/GRALP.
Acknowledgments
This work was supported in part by the United States Department of Agriculture (USDA) National Institute of Food and Agriculture (NIFA) under Award Number 2023-67021-44838. Any opinions, findings, and conclusions or recommendations expressed in this material are those of the author(s) and do not necessarily reflect the views of the USDA NIFA.
@inproceedings{ijcai2026p846,
title = {GRALP: A Generative Representation Framework for Action Refinement and Latent Planning in Offline Robotic Control},
author = {Zaidi, Talha and Munir, Arslan and Abbas, Sardar Ali},
booktitle = {Proceedings of the Thirty-Fifth International Joint Conference on
Artificial Intelligence, {IJCAI-26}},
publisher = {International Joint Conferences on Artificial Intelligence Organization},
editor = {Diego Calvanese},
pages = {7610--7618},
year = {2026},
month = {8},
note = {AI and Robotics},
doi = {10.24963/ijcai.2026/846},
url = {https://doi.org/10.24963/ijcai.2026/846},
}
TY - CPAPER TI - GRALP: A Generative Representation Framework for Action Refinement and Latent Planning in Offline Robotic Control AU - Zaidi, Talha AU - Munir, Arslan AU - Abbas, Sardar Ali T2 - Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence, IJCAI-26 PB - International Joint Conferences on Artificial Intelligence Organization PY - 2026 SP - 7610 EP - 7618 DO - 10.24963/ijcai.2026/846 UR - https://doi.org/10.24963/ijcai.2026/846 ER -Copied
Machine-readable versions of this paper, for tools and AI agents: