Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Yuzheng Zhuang

dblp:193/7920 · DBLP profile ↗
← Back
20ranked-venue papers
0as first author
17since 2021 · last 2026
0000-0002-0915-0254ORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 18 · 16 since 2021Systems, architecture and hardware · 4 · 4 since 2021Graphics, computer vision, multimedia, augmented reality and games · 2 · 1 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
13 papers
Reinforcement learning · 37% Robot manipulation · 15% Planning, search and constraint satisfaction · 11%
Human-computer interaction and pervasive computing
1 paper
Games and playful interaction · 100%

Topics — the 29 heaviest of 34, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Reinforcement learning
imitation learning
1.932025
Astra: Efficient Transformer Architecture and Contrastive Dynamics Learning for Embodied Instruction Following · EMNLP 2025
Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization · ICML 2022
Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets · IJCAI 2020
Robotics › Motion planning and robot control › motion planning › manipulation planning
long-horizon manipulation
1.522024
PERIA: Perceive, Reason, Imagine, Act via Holistic Language and Vision Planning for Manipulation · NeurIPS 2024
Generate Subgoal Images Before Act: Unlocking the Chain-of-Thought Reasoning in Diffusion Model for Robot Manipulation with Multimodal Prompts · CVPR 2024
Knowledge, reasoning and agents › Planning, search and constraint satisfaction › robot task planning
spatial planning
0.912025
MineAnyBuild: Benchmarking Spatial Planning for Open-world AI Agents · NeurIPS 2025
Knowledge, reasoning and agents › Planning, search and constraint satisfaction
task planning
0.912025
Structured Preference Optimization for Vision-Language Long-Horizon Task Planning · EMNLP 2025
Robotics › Robot manipulation › embodied foundation models
vision-language-action model
0.912025
Astra: Efficient Transformer Architecture and Contrastive Dynamics Learning for Embodied Instruction Following · EMNLP 2025
Robotics › Robot manipulation
affordance learning
0.812024
Articulated Object Manipulation with Coarse-to-fine Affordance for Mitigating the Effect of Point Cloud Noise · ICRA 2024
Robotics › Robot manipulation › grasping
articulated object manipulation
0.812024
Articulated Object Manipulation with Coarse-to-fine Affordance for Mitigating the Effect of Point Cloud Noise · ICRA 2024
Machine learning › Generative modeling
diffusion model
0.812024
Generate Subgoal Images Before Act: Unlocking the Chain-of-Thought Reasoning in Diffusion Model for Robot Manipulation with Multimodal Prompts · CVPR 2024
Machine learning › Generative modeling › diffusion model
diffusion planning
0.812024
Generate Subgoal Images Before Act: Unlocking the Chain-of-Thought Reasoning in Diffusion Model for Robot Manipulation with Multimodal Prompts · CVPR 2024
Knowledge, reasoning and agents › Multi-agent systems › multimodal agent
multimodal planning
0.812024
PERIA: Perceive, Reason, Imagine, Act via Holistic Language and Vision Planning for Manipulation · NeurIPS 2024
Computer vision › 3D vision
point cloud processing
0.812024
Articulated Object Manipulation with Coarse-to-fine Affordance for Mitigating the Effect of Point Cloud Noise · ICRA 2024
Robotics › Robot manipulation
subgoal image generation
0.812024
Generate Subgoal Images Before Act: Unlocking the Chain-of-Thought Reasoning in Diffusion Model for Robot Manipulation with Multimodal Prompts · CVPR 2024
Robotics › Robot navigation and mapping
visual navigation
0.812024
SCALE: Self-Correcting Visual Navigation for Mobile Robots via Anti-Novelty Estimation · ICRA 2024
Knowledge, reasoning and agents › Planning, search and constraint satisfaction › intelligent control
behavior selection
0.712023
Learnable Behavior Control: Breaking Atari Human World Records via Sample-Efficient Behavior Selection · ICLR 2023
Machine learning › Reinforcement learning › exploration › efficient exploration
sample-efficient exploration
0.712023
Learnable Behavior Control: Breaking Atari Human World Records via Sample-Efficient Behavior Selection · ICLR 2023
Machine learning › Reinforcement learning › meta-reinforcement learning
context-based meta RL
0.612022
DOMINO: Decomposed Mutual Information Optimization for Generalized Context in Meta-Reinforcement Learning · NeurIPS 2022
Machine learning › Reinforcement learning › generalization in reinforcement learning
dynamics generalization
0.612022
DOMINO: Decomposed Mutual Information Optimization for Generalized Context in Meta-Reinforcement Learning · NeurIPS 2022
Machine learning › Reinforcement learning › imitation learning
learning from observation
0.612022
Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization · ICML 2022
Machine learning › Reinforcement learning
meta-reinforcement learning
0.612022
DOMINO: Decomposed Mutual Information Optimization for Generalized Context in Meta-Reinforcement Learning · NeurIPS 2022
Machine learning › Reinforcement learning
policy optimization
0.612022
Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization · ICML 2022
Machine learning › Reinforcement learning › model-based reinforcement learning › model-based planning
imagination-based planning
0.512021
Model-Based Reinforcement Learning via Imagination with Derived Memory · NeurIPS 2021
Machine learning › Representation and self-supervised learning › representation learning › latent representation learning › state representation learning
latent dynamics model
0.512021
Model-Based Reinforcement Learning via Imagination with Derived Memory · NeurIPS 2021
Machine learning › Reinforcement learning
model-based reinforcement learning
0.512021
Model-Based Reinforcement Learning via Imagination with Derived Memory · NeurIPS 2021
Machine learning › Reinforcement learning › imitation learning › generative imitation learning
generative adversarial imitation learning
0.412020
Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets · IJCAI 2020
Knowledge, reasoning and agents › Multi-agent systems › agent interaction › multi-agent interaction
multi-agent interaction modeling
0.412020
Multi-Agent Interactions Modeling with Correlated Policies · ICLR 2020
Machine learning › Reinforcement learning
multi-agent reinforcement learning
0.412020
Multi-Agent Interactions Modeling with Correlated Policies · ICLR 2020
Machine learning › Reinforcement learning › imitation learning
multi-modal imitation learning
0.412020
Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets · IJCAI 2020
Computer vision › Vision and language › vision-language model
multimodal large language model
0.312025
MineAnyBuild: Benchmarking Spatial Planning for Open-world AI Agents · NeurIPS 2025
Natural language and speech › Language models and text generation
preference optimization
0.312025
Structured Preference Optimization for Vision-Language Long-Horizon Task Planning · EMNLP 2025

Methods — techniques the papers use, named apart from their topics

multimodal large language model · 1.6transformer architecture · 0.9preference optimization · 0.9contrastive learning · 0.9benchmark construction · 0.9semantic alignment · 0.8implicit q-learning · 0.8diffusion model · 0.8bidirectional generation · 0.8affordance learning · 0.8sample-efficient reinforcement learning · 0.7behavior control · 0.7
YearPublicationVenuePosition
2026 A Survey on Vision-Language-Action Models for Embodied AI
abstract
Embodied AI is widely recognized as a cornerstone of artificial general intelligence (AGI) because it involves controlling embodied agents to perform tasks in the physical world. Building on the success of large language models (LLMs) and vision-language models (VLMs), a new category of multimodal models-referred to as vision-language-action (VLA) models-has emerged to address language-conditioned robotic tasks in embodied AI by leveraging their distinct ability to generate actions. The recent proliferation of VLAs necessitates a comprehensive survey to capture the rapidly evolving landscape. To this end, we present the first survey on VLAs for embodied AI. This work provides a detailed taxonomy of VLAs, organized into three major lines of research. The first line focuses on individual components of VLAs. The second line is dedicated to developing VLA-based control policies adept at predicting low-level actions. The third line comprises high-level task planners capable of decomposing long-horizon tasks into a sequence of subtasks, thereby guiding VLAs to follow more general user instructions. Furthermore, we provide an extensive summary of relevant resources, including datasets, simulators, and benchmarks. Finally, we discuss the challenges facing VLAs and outline promising future directions in embodied AI. A curated repository associated with this survey is available at: https://github.com/yueen-ma/Awesome-VLA.
Yueen Ma 0001, Zixing Song, Yuzheng Zhuang, Jianye Hao, Irwin King
IEEE Trans. Neural Networks Learn. Syst.3
2025 Structured Preference Optimization for Vision-Language Long-Horizon Task Planning
abstract
Xiwen Liang, Min Lin, Weiqi Ruan, Rongtao Xu, Yuecheng Liu, Jiaqi Chen, Bingqian Lin, Yuzheng Zhuang, Xiaodan Liang. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025.
Xiwen Liang, Weiqi Ruan, Rongtao Xu, Yuecheng Liu, Bingqian Lin, Yuzheng Zhuang, Xiaodan Liang
EMNLP8
2025 Astra: Efficient Transformer Architecture and Contrastive Dynamics Learning for Embodied Instruction Following
abstract
Vision-language-action models have gained significant attention for their ability to model multimodal sequences in embodied instruction following tasks. However, most existing models rely on causal attention, which we find suboptimal for processing sequences composed of interleaved segments from different modalities. In this paper, we introduce Astra, a novel Transformer architecture featuring trajectory attention and learnable action queries, designed to efficiently process segmented multimodal trajectories and predict actions for imitation learning. Furthermore, we propose a contrastive dynamics learning objective to enhance the model’s understanding of environment dynamics and multimodal alignment, complementing the primary behavior cloning objective. Through extensive experiments on three large-scale robot manipulation benchmarks, Astra demonstrates substantial performance improvements over previous models.
Yueen Ma 0001, Dafeng Chi, Shiguang Wu 0004, Yuecheng Liu, Yuzheng Zhuang, Irwin King
EMNLP5
2025 ET-Plan-Bench: Embodied Task-level Planning Benchmark Towards Spatial-Temporal Cognition with Foundation Models
abstract
Recent advancements in Large Language Models (LLMs) have catalyzed numerous efforts to apply these technologies to embodied tasks, with a particular focus on high-level task planning and task decomposition. LLMs face challenges in understanding the physical world, especially regarding spatial, temporal, and causal relationships among objects and actions. Moreover, the current benchmarks for evaluating these relationships are limited. To further investigate this domain, we introduce a novel embodied task planning benchmark, ET-Plan-Bench. This benchmark features a controllable and diverse array of embodied tasks, varying in levels of difficulty and complexity. It is designed to evaluate two critical dimensions of LLMs’ application in embodied task understanding: spatial understanding (including relation constraints and occlusion of target objects) and temporal and causal comprehension of sequences of actions within an environment. Utilizing multi-source simulators as the backend simulator, ET-Plan-Bench provides immediate environmental feedback to LLMs, enabling dynamic interaction with the environment and the capacity for re-planning as necessary. We evaluated state-of-the-art open-source and closed-source foundational models, including GPT-4, Llama, and Mistral, using our proposed benchmark. While these models perform adequately on simple navigation tasks, their performance significantly deteriorates when con-fronted with tasks that demand a deeper understanding of spatial, temporal, and causal relationships. Consequently, our benchmark distinguishes itself as a large-scale, quantifiable, highly automated, and fine-grained diagnostic framework that presents a substantial challenge to the latest foundational models. We hope it will inspire and propel further research in embodied task planning utilizing foundational models. Code available at: https://github.com/ET-Plan-Bench/ET-Plan-Bench
Yuening Wang, Hongjian Gu, Atia Hamidizadeh, Zhanguang Zhang, Yuecheng Liu, David Gamaliel Arcos Bravo, Junyi Dong, Shunbo Zhou, Tongtong Cao, Xingyue Quan, Yuzheng Zhuang, Yingxue Zhang 0001, Jianye Hao
IROS13
2025 MineAnyBuild: Benchmarking Spatial Planning for Open-world AI Agents
abstract
Spatial Planning is a crucial part in the field of spatial intelligence, which requires the understanding and planning about object arrangements in space perspective. AI agents with the spatial planning ability can better adapt to various real-world applications, including robotic manipulation, automatic assembly, urban planning etc. Recent works have attempted to construct benchmarks for evaluating the spatial intelligence of Multimodal Large Language Models (MLLMs). Nevertheless, these benchmarks primarily focus on spatial reasoning based on typical Visual Question-Answering (VQA) forms, which suffers from the gap between abstract spatial understanding and concrete task execution. In this work, we take a step further to build a comprehensive benchmark called MineAnyBuild, aiming to evaluate the spatial planning ability of open-world AI agents in the Minecraft game. Specifically, MineAnyBuild requires an agent to generate executable architecture building plans based on the given multi-modal human instructions. It involves 4,000 curated spatial planning tasks and also provides a paradigm for infinitely expandable data collection by utilizing rich player-generated content. MineAnyBuild evaluates spatial planning through four core supporting dimensions: spatial understanding, spatial reasoning, creativity, and spatial commonsense. Based on MineAnyBuild, we perform a comprehensive evaluation for existing MLLM-based agents, revealing the severe limitations but enormous potential in their spatial planning abilities. We believe our MineAnyBuild will open new avenues for the evaluation of spatial intelligence and help promote further development for open-world AI agents capable of spatial planning.
Ziming Wei 0001, Bingqian Lin, Zijian Jiao, Yunshuang Nie, Yuecheng Liu, Yuzheng Zhuang, Xiaodan Liang
NeurIPS7
2024 Generate Subgoal Images Before Act: Unlocking the Chain-of-Thought Reasoning in Diffusion Model for Robot Manipulation with Multimodal Prompts
abstract
Robotics agents often struggle to understand and follow the multi-modal prompts in complex manipulation scenes which are challenging to be sufficiently and accurately described by text alone. Moreover, for long-horizon manipulation tasks, the deviation from general instruction tends to accumulate if lack of intermediate guidance from high-level subgoals. For this, we consider can we generate subgoal images before act to enhance the instruction following in long-horizon manipulation with multi-modal prompts? Inspired by the great success of diffusion model in image generation tasks, we propose a novel hierarchical framework named as CoTDiffusion that incorporates diffusion model as a high-level planner to convert the general and multimodal prompts into coherent visual subgoal plans, which further guide the low-level policy model before action execution. We design a semantic alignment module that can anchor the progress of generated keyframes along a coherent generation chain, unlocking the chain-of-thought reasoning ability of diffusion model. Additionally, we propose bi-directional generation and frame concat mechanism to further enhance the fidelity of generated subgoal images and the accuracy of instruction following. The experiments cover various robotics manipulation scenarios including visual reasoning, visual rearrange, and visual constraints. CoTDiffusion achieves outstanding performance gain compared to the baselines without explicit subgoal generation, which proves that a subgoal image is worth a thousand words of instruction. The details and visualizations are available at https://cotdiffusion.github.io.
Fei Ni 0001, Jianye Hao, Shiguang Wu 0001, Longxin Kou, Jiashun Liu, Yan Zheng 0002, Bin Wang 0034, Yuzheng Zhuang
CVPR8
2024 Multi-Agent Trajectory Prediction with Scalable Diffusion Transformer
abstract
Accurate prediction of multi-agent spatiotemporal systems is critical to various real-world applications, such as autonomous driving, sports, and multiplayer games.Unfortunately, modeling multiagent trajectories is challenging due to its complicated, interactive, and multi-modal nature.Recently, diffusion models have achieved great success in modeling multi-modal distribution and trajectory generation, showing promising ability in resolving this problem.Motivated by this, in this paper, we propose a novel multi-agent trajectory prediction framework, dubbed Scalable Diffusion Transformer (SDT), which is naturally designed to learn the complicated distribution and implicit interactions among agents.We evaluate SDT on a set of real-world benchmark datasets and compare it with representative baseline methods, which demonstrates the better multi-agent trajectory prediction ability of SDT in terms of accuracy and diversity.
Shenyu Zhang 0001, Shixiong Kai, Chang Chen 0015, Yuzheng Zhuang, Zhengbang Zhu, Minghuan Liu, Weinan Zhang 0001
DAI4
2024 SCALE: Self-Correcting Visual Navigation for Mobile Robots via Anti-Novelty Estimation
abstract
Although visual navigation has been extensively studied using deep reinforcement learning, online learning for real-world robots remains a challenging task. Recent work directly learned from offline dataset to achieve broader generalization in the real-world tasks, which, however, faces the out-of-distribution (OOD) issue and potential robot localization failures in a given map for unseen observation. This significantly drops the success rates and even induces collision. In this paper, we present a self-correcting visual navigation method, SCALE, that can autonomously prevent the robot from the OOD situations without human intervention. Specifically, we develop an image-goal conditioned offline reinforcement learning method based on implicit Q-learning (IQL). When facing OOD observation, our novel localization recovery method generates the potential future trajectories by learning from the navigation affordance, and estimates the future novelty via random network distillation (RND). A tailored cost function searches for the candidates with the least novelty that can lead the robot to the familiar places. We collect offline data and conduct evaluation experiments in three real-world urban scenarios. Experiment results show that SCALE outperforms the previous state-of-the-art methods for open-world navigation with a unique capability of localization recovery, significantly reducing the need for human intervention. Code is available at https://github.com/KubeEdge4Robotics/ScaleNav.
Yuecheng Liu, Yuzheng Zhuang, Sitong Mao, Shunbo Zhou
ICRA3
2024 Articulated Object Manipulation with Coarse-to-fine Affordance for Mitigating the Effect of Point Cloud Noise
abstract
3D articulated objects are inherently challenging for manipulation due to the varied geometries and intricate functionalities associated with articulated objects. Point-level affordance, which predicts the per-point actionable score and thus proposes the best point to interact with, has demonstrated excellent performance and generalization capabilities in articulated object manipulation. However, a significant challenge remains: while previous works use perfect point cloud generated in simulation, the models cannot directly apply to the noisy point cloud in the real-world. To tackle this challenge, we leverage the property of real-world scanned point cloud that, the point cloud becomes less noisy when the camera is closer to the object. Therefore, we propose a novel coarse-to-fine affordance learning pipeline to mitigate the effect of point cloud noise in two stages. In the first stage, we learn the affordance on the noisy far point cloud which includes the whole object to propose the approximated place to manipulate. Then, we move the camera in front of the approximated place, scan a less noisy point cloud containing precise local geometries for manipulation, and learn affordance on such point cloud to propose fine-grained final actions. The proposed method is thoroughly evaluated both using large-scale simulated noisy point clouds mimicking real-world scans, and in the real world scenarios, with superiority over existing methods, demonstrating the effectiveness in tackling the noisy real-world point cloud problem.
Suhan Ling, Ruihai Wu, Shiguang Wu 0004, Yuzheng Zhuang, Yu Li 0022, Chang Liu 0077, Hao Dong 0003
ICRA5
2024 PERIA: Perceive, Reason, Imagine, Act via Holistic Language and Vision Planning for Manipulation
abstract
Long-horizon manipulation tasks with general instructions often implicitly encapsulate multiple sub-tasks, posing significant challenges in instruction following. While language planning is a common approach to decompose general instructions into stepwise sub-instructions, text-only guidance may lack expressiveness and lead to potential ambiguity. Considering that humans often imagine and visualize sub-instructions reasoning out before acting, the imagined subgoal images can provide more intuitive guidance and enhance the reliability of decomposition. Inspired by this, we propose **PERIA**(**PE**rceive, **R**eason, **I**magine, **A**ct), a novel framework that integrates holistic language planning and vision planning for long-horizon manipulation tasks with complex instructions, leveraging both logical and intuitive aspects of task decomposition. Specifically, we first perform a lightweight multimodal alignment on the encoding side to empower the MLLM to perceive visual details and language instructions. The MLLM is then jointly instruction-tuned with a pretrained image-editing model to unlock capabilities of simultaneous reasoning of language instructions and generation of imagined subgoals. Furthermore, we introduce a consistency alignment loss to encourage coherent subgoal images and align with their corresponding instructions, mitigating potential hallucinations and semantic conflicts between the two planning manners. Comprehensive evaluations across three task domains demonstrate that PERIA, benefiting from holistic language and vision planning, significantly outperforms competitive baselines in both instruction following accuracy and task success rate on complex manipulation tasks.
Fei Ni 0001, Jianye Hao, Shiguang Wu 0001, Longxin Kou, Yifu Yuan, Zibin Dong, Jinyi Liu 0002, Mingzhi Li, Yuzheng Zhuang, Yan Zheng 0002
NeurIPS9
2024 Prototypical Context-Aware Dynamics for Generalization in Visual Control With Model-Based Reinforcement Learning
abstract
The latent world model, which efficiently represents high-dimensional observations within a latent space, has shown promise in reinforcement learning-based policies for visual control tasks. Due to a lack of clear environmental context comprehension, its applicability in a variety of contexts with unknown dynamics is constrained. We propose a prototypical context- aware dynamics (ProtoCAD) model to address this issue. This model captures local dynamics using temporally consistent latent contexts and aids generalization in visual control tasks. By grouping prototypes over historical experiences, ProtoCAD collects useful contextual information that improves model-based reinforcement learning dynamics generalization in two ways. First, to guarantee the consistency of prototype assignments for various temporal segments of the same latent trajectory, a temporally consistent prototypes regularizer is used. Then, a context representation is devised to combine the aggregated prototype with the projection embedding of latent states. According to extensive trials, ProtoCAD outperforms competing approaches in terms of dynamics generalization for visual robotic control and autonomous driving applications.
Yao Mu 0001, Dong Li 0016, Dongbin Zhao, Yuzheng Zhuang, Ping Luo 0002, Bin Wang 0034, Jianye Hao
IEEE Trans. Ind. Informatics6
2023 RITA: Boost Driving Simulators with Realistic Interactive Traffic Flow
abstract
High-quality traffic flow generation is the core module in building simulators for autonomous driving. However, the majority of available simulators are incapable of replicating traffic patterns that accurately reflect the various features of real-world data while also simulating human-like reactive responses to the tested autopilot driving strategies. Taking one step forward to addressing such a problem, we propose Realistic Interactive TrAffic flow (RITA) as an integrated component of existing driving simulators to provide high-quality traffic flow for the evaluation and optimization of the tested driving strategies. RITA is developed with consideration of three key features, i.e., fidelity, diversity, and controllability, and consists of two core modules called RITABackend and RITAKit. RITABackend is built to support vehicle-wise control and provide traffic generation models from real-world datasets, while RITAKit is developed with easy-to-use interfaces for controllable traffic generation via RITABackend. We demonstrate RITA’s capacity to create diversified and high-fidelity traffic simulations in several highly interactive highway scenarios. The experimental findings demonstrate that our produced RITA traffic flows exhibit all three key features, hence enhancing the completeness of driving strategy evaluation. Moreover, we showcase the possibility for further improvement of baseline strategies through online fine-tuning with RITA traffic flows.
Zhengbang Zhu, Shenyu Zhang 0001, Yuzheng Zhuang, Yuecheng Liu, Minghuan Liu, Ziqing Gong, Shixiong Kai, Qiang Gu, Bin Wang 0034, Siyuan Cheng 0012, Xinyu Wang 0001, Jianye Hao, Yong Yu 0001
DAI3
2023 Learnable Behavior Control: Breaking Atari Human World Records via Sample-Efficient Behavior Selection
Jiajun Fan, Yuzheng Zhuang, Yuecheng Liu, Jianye Hao, Bin Wang 0034, Jiangcheng Zhu, Hao Wang 0049, Shutao Xia
ICLR2
2022 Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization
abstract
Recent progress in state-only imitation learning extends the scope of applicability of imitation learning to real-world settings by relieving the need for observing expert actions. However, existing solutions only learn to extract a state-to-action mapping policy from the data, without considering how the expert plans to the target. This hinders the ability to leverage demonstrations and limits the flexibility of the policy. In this paper, we introduce Decoupled Policy Optimization (DePO), which explicitly decouples the policy as a high-level state planner and an inverse dynamics model. With embedded decoupled policy gradient and generative adversarial training, DePO enables knowledge transfer to different action spaces or state transition dynamics, and can generalize the planner to out-of-demonstration state regions. Our in-depth experimental analysis shows the effectiveness of DePO on learning a generalized target state planner while achieving the best imitation performance. We demonstrate the appealing usage of DePO for transferring across different tasks by pre-training, and the potential for co-training agents with various skills.
Minghuan Liu, Zhengbang Zhu, Yuzheng Zhuang, Weinan Zhang 0001, Jianye Hao, Yong Yu 0001, Jun Wang 0012
ICML3
2022 DOMINO: Decomposed Mutual Information Optimization for Generalized Context in Meta-Reinforcement Learning
abstract
Adapting to the changes in transition dynamics is essential in robotic applications. By learning a conditional policy with a compact context, context-aware meta-reinforcement learning provides a flexible way to adjust behavior according to dynamics changes. However, in real-world applications, the agent may encounter complex dynamics changes. Multiple confounders can influence the transition dynamics, making it challenging to infer accurate context for decision-making. This paper addresses such a challenge by decomposed mutual information optimization (DOMINO) for context learning, which explicitly learns a disentangled context to maximize the mutual information between the context and historical trajectories while minimizing the state transition prediction error. Our theoretical analysis shows that DOMINO can overcome the underestimation of the mutual information caused by multi-confounded challenges via learning disentangled context and reduce the demand for the number of samples collected in various environments. Extensive experiments show that the context learned by DOMINO benefits both model-based and model-free reinforcement learning algorithms for dynamics generalization in terms of sample efficiency and performance in unseen environments.
Yao Mu 0001, Yuzheng Zhuang, Fei Ni 0001, Bin Wang 0034, Jianyu Chen 0002, Jianye Hao, Ping Luo 0002
NeurIPS2
2021 Reinforcement Learning based Negotiation-aware Motion Planning of Autonomous Vehicles
abstract
For autonomous vehicles integrating onto road-ways with human traffic participants, it requires understanding and adapting to the participants’ intention by responding in predictable ways. This paper proposes a reinforcement learning based negotiation-aware motion planning framework, which adopts RL to adjust the driving style of the planner by dynamically modifying the prediction horizon length of the motion planner in real time adaptively. The framework models the interaction between the autonomous vehicle and other traffic participants as a Markov Decision Process. A temporal sequence of occupancy grid maps are taken as inputs for RL module to embed an implicit intention reasoning. Curriculum learning is employed to enhance the training efficiency and the robustness of the algorithm. We applied our method to narrow lane navigation in both simulation and real world to demonstrate that the proposed method outperforms the common alternative due to its advantage in alleviating the social dilemma problem with proper negotiation skills.
Zhitao Wang, Yuzheng Zhuang, Qiang Gu, Wulong Liu
IROS2
2021 Model-Based Reinforcement Learning via Imagination with Derived Memory
abstract
Model-based reinforcement learning aims to improve the sample efficiency of policy learning by modeling the dynamics of the environment. Recently, the latent dynamics model is further developed to enable fast planning in a compact space. It summarizes the high-dimensional experiences of an agent, which mimics the memory function of humans. Learning policies via imagination with the latent model shows great potential for solving complex tasks. However, only considering memories from the true experiences in the process of imagination could limit its advantages. Inspired by the memory prosthesis proposed by neuroscientists, we present a novel model-based reinforcement learning framework called Imagining with Derived Memory (IDM). It enables the agent to learn policy from enriched diverse imagination with prediction-reliability weight, thus improving sample efficiency and policy robustness. Experiments on various high-dimensional visual control tasks in the DMControl benchmark demonstrate that IDM outperforms previous state-of-the-art methods in terms of policy robustness and further improves the sample efficiency of the model-based method.
Yao Mu 0001, Yuzheng Zhuang, Bin Wang 0034, Guangxiang Zhu, Wulong Liu, Jianyu Chen 0002, Ping Luo 0002, Shengbo Eben Li, Chongjie Zhang, Jianye Hao
NeurIPS2
2020 Multi-Agent Interactions Modeling with Correlated Policies
Minghuan Liu, Ming Zhou 0006, Weinan Zhang 0001, Yuzheng Zhuang, Jun Wang 0012, Wulong Liu, Yong Yu 0001
ICLR4
2020 Triple-GAIL: A Multi-Modal Imitation Learning Framework with Generative Adversarial Nets
abstract
Generative adversarial imitation learning (GAIL) has shown promising results by taking advantage of generative adversarial nets, especially in the field of robot learning. However, the requirement of isolated single modal demonstrations limits the scalability of the approach to real world scenarios such as autonomous vehicles' demand for a proper understanding of human drivers' behavior. In this paper, we propose a novel multi-modal GAIL framework, named Triple-GAIL, that is able to learn skill selection and imitation jointly from both expert demonstrations and continuously generated experiences with data augmentation purpose by introducing an auxiliary selector. We provide theoretical guarantees on the convergence to optima for both of the generator and the selector respectively. Experiments on real driver trajectories and real-time strategy game datasets demonstrate that Triple-GAIL can better fit multi-modal behaviors close to the demonstrators and outperforms state-of-the-art methods.
Cong Fei, Bin Wang 0034, Yuzheng Zhuang, Zongzhang Zhang, Jianye Hao, Xuewu Ji, Wulong Liu
IJCAI3
2016 HiGene: A high-performance platform for genomic data analysis
abstract
Post-sequencing genomic data analysis becomes a major challenge while next-generation sequencing technologies evolve by leaps and bounds. The data-intensive and compute-intensive nature of genome analysis makes cluster computing an attractive choice for building efficient solutions. This paper presents HiGene, a high-performance genome analysis platform that exploits big data technology to revolutionize genomics data crunching power. HiGene reconstructs the genome analysis pipeline by exploiting both multi-core and multi-node parallelization using Apache Spark, and employs two key techniques to further boost the performance. First, a dynamic computing resource re-allocator is implemented, which allows flexible on-demand resource allocation for operations inside tasks. Second, an efficient skew mitigation approach is proposed, which automatically identifies and resolves data skew and computation skew through task repartitioning and resource reallocating respectively. HiGene has been evaluated with a whole human genome dataset on a 10-node Huawei 5885 cluster. Experimental results show that HiGene achieves remarkable high performance that reduces the total running time on a whole genome sequence dataset from days to nearly one hour. Furthermore, it is two times faster than state-of-the-art cluster based approaches.
Liqun Deng, Guowei Huang 0002, Yuzheng Zhuang, Jiansheng Wei, Youliang Yan
BIBM3