← Search

Yihao Sun

14 accepted papers

2026

ADM-v2: Pursuing Full-Horizon Roll-out in Dynamics Models for Offline Policy Learning and Evaluation

ICLR 2026poster

Model-based methods for offline Reinforcement Learning transfer extensive policy exploration and evaluation to data-driven dynamics models, effectively saving real-world samples in the offline setting. We expect the dynamics model to allow the policy to roll out full-horizon episodes, which is cruci…

Cited by 0SourcecodeScholar
2026

Hierarchical Value-Decomposed Offline Reinforcement Learning for Whole-Body Control

ICLR 2026poster

Scaling imitation learning to high-DoF whole-body robots is fundamentally limited by the \textbf{curse of dimensionality} and the prohibitive cost of collecting expert demonstrations. We argue that the core bottleneck is paradigmatic: real-world supervision for whole-body control is inherently imper…

Cited by 0SourceScholar
2026

Speedup Patch: Learning a Plug-and-Play Policy to Accelerate Embodied Manipulation

ICML 2026poster

While current embodied policies exhibit remarkable manipulation skills, their execution remains unsatisfactorily slow as they inherit the tardy pacing of human demonstrations. Existing acceleration methods typically require policy retraining or costly online interactions, limiting their scalability …

Cited by 0SourceScholar
2026

Towards Practical World Model-based Reinforcement Learning for Vision-Language-Action Models

ICML 2026poster

Vision-Language-Action (VLA) models show strong generalization for robotic control, but finetuning them with reinforcement learning (RL) is constrained by the high cost and safety risks of real-world interaction. Training VLA models in interactive world models avoids these issues but introduces seve…

Cited by 0SourceScholar
2025

Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning

ICLR 2025poster

Model-based methods in reinforcement learning offer a promising approach to enhance data efficiency by facilitating policy exploration within a dynamics model. However, accurately predicting sequential steps in the dynamics model remains a challenge due to the bootstrapping prediction, which attribu…

2025

Emergent Cooperative Strategies for Pursuit-Evasion in Cluttered Environments: A Knowledge-Enhanced Multi-Agent Deep Reinforcement Learning Approach

IROS 2025

Deep reinforcement learning (DRL) has recently emerged as a promising tool for tackling pursuit-evasion tasks. However, most existing DRL-based pursuit approaches still rely on individual rewards and struggle with complex scenarios. To address these challenges, we propose a knowledge-enhanced DRL ap

Cited by 0SourceScholar
2025

Improving Reward Model Generalization from Adversarial Process Enhanced Preferences

ICML 2025poster

In sequential decision-making, the reward function serves as the primary supervision signal, guiding agents to acquire the desired behaviors. Traditional reward modeling methods rely heavily on human expertise, limiting their scalability. Automated preference generation from suboptimal demonstration…

Cited by 0SourcePDFScholar
2024

Assemblage: Automatic Binary Dataset Construction for Machine Learning

NeurIPS 2024poster

Binary code is pervasive, and binary analysis is a key task in reverse engineering, malware classification, and vulnerability discovery. Unfortunately, while there exist large corpuses of malicious binaries, obtaining high-quality corpuses of benign binaries for modern systems has proven challenging…

2024

Episodic Return Decomposition by Difference of Implicitly Assigned Sub-trajectory Reward

AAAI 2024technical

Real-world decision-making problems are usually accompanied by delayed rewards, which affects the sample efficiency of Reinforcement Learning, especially in the extremely delayed case where the only feedback is the episodic reward obtained at the end of an episode. Episodic return decomposition is a…

2024

Flow to Better: Offline Preference-based Reinforcement Learning via Preferred Trajectory Generation

ICLR 2024poster

Offline preference-based reinforcement learning (PbRL) offers an effective solution to overcome the challenges associated with designing rewards and the high costs of online interactions. In offline PbRL, agents are provided with a fixed dataset containing human preferences between pairs of trajecto…

Cited by 20SourcePDFScholar
2024

Policy-conditioned Environment Models are More Generalizable

ICML 2024poster

In reinforcement learning, it is crucial to have an accurate environment dynamics model to evaluate different policies' value in downstream tasks like offline policy optimization and policy evaluation. However, the learned model is known to be inaccurate in predictions when evaluating target policie…

Cited by 2SourcePDFScholar
2024

Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximation

NeurIPS 2024poster

As a prominent category of imitation learning methods, adversarial imitation learning (AIL) has garnered significant practical success powered by neural network approximation. However, existing theoretical studies on AIL are primarily limited to simplified scenarios such as tabular and linear functi…

Cited by 1SourcePDFScholar
2023

Model-Bellman Inconsistency for Model-based Offline Reinforcement Learning

ICML 2023poster

For offline reinforcement learning (RL), model-based methods are expected to be data-efficient as they incorporate dynamics models to generate more data. However, due to inevitable model errors, straightforwardly learning a policy in the model typically fails in the offline setting. Previous studies…