← Search

Jianxiong Li

19 accepted papers

2026

Demystifying Action Space Design for Robotic Manipulation Policies

ICML 2026poster

The specification of the action space plays a pivotal role in imitation-based robotic manipulation policy learning, fundamentally shaping the optimization landscape of policy learning. While recent advances have focused heavily on scaling training data and model capacity, the choice of action space …

Cited by 0SourceScholar
2026

Dichotomous Diffusion Policy Optimization

ICLR 2026poster

Diffusion-based policies have gained growing popularity in solving a wide range of decision-making tasks due to their superior expressiveness and controllable generation during inference. However, effectively training large diffusion policies using reinforcement learning (RL) remains challenging. Ex…

Cited by 0SourcecodeScholar
2026

Sample Efficient Offline RL via T-Symmetry Enforced Latent State-Stitching

ICLR 2026poster

Offline reinforcement learning (RL) has achieved notable progress in recent years. However, most existing offline RL methods require a large amount of training data to achieve reasonable performance and offer limited out-of-distribution (OOD) generalization capability due to conservative data-relate…

Cited by 0SourceScholar
2026

X-VLA: Soft-Prompted Transformer as Scalable Cross-Embodiment Vision-Language-Action Model

ICLR 2026poster

Successful generalist Vision-Language-Action (VLA) models that rely on effective training across diverse robotic platforms with large-scale, cross-embodiment, heterogeneous datasets. To facilitate and leverage the heterogeneity in rich, diverse robotic data sources, we propose a novel Soft Prompt ap…

Cited by 0SourcecodeScholar
2025

Are Expressive Models Truly Necessary for Offline RL?

AAAI 2025technical

Among various branches of offline reinforcement learning (RL) methods, goal-conditioned supervised learning (GCSL) has gained increasing popularity as it formulates the offline RL problem as a sequential modeling task, therefore bypassing the notoriously difficult credit assignment challenge of valu…

2025

Diffusion-Based Planning for Autonomous Driving with Flexible Guidance

ICLR 2025oral

Achieving human-like driving behaviors in complex open-world environments is a critical challenge in autonomous driving. Contemporary learning-based planning approaches such as imitation learning methods often struggle to balance competing objectives and lack of safety assurance,due to limited adapt…

Cited by 3SourcePDFScholar
2025

Efficient Robotic Policy Learning via Latent Space Backward Planning

ICML 2025poster

Current robotic planning methods often rely on predicting multi-frame images with full pixel details. While this fine-grained approach can serve as a generic world model, it introduces two significant challenges for downstream policy learning: substantial computational costs that hinder real-time de…

2025

Flow Matching-Based Autonomous Driving Planning with Advanced Interactive Behavior Modeling

NeurIPS 2025poster

Modeling interactive driving behaviors in complex scenarios remains a fundamental challenge for autonomous driving planning. Learning-based approaches attempt to address this challenge with advanced generative models, removing the dependency on over-engineered architectures for representation fusion…

Cited by 0SourceScholar
2025

Robo-MUTUAL: Robotic Multimodal Task Specification via Unimodal Learning

ICRA 2025

Multimodal task specification is essential for enhanced robotic performance, where Cross-modality Alignment enables the robot to holistically understand complex task instructions. Directly annotating multimodal instructions for model training proves impractical, due to the sparsity of paired multimo

Cited by 5SourceScholar
2025

Skill Expansion and Composition in Parameter Space

ICLR 2025poster

Humans excel at reusing prior knowledge to address new challenges and developing skills while solving problems. This paradigm becomes increasingly popular in the development of autonomous agents, as it develops systems that can self-evolve in response to new challenges like human beings. However, pr…

2025

Universal Actions for Enhanced Embodied Foundation Models

CVPR 2025poster

Training on diverse, internet-scale data is a key factor in the success of recent large foundation models. Yet, using the same recipe for building embodied agents has faced noticeable difficulties. Despite the availability of many crowd-sourced embodied datasets, their action spaces often exhibit si…

2024

DecisionNCE: Embodied Multimodal Representations via Implicit Preference Learning

ICML 2024poster

Multimodal pretraining is an effective strategy for the trinity of goals of representation learning in autonomous robots: $1)$ extracting both local and global task progressions; $2)$ enforcing temporal consistency of visual representation; $3)$ capturing trajectory-level language grounding. Most ex…

2024

Instruction-Guided Visual Masking

NeurIPS 2024poster

Instruction following is crucial in contemporary LLM. However, when extended to multimodal setting, it often suffers from misalignment between specific textual instruction and targeted local region of an image. To achieve more accurate and nuanced multimodal instruction following, we introduce Instr…

2024

Query-Policy Misalignment in Preference-Based Reinforcement Learning

ICLR 2024spotlight

Preference-based reinforcement learning (PbRL) provides a natural way to align RL agents’ behavior with human desired outcomes, but is often restrained by costly human feedback. To improve feedback efficiency, most existing PbRL methods focus on selecting queries to maximally improve the overall qua…

2024

Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Model

ICLR 2024poster

Safe offline reinforcement learning is a promising way to bypass risky online interactions towards safe policy learning. Most existing methods only enforce soft constraints, i.e., constraining safety violations in expectation below thresholds predetermined. This can lead to potentially unsafe outcom…

2023

Mind the Gap: Offline Policy Optimization for Imperfect Rewards

ICLR 2023poster

Reward function is essential in reinforcement learning (RL), serving as the guiding signal to incentivize agents to solve given tasks, however, is also notoriously difficult to design. In many cases, only imperfect rewards are available, which inflicts substantial performance loss for RL agents. In…

2023

Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Regularization

ICLR 2023top-5%

Most offline reinforcement learning (RL) methods suffer from the trade-off between improving the policy to surpass the behavior policy and constraining the policy to limit the deviation from the behavior policy as computing $Q$-values using out-of-distribution (OOD) actions will suffer from errors d…

2023

When Data Geometry Meets Deep Function: Generalizing Offline Reinforcement Learning

ICLR 2023poster

In offline reinforcement learning (RL), one detrimental issue to policy learning is the error accumulation of deep \textit{Q} function in out-of-distribution (OOD) areas. Unfortunately, existing offline RL methods are often over-conservative, inevitably hurting generalization performance outside dat…

2022

A Policy-Guided Imitation Approach for Offline Reinforcement Learning

NeurIPS 2022accept

Offline reinforcement learning (RL) methods can generally be categorized into two types: RL-based and Imitation-based. RL-based methods could in principle enjoy out-of-distribution generalization but suffer from erroneous off-policy evaluation. Imitation-based methods avoid off-policy evaluation but…