← Search

Yuhui Wang

20 accepted papers

2026

Efficient Morphology–Control Co-Design via Stackelberg PPO under Non-Differentiable Leader–Follower Interfaces

ICLR 2026poster

Morphology-control co-design concerns the coupled optimization of an agent’s body structure and control policy. A key challenge is that evaluating each candidate morphology requires extensive rollouts to re-optimize control and assess quality, leading to high computational costs and slow convergence…

Cited by 0SourcecodeScholar
2026

MetaAct-RL: Training Language Models for Reasoning Through Meta-Action-Based Reinforcement Learning

AAAI 2026technical

Outcome-based reinforcement learning has made notable advances in training language models (LMs) for reasoning. However, without explicit incentives and controls, this paradigm has limitations and instability in eliciting high-quality reasoning trajectories with diverse actions—particularly for mode

Cited by 0SourcePDFScholar
2026

Reasoning or Retrieval? A Study of Answer Attribution on Large Reasoning Models

ICLR 2026poster

Large reasoning models (LRMs) exhibit unprecedented capabilities in solving complex problems through Chain-of-Thought (CoT) reasoning. However, recent studies reveal that their final answers often contradict their own reasoning traces. We hypothesize that this inconsistency stems from two competing…

Cited by 0SourceScholar
2025

Directly Forecasting Belief for Reinforcement Learning with Delays

ICML 2025poster

Reinforcement learning (RL) with delays is challenging as sensory perceptions lag behind the actual events: the RL agent needs to estimate the real state of its environment based on past observations. State-of-the-art (SOTA) methods typically employ recursive, step-by-step forecasting of states. Thi…

2025

LLMEval-Med: A Real-world Clinical Benchmark for Medical LLMs with Physician Validation

EMNLP 2025

Evaluating large language models (LLMs) in medicine is crucial because medical applications require high accuracy with little room for error. Current medical benchmarks have three main types: medical exam-based, comprehensive medical, and specialized assessments. However, these benchmarks have limit

2025

PFDial: A Structured Dialogue Instruction Fine-tuning Method Based on UML Flowcharts

ACL 2025finding

Process-driven dialogue systems, which operate under strict predefined process constraints, are essential in customer service and equipment maintenance scenarios. Although Large Language Models (LLMs) have shown remarkable progress in dialogue and reasoning, they still struggle to solve these strict…

2025

Parrot: A Training Pipeline Enhances Both Program CoT and Natural Language CoT for Reasoning

EMNLP 2025

Natural language chain-of-thought (N-CoT) and Program chain-of-thought (P-CoT) have emerged as two primary paradigms for large language models (LLMs) to solve mathematical reasoning problems. Current research typically endeavors to achieve unidirectional enhancement: P-CoT enhanced N-CoT or N-CoT en

2025

RobustKV: Defending Large Language Models against Jailbreak Attacks via KV Eviction

ICLR 2025poster

Jailbreak attacks circumvent LLMs' built-in safeguards by concealing harmful queries within adversarial prompts. While most existing defenses attempt to mitigate the effects of adversarial prompts, they often prove inadequate as adversarial prompts can take arbitrary, adaptive forms. This paper intr…

Cited by 4SourcePDFScholar
2025

Scaling Value Iteration Networks to 5000 Layers for Extreme Long-Term Planning

ICML 2025poster

The Value Iteration Network (VIN) is an end-to-end differentiable neural network architecture for planning. It exhibits strong generalization to unseen domains by incorporating a differentiable planning module that operates on a latent Markov Decision Process (MDP). However, VINs struggle to scale t…

Cited by 1SourcePDFScholar
2024

Boosting Reinforcement Learning with Strongly Delayed Feedback Through Auxiliary Short Delays

ICML 2024poster

Reinforcement learning (RL) is challenging in the common case of delays between events and their sensory perceptions. State-of-the-art (SOTA) state augmentation techniques either suffer from state space explosion or performance degeneration in stochastic environments. To address these challenges, we…

2024

Variational Delayed Policy Optimization

NeurIPS 2024spotlight

In environments with delayed observation, state augmentation by including actions within the delay window is adopted to retrieve Markovian property to enable reinforcement learning (RL). Whereas, state-of-the-art (SOTA) RL techniques with Temporal-Difference (TD) learning frameworks commonly suffer…

2023

DropIT: Dropping Intermediate Tensors for Memory-Efficient DNN Training

ICLR 2023poster

A standard hardware bottleneck when training deep neural networks is GPU memory. The bulk of memory is occupied by caching intermediate tensors for gradient computation in the backward pass. We propose a novel method to reduce this footprint - Dropping Intermediate Tensors (DropIT). DropIT drops mi…

2023

Learning to Identify Critical States for Reinforcement Learning from Videos

ICCV 2023poster

Recent work on deep reinforcement learning (DRL) has pointed out that algorithmic information about good policies can be extracted from offline data which lack explicit information about executed actions. For example, videos of humans or robots may convey a lot of implicit information about rewardin…

Cited by 12PDFcodeScholar