← Search

Yuheng Zhang

16 accepted papers

2026

ArtVIP: Articulated Digital Assets of Visual Realism, Modular Interaction, and Physical Fidelity for Robot Learning

ICLR 2026poster

Robot learning increasingly relies on simulation to advance complex ability such as dexterous manipulations and precise interactions, necessitating high-quality digital assets to bridge the sim-to-real gap. However, existing open-source articulated object datasets for simulation are limited by insuf…

Cited by 0SourceScholar
2026

Offline Two-Player Zero-Sum Markov Games with KL Regularization

ICML 2026poster

We study the problem of learning Nash equilibria in offline two-player zero-sum Markov games. While existing approaches often rely on explicit pessimism to address distribution shift, we show that KL regularization alone suffices to stabilize learning and guarantee convergence. We first introduce Re…

Cited by 0SourceScholar
2026

ProOOD: Prototype-Guided Out-of-Distribution 3D Occupancy Prediction

CVPR 2026

3D semantic occupancy prediction is central to autonomous driving, yet current methods are vulnerable to long-tailed class bias and out-of-distribution (OOD) inputs, often overconfidently assigning anomalies to rare classes. We present ProOOD, a lightweight, plug-and-play method that couples prototy

Cited by 0SourcecodeScholar
2025

Improving LLM General Preference Alignment via Optimistic Online Mirror Descent

NeurIPS 2025spotlight

Reinforcement learning from human feedback (RLHF) has demonstrated remarkable effectiveness in aligning large language models (LLMs) with human preferences. Many existing alignment approaches rely on the Bradley-Terry (BT) model assumption, which assumes the existence of a ground-truth reward for ea…

Cited by 0SourceScholar
2025

Iterative Nash Policy Optimization: Aligning LLMs with General Preferences via No-Regret Learning

ICLR 2025oral

Reinforcement Learning with Human Feedback (RLHF) has achieved great success in aligning large language models (LLMs) with human preferences. Prevalent RLHF approaches are reward-based, following the Bradley-Terry (BT) model assumption, which may not fully capture the complexity of human preferences…

Cited by 4SourcePDFScholar
2025

Statistical Tractability of Off-policy Evaluation of History-dependent Policies in POMDPs

ICLR 2025poster

We investigate off-policy evaluation (OPE), a central and fundamental problem in reinforcement learning (RL), in the challenging setting of Partially Observable Markov Decision Processes (POMDPs) with large observation spaces. Recent works of Uehara et al. (2023a); Zhang & Jiang (2024) developed a m…

Cited by 0SourcePDFScholar
2024

Efficient Contextual Bandits with Uninformed Feedback Graphs

ICML 2024poster

Bandits with feedback graphs are powerful online learning models that interpolate between the full information and classic bandit problems, capturing many real-life applications. A recent work by [Zhang et al., 2023] studies the contextual version of this problem and proposes an efficient and optima…

Cited by 3SourcePDFScholar
2024

On the Curses of Future and History in Future-dependent Value Functions for Off-policy Evaluation

NeurIPS 2024poster

We study off-policy evaluation (OPE) in partially observable environments with complex observations, with the goal of developing estimators whose guarantee avoids exponential dependence on the horizon. While such estimators exist for MDPs and POMDPs can be converted to history-based MDPs, their esti…

Cited by 4SourcePDFScholar
2024

Online Iterative Reinforcement Learning from Human Feedback with General Preference Model

NeurIPS 2024poster

We investigate Reinforcement Learning from Human Feedback (RLHF) in the context of a general preference oracle. In particular, we do not assume the existence of a reward function and an oracle preference signal drawn from the Bradley-Terry model as most of the prior works do. We consider a standard…

2024

Provably Efficient Interactive-Grounded Learning with Personalized Reward

NeurIPS 2024poster

Interactive-Grounded Learning (IGL) [Xie et al., 2021] is a powerful framework in which a learner aims at maximizing unobservable rewards through interacting with an environment and observing reward-dependent feedback on the taken actions. To deal with personalized rewards that are ubiquitous in app…

Cited by 0SourcePDFScholar
2023

Practical Contextual Bandits with Feedback Graphs

NeurIPS 2023poster

While contextual bandit has a mature theory, effectively leveraging different feedback patterns to enhance the pace of learning remains unclear. Bandits with feedback graphs, which interpolates between the full information and bandit regimes, provides a promising framework to mitigate the statistica…

Cited by 5SourcePDFScholar
2022

Batch Active Learning with Graph Neural Networks via Multi-Agent Deep Reinforcement Learning

AAAI 2022technical

Graph neural networks (GNNs) have achieved tremendous success in many graph learning tasks such as node classification, graph classification and link prediction. For the classification task, GNNs' performance often highly depends on the number of labeled nodes and thus could be significantly hampere…

Cited by 25SourcePDFScholar
2022

Improved Algorithms for Neural Active Learning

NeurIPS 2022accept

We improve the theoretical and empirical performance of neural-network(NN)-based active learning algorithms for the non-parametric streaming setting. In particular, we introduce two regret metrics by minimizing the population loss that are more suitable in active learning than the one used in state-…

2021

Improving Robustness to Model Inversion Attacks via Mutual Information Regularization

AAAI 2021technical

This paper studies defense mechanisms against model inversion (MI) attacks -- a type of privacy attacks aimed at inferring information about the training data distribution given the access to a target machine learning model. Existing defense mechanisms rely on model-specific heuristics or noise inje…

Cited by 99SourcePDFScholar
2020

The Secret Revealer: Generative Model-Inversion Attacks Against Deep Neural Networks

CVPR 2020oral

This paper studies model-inversion attacks, in which the access to a model is abused to infer information about the training data. Since its first introduction by [??], such attacks have raised serious concerns given that training data usually contain privacy sensitive information. Thus far, success…

Cited by 614PDFcodeScholar