← Search

Chenjia Bai

43 accepted papers

2026

Align-Then-stEer: Adapting the Vision-Language Action Models through Unified Latent Guidance

ICLR 2026poster

Vision-Language-Action (VLA) models pre-trained on large, diverse datasets show remarkable potential for general-purpose robotic manipulation. However, a primary bottleneck remains in adapting these models to downstream tasks, especially when the robot's embodiment or the task itself differs from th…

Cited by 0SourcecodeScholar
2026

CoNavBench: Collaborative Long-Horizon Vision-Language Navigation Benchmark

ICLR 2026poster

Vision-and-Language Navigation (VLN) primarily focuses on a single-agent-centric approach that executes human instructions step-by-step. In real environments with high demand or parallel workflows, collaboration VLN offers distinct benefits including shorter makespan and greater robustness through p…

Cited by 0SourcecodeScholar
2026

HUSKY: Humanoid Skateboarding System via Physics-Aware Whole-Body Control

RSS 2026poster

While current humanoid whole-body control frameworks predominantly rely on the static environment assumptions, addressing tasks characterized by high dynamism and complex interactions presents a formidable challenge. In this paper, we address humanoid skateboarding, a highly challenging task requiri…

Cited by 4SourceScholar
2026

Kungfubot2: Learning Versatile Motion Skills for Humanoid Whole-Body Control

ICRA 2026poster

Learning versatile whole-body skills by tracking various human motions is a fundamental step toward general-purpose humanoid robots. This task is particularly challenging because a single policy must master a broad repertoire of motion skills while ensuring stability over long-horizon sequences. To …

2026

Towards Adaptive Humanoid Control via Multi-Behavior Distillation and Reinforced Fine-Tuning

AAAI 2026technical

Humanoid robots are promising to learn a diverse set of human-like locomotion behaviors, including standing up, walking, running, and jumping. However, existing methods predominantly require training independent policies for each skill, yielding behavior-specific controllers that exhibit limited gen

Cited by 0SourcePDFScholar
2026

Unifying Value Alignment and Assignment in Cross-Domain Offline Reinforcement Learning with Heterogeneous Datasets

ICML 2026poster

Cross-domain offline reinforcement learning (RL) aims to train an agent that performs well in the target domain using a limited target domain dataset and a source domain dataset that exhibits a dynamics shift. Training directly on the original source dataset typically leads to performance collapse. …

Cited by 0SourceScholar
2026

X-Loco: Towards Generalist Humanoid Locomotion Control via Synergetic Policy Distillation

RSS 2026poster

While recent advances have demonstrated strong performance in individual humanoid skills such as upright locomotion, fall recovery and whole-body coordination, learning a single policy that masters all these skills remains challenging due to the diverse dynamics and conflicting control objectives in…

Cited by 0SourceScholar
2025

Adversarial Locomotion and Motion Imitation for Humanoid Policy Learning

NeurIPS 2025poster

Humans exhibit diverse and expressive whole-body movements. However, attaining human-like whole-body coordination in humanoid robots remains challenging, as conventional approaches that mimic whole-body motions often neglect the distinct roles of upper and lower body. This oversight leads to computa…

Cited by 0SourcecodeScholar
2025

Discriminator-Guided Embodied Planning for LLM Agent

ICLR 2025poster

Large Language Models (LLMs) have showcased remarkable reasoning capabilities in various domains, yet face challenges in complex embodied tasks due to the need for a coherent long-term policy and context-sensitive environmental understanding. Previous work performed LLM refinement relying on outcome…

Cited by 1SourcePDFScholar
2025

Exponential Topology-enabled Scalable Communication in Multi-agent Reinforcement Learning

ICLR 2025poster

In cooperative multi-agent reinforcement learning (MARL), well-designed communication protocols can effectively facilitate consensus among agents, thereby enhancing task performance. Moreover, in large-scale multi-agent systems commonly found in real-world applications, effective communication plays…

2025

Forward KL Regularized Preference Optimization for Aligning Diffusion Policies

AAAI 2025technical

Diffusion models have achieved remarkable success in sequential decision-making by leveraging the highly expressive model capabilities in policy learning. A central problem for learning diffusion policies is to align the policy output with human intents in various tasks. To achieve this, previous me…

Cited by 3SourcePDFScholar
2025

Humanoid Whole-Body Locomotion on Narrow Terrain via Dynamic Balance and Reinforcement Learning

IROS 2025

Humans possess delicate dynamic balance mechanisms that enable them to maintain stability across diverse terrains and under extreme conditions. However, despite significant advances recently, existing locomotion algorithms for humanoid robots are still struggle to traverse extreme environments, espe

Cited by 17SourcecodeScholar
2025

HumanoidGen: Data Generation for Bimanual Dexterous Manipulation via LLM Reasoning

NeurIPS 2025poster

For robotic manipulation, existing robotics datasets and simulation benchmarks predominantly cater to robot-arm platforms. However, for humanoid robots equipped with dual arms and dexterous hands, simulation tasks and high-quality demonstrations are notably lacking. Bimanual dexterous manipulation i…

Cited by 0SourcecodeScholar
2025

Information-Theoretic Reward Decomposition for Generalizable RLHF

NeurIPS 2025poster

Obtaining a generalizable reward model is crucial in Reinforcement Learning from Human Feedback (RLHF) as it enables correctly evaluating unseen prompt-response pairs. However, existing reward models lack this ability, as they are typically trained by increasing the reward gap between chosen and rej…

Cited by 0SourceScholar
2025

KungfuBot: Physics-Based Humanoid Whole-Body Control for Learning Highly-Dynamic Skills

NeurIPS 2025poster

Humanoid robots are promising to acquire various skills by imitating human behaviors. However, existing algorithms are only capable of tracking smooth, low-speed human motions, even with delicate reward and curriculum design. This paper presents a physics-based humanoid control framework, aiming to…

Cited by 0SourcecodeScholar
2025

Online Iterative Self-Alignment for Radiology Report Generation

ACL 2025long

Radiology Report Generation (RRG) is an important research topic for relieving radiologists’ heavy workload. Existing RRG models mainly rely on supervised fine-tuning (SFT) based on different model architectures using data pairs of radiological images and corresponding radiologist-annotated reports.…

Cited by 0SourcePDFScholar
2025

Online Preference Alignment for Language Models via Count-based Exploration

ICLR 2025spotlight

Reinforcement Learning from Human Feedback (RLHF) has shown great potential in fine-tuning Large Language Models (LLMs) to align with human preferences. Existing methods perform preference alignment from a fixed dataset, which can be limited in data coverage and the resulting reward model is hard to…

2025

Preference Aligned Diffusion Planner for Quadrupedal Locomotion Control

IROS 2025

Diffusion models demonstrate superior performance in capturing complex distributions from large-scale datasets, providing a promising solution for quadrupedal locomotion control. However, the robustness of the diffusion planner is inherently dependent on the diversity of the pre-collected datasets.

Cited by 9SourcecodeScholar
2025

Radiology Report Generation via Multi-objective Preference Optimization

AAAI 2025technical

Automatic Radiology Report Generation (RRG) is an important topic for alleviating the substantial workload of radiologists. Existing RRG approaches rely on supervised regression based on different architectures or additional knowledge injection, while the generated report may not align optimally wit…

Cited by 2SourcePDFScholar
2025

Revisiting Multi-Agent World Modeling from a Diffusion-Inspired Perspective

NeurIPS 2025poster

World models have recently attracted growing interest in Multi-Agent Reinforcement Learning (MARL) due to their ability to improve sample efficiency for policy learning. However, accurately modeling environments in MARL is challenging due to the exponentially large joint action space and highly unce…

Cited by 0SourcecodeScholar
2025

Task-Agnostic Pre-training and Task-Guided Fine-tuning for Versatile Diffusion Planner

ICML 2025poster

Diffusion models have demonstrated their capabilities in modeling trajectories of multi-tasks. However, existing multi-task planners or policies typically rely on task-specific demonstrations via multi-task imitation, or require task-specific reward labels to facilitate policy optimization via Reinf…

Cited by 5SourcePDFScholar
2025

Towards Efficient LLM Grounding for Embodied Multi-Agent Collaboration

ACL 2025finding

Grounding the reasoning ability of large language models (LLMs) for embodied tasks is challenging due to the complexity of the physical world. Especially, LLM planning for multi-agent collaboration requires communication of agents or credit assignment as the feedback to re-adjust the proposed plans…

2025

Towards Reliable LLM-based Robots Planning via Combined Uncertainty Estimation

NeurIPS 2025poster

Large language models (LLMs) demonstrate advanced reasoning abilities, enabling robots to understand natural language instructions and generate high-level plans with appropriate grounding. However, LLM hallucinations present a significant challenge, often leading to overconfident yet potentially mis…

Cited by 0SourcecodeScholar
2025

VLP: Vision-Language Preference Learning for Embodied Manipulation

EMNLP 2025

Reward engineering is one of the key challenges in Reinforcement Learning (RL). Preference-based RL effectively addresses this issue by learning from human feedback. However, it is both time-consuming and expensive to collect human preference labels. In this paper, we propose a novel V ision- L angu

2024

Bridging the Sim-to-Real Gap from the Information Bottleneck Perspective

CoRL 2024poster

Reinforcement Learning (RL) has recently achieved remarkable success in robotic control. However, most works in RL operate in simulated environments where privileged knowledge (e.g., dynamics, surroundings, terrains) is readily available. Conversely, in real-world scenarios, robot agents usually rel…

Cited by 9SourcecodeScholar
2024

Constrained Ensemble Exploration for Unsupervised Skill Discovery

ICML 2024poster

Unsupervised Reinforcement Learning (RL) provides a promising paradigm for learning useful behaviors via reward-free per-training. Existing methods for unsupervised RL mainly conduct empowerment-driven skill discovery or entropy-based exploration. However, empowerment often leads to static skills, a…

Cited by 6SourcePDFScholar
2024

Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learning

ICML 2024poster

Cross-domain offline reinforcement learning leverages source domain data with diverse transition dynamics to alleviate the data requirement for the target domain. However, simply merging the data of two domains leads to performance degradation due to the dynamics mismatch. Existing methods address t…

2024

Cross-Domain Policy Adaptation by Capturing Representation Mismatch

ICML 2024poster

It is vital to learn effective policies that can be transferred to different domains with dynamics discrepancies in reinforcement learning (RL). In this paper, we consider dynamics adaptation settings where there exists dynamics mismatch between the source domain and the target domain, and one can g…

2024

Learning an Actionable Discrete Diffusion Policy via Large-Scale Actionless Video Pre-Training

NeurIPS 2024poster

Learning a generalist embodied agent capable of completing multiple tasks poses challenges, primarily stemming from the scarcity of action-labeled robotic datasets. In contrast, a vast amount of human videos exist, capturing intricate tasks and interactions with the physical world. Promising prospec…

2024

ODRL: A Benchmark for Off-Dynamics Reinforcement Learning

NeurIPS 2024poster

We consider off-dynamics reinforcement learning (RL) where one needs to transfer policies across different domains with dynamics mismatch. Despite the focus on developing dynamics-aware algorithms, this field is hindered due to the lack of a standard benchmark. To bridge this gap, we introduce ODRL,…

2024

OVD-Explorer: Optimism Should Not Be the Sole Pursuit of Exploration in Noisy Environments

AAAI 2024technical

In reinforcement learning, the optimism in the face of uncertainty (OFU) is a mainstream principle for directing exploration towards less explored areas, characterized by higher uncertainty. However, in the presence of environmental stochasticity (noise), purely optimistic exploration may lead to ex…

2024

Regularized Conditional Diffusion Model for Multi-Task Preference Alignment

NeurIPS 2024poster

Sequential decision-making can be formulated as a conditional generation process, with targets for alignment with human intents and versatility across various tasks. Previous return-conditioned diffusion models manifest comparable performance but rely on well-defined reward functions, which requires…

Cited by 7SourcePDFScholar
2024

Robust Quadrupedal Locomotion via Risk-Averse Policy Learning

ICRA 2024poster

The robustness of legged locomotion is crucial for quadrupedal robots in challenging terrains. Recently, Reinforcement Learning (RL) has shown promising results in legged locomotion and various methods try to integrate privileged distillation, scene modeling, and external sensors to improve the gene…

Cited by 13SourceScholar
2024

SAM-E: Leveraging Visual Foundation Model with Sequence Imitation for Embodied Manipulation

ICML 2024poster

Acquiring a multi-task imitation policy in 3D manipulation poses challenges in terms of scene understanding and action prediction. Current methods employ both 3D representation and multi-view 2D representation to predict the poses of the robot’s end-effector. However, they still require a considerab…

Cited by 11SourcePDFScholar
2023

Behavior Contrastive Learning for Unsupervised Skill Discovery

ICML 2023poster

In reinforcement learning, unsupervised skill discovery aims to learn diverse skills without extrinsic rewards. Previous methods discover skills by maximizing the mutual information (MI) between states and skills. However, such an MI objective tends to learn simple and static skills and may hinder e…

2023

Cross-Domain Policy Adaptation via Value-Guided Data Filtering

NeurIPS 2023poster

Generalizing policies across different domains with dynamics mismatch poses a significant challenge in reinforcement learning. For example, a robot learns the policy in a simulator, but when it is deployed in the real world, the dynamics of the environment may be different. Given the source and targ…

Cited by 19SourcePDFScholar
2023

Diffusion Model is an Effective Planner and Data Synthesizer for Multi-Task Reinforcement Learning

NeurIPS 2023poster

Diffusion models have demonstrated highly-expressive generative capabilities in vision and NLP. Recent studies in reinforcement learning (RL) have shown that diffusion models are also powerful in modeling complex policies or trajectories in offline datasets. However, these works have been limited to…

2022

Contrastive UCB: Provably Efficient Contrastive Self-Supervised Learning in Online Reinforcement Learning

ICML 2022spotlight

In view of its power in extracting feature representation, contrastive self-supervised learning has been successfully integrated into the practice of (deep) reinforcement learning (RL), leading to efficient policy learning on various applications. Despite its tremendous empirical successes, the unde…

2022

Pessimistic Bootstrapping for Uncertainty-Driven Offline Reinforcement Learning

ICLR 2022spotlight

Offline Reinforcement Learning (RL) aims to learn policies from previously collected datasets without exploring the environment. Directly applying off-policy algorithms to offline RL usually fails due to the extrapolation error caused by the out-of-distribution (OOD) actions. Previous methods tackle…

2022

RORL: Robust Offline Reinforcement Learning via Conservative Smoothing

NeurIPS 2022accept

Offline reinforcement learning (RL) provides a promising direction to exploit massive amount of offline data for complex decision-making tasks. Due to the distribution shift issue, current offline RL algorithms are generally designed to be conservative in value estimation and action selection. Howev…

2021

Dynamic Bottleneck for Robust Self-Supervised Exploration

NeurIPS 2021poster

Exploration methods based on pseudo-count of transitions or curiosity of dynamics have achieved promising results in solving reinforcement learning with sparse rewards. However, such methods are usually sensitive to environmental dynamics-irrelevant information, e.g., white-noise. To handle such dyn…

2021

Principled Exploration via Optimistic Bootstrapping and Backward Induction

ICML 2021spotlight

One principled approach for provably efficient exploration is incorporating the upper confidence bound (UCB) into the value function as a bonus. However, UCB is specified to deal with linear and tabular settings and is incompatible with Deep Reinforcement Learning (DRL). In this paper, we propose a…