← Search

Kwok Yan Lam

5 accepted papers

2026

Failure-Driven Workflow Refinement

ICML 2026spotlight

Workflow optimization for tool-using LLM agents is often cast as global search over candidate graphs, scored by a scalar metric. This collapses rich, multi-step failure traces into binary outcomes, obscuring recurring failure structure and making refinement inefficient. We reframe optimization as \e…

Cited by 0SourceScholar
2026

RedVisor: Reasoning-Aware Prompt Injection Defense via Zero-Copy KV Cache Reuse

ICML 2026poster

Large Language Models (LLMs) are increasingly vulnerable to *Prompt Injection (PI)* attacks, where adversarial instructions hidden within retrieved contexts hijack the model's execution flow. Current defenses typically face a critical trade-off: *prevention-based* fine-tuning often degrades general …

Cited by 0SourceScholar
2026

SOLAR for Offline MARL: Plateau-Triggered Potential Shaping under World-Model Uncertainty

ICML 2026poster

Reward shaping can accelerate reinforcement learning, but in sparse-reward \emph{offline} multi-agent RL it is often brittle: dense intrinsic rewards may alter the underlying Markov game, while world-model guidance can amplify model bias. We find that shaping becomes reliable when it is (i) activate…

Cited by 0SourceScholar
2026

When Search Goes Wrong: Red-Teaming Web-Augmented Large Language Models

ICML 2026poster

Large Language Models (LLMs) have been augmented with web search to overcome the limitations of the static knowledge boundary by accessing up-to-date information from the open Internet. While this integration enhances model capability, it also introduces a distinct safety threat surface: the retriev…

Cited by 0SourceScholar