← Search

Muning Wen

13 accepted papers

2026

Towards Cold-Start Drafting and Continual Refining: A Value-Driven Memory Approach with Application to NPU Kernel Synthesis

ICML 2026poster

Deploying Large Language Models to data-scarce programming domains poses significant challenges, particularly for kernel synthesis on emerging Domain-Specific Architectures where a "Data Wall" limits available training data. While models excel on data-rich platforms like CUDA, they suffer catastroph…

Cited by 0SourceScholar
2025

Autonomous Goal Detection and Cessation in Reinforcement Learning: A Case Study on Source Term Estimation

AAAI 2025technical

Reinforcement Learning has revolutionized decision-making processes in dynamic environments, yet it often struggles with autonomously detecting and achieving goals without clear feedback signals. For example, in a Source Term Estimation problem, the lack of precise environmental information makes it…

Cited by 3SourcePDFScholar
2025

HammerBench: Fine-Grained Function-Calling Evaluation in Real Mobile Assistant Scenarios

ACL 2025finding

Evaluating the performance of LLMs in multi-turn human-agent interactions presents significant challenges, particularly due to the complexity and variability of user behavior. In this paper, we introduce HammerBench, a novel benchmark framework for assessing LLMs’ function-calling capabilities in re…

2025

Learning Humanoid Standing-up Control across Diverse Postures

RSS 2025poster

Standing-up control is crucial for humanoid robots, with the potential for integration into current locomotion and loco-manipulation systems. Existing approaches are either limited to simulations that neglect hardware constraints or rely on predefined ground-specific motion trajectories, failing to…

Cited by 6PDFScholar
2025

MobileUse: A Hierarchical Reflection-Driven GUI Agent for Autonomous Mobile Operation

NeurIPS 2025poster

Recent advances in Multimodal Large Language Models (MLLMs) have enabled the development of mobile agents that can understand visual inputs and follow user instructions, unlocking new possibilities for automating complex tasks on mobile devices. However, applying these models to real-world mobile sc…

Cited by 0SourcecodeScholar
2025

Robust Function-Calling for On-Device Language Model via Function Masking

ICLR 2025spotlight

Large language models have demonstrated impressive value in performing as autonomous agents when equipped with external tools and API calls. Nonetheless, effectively harnessing their potential for executing complex tasks crucially relies on enhancements in their function-calling capabilities. This p…

Cited by 1SourcePDFScholar
2025

Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learning

ICLR 2025poster

Driven by inherent uncertainty and the sim-to-real gap, robust reinforcement learning (RL) seeks to improve resilience against the complexity and variability in agent-environment sequential interactions. Despite the existence of a large number of RL benchmarks, there is a lack of standardized benchm…

Cited by 1SourcePDFScholar
2024

AlphaZero-Like Tree-Search can Guide Large Language Model Decoding and Training

ICML 2024poster

Recent works like Tree-of-Thought (ToT) and Reasoning via Planning (RAP) aim to augment the multi-step reasoning capabilities of LLMs by using tree-search algorithms. These methods rely on prompting a pre-trained model to serve as a value function and focus on problems with low search depth. As a re…

2024

Reinforcing LLM Agents via Policy Optimization with Action Decomposition

NeurIPS 2024poster

Language models as intelligent agents push the boundaries of sequential decision-making agents but struggle with limited knowledge of environmental dynamics and exponentially huge action space. Recent efforts like GLAM and TWOSOME manually constrain the action space to a restricted subset and employ…

2022

Multi-Agent Reinforcement Learning is a Sequence Modeling Problem

NeurIPS 2022accept

Large sequence models (SM) such as GPT series and BERT have displayed outstanding performance and generalization capabilities in natural language process, vision and recently reinforcement learning. A natural follow-up question is how to abstract multi-agent decision making also as an sequence model…

2022

Trust Region Policy Optimisation in Multi-Agent Reinforcement Learning

ICLR 2022poster

Trust region methods rigorously enabled reinforcement learning (RL) agents to learn monotonically improving policies, leading to superior performance on a variety of tasks. Unfortunately, when it comes to multi-agent reinforcement learning (MARL), the property of monotonic improvement may not simpl…

Cited by 329SourcePDFScholar
2021

Settling the Variance of Multi-Agent Policy Gradients

NeurIPS 2021poster

Policy gradient (PG) methods are popular reinforcement learning (RL) methods where a baseline is often applied to reduce the variance of gradient estimates. In multi-agent RL (MARL), although the PG theorem can be naturally extended, the effectiveness of multi-agent PG (MAPG) methods degrades as th…