← Search

Zhiwei Shang

4 accepted papers

2026

MASPOB: Bandit-Based Prompt Optimization for Multi-Agent Systems with Graph Neural Networks

ICML 2026spotlight

Large Language Models (LLMs) have achieved significant success across a wide range of tasks, serving as the cognitive backbone for Multi-Agent Systems (MAS) designed to orchestrate complex practical workflows. Given that MAS performance is highly sensitive to input prompts and many deployment scenar…

Cited by 0SourceScholar
2026

T-POP: Test-Time Personalization with Online Preference Feedback

ICML 2026poster

Personalizing large language models (LLMs) to individual user preferences is a critical step beyond generating generically helpful responses. However, current personalization methods are ill-suited for new users, as they typically require either slow, resource-intensive fine-tuning or a substantial …

Cited by 0SourceScholar
2025

Preference Aligned Diffusion Planner for Quadrupedal Locomotion Control

IROS 2025

Diffusion models demonstrate superior performance in capturing complex distributions from large-scale datasets, providing a promising solution for quadrupedal locomotion control. However, the robustness of the diffusion planner is inherently dependent on the diversity of the pre-collected datasets.

Cited by 9SourcecodeScholar