← Search

Yuzhou Nie

7 accepted papers

2026

BlueCodeAgent: A Blue Teaming Agent Powered by Automated Red Teaming for CodeGen AI

ICML 2026poster

Existing research on CodeGen AI security mainly focuses on red teaming, which aims to uncover vulnerabilities and risks in AI-generated code. However, progress on the blue teaming side remains limited, as effective defenses require a deep security analysis of given tasks and edge cases. To fill in t…

Cited by 0SourceScholar
2026

OpenSage: Self-programming Agent Generation Engine

ICML 2026poster

Agent development kits (ADKs) provide effective platforms and tooling for constructing agents, and their designs are critical to the constructed agents' performance, especially the functionality for agent topology, tools, and memory. However, current ADKs either lack sufficient functional support or…

Cited by 0SourceScholar
2026

THETA: Threshold-Based Exclusive Batching for Memory-Bandwidth-Constrained LLM Inference

ICML 2026poster

Chunked prefill has become the dominant scheduling strategy for large language model (LLM) inference, interleaving prefill and decode operations to improve GPU utilization. However, this approach does not universally outperform exclusive batching: on bandwidth-constrained GPUs, mixed batches can int…

Cited by 0SourceScholar
2025

AGENTVIGIL: Automatic Black-Box Red-teaming for Indirect Prompt Injection against LLM Agents

EMNLP 2025

There emerges a critical security risk of LLM agents: indirect prompt injection, a sophisticated attack vector that compromises thecore of these agents, the LLM, by manipulating contextual information rather than direct user prompts. In this work, we propose a generic black-box optimization framewor

Cited by 0SourcePDFScholar
2025

OWL: Optimized Workforce Learning for General Multi-Agent Assistance in Real-World Task Automation

NeurIPS 2025poster

Large Language Model (LLM)-based multi-agent systems show promise for automating real-world tasks but struggle to transfer across domains due to their domain-specific nature. Current approaches face two critical shortcomings: they require complete architectural redesign and full retraining of all co…

Cited by 0SourcecodeScholar
2025

SECODEPLT: A Unified Benchmark for Evaluating the Security Risks and Capabilities of Code GenAI

NeurIPS 2025poster

Existing benchmarks for evaluating the security risks and capabilities (e.g., vulnerability detection) of code-generating large language models (LLMs) face several key limitations: (1) limited coverage of risk and capabilities; (2) reliance on static evaluation metrics such as LLM judgments or rule-…

Cited by 0SourcecodeScholar
2024

When LLM Meets DRL: Advancing Jailbreaking Efficiency via DRL-guided Search

NeurIPS 2024poster

Recent studies developed jailbreaking attacks, which construct jailbreaking prompts to "fool" LLMs into responding to harmful questions. Early-stage jailbreaking attacks require access to model internals or significant human efforts. More advanced attacks utilize genetic algorithms for automatic an…