← Search

Kai Mei

12 accepted papers

2026

R-WoM: Retrieval-augmented World Model For Computer-use Agents

ICLR 2026poster

Large Language Models (LLMs) can serve as world models to enhance agent decision-making in digital environments by simulating future states and predicting action outcomes, potentially eliminating costly trial-and-error exploration. However, this capability is fundamentally limited by LLM’s tendency…

Cited by 0SourceScholar
2025

Agent Security Bench (ASB): Formalizing and Benchmarking Attacks and Defenses in LLM-based Agents

ICLR 2025poster

Although LLM-based agents, powered by Large Language Models (LLMs), can use external tools and memory mechanisms to solve complex real-world tasks, they may also introduce critical security vulnerabilities. However, the existing literature does not comprehensively evaluate attacks and defenses again…

2025

Cerebrum (AIOS SDK): A Platform for Agent Development, Deployment, Distribution, and Discovery

NAACL 2025system demonstrations

Autonomous LLM-based agents have emerged as a powerful paradigm for complex task execution, yet the field lacks standardized tools for development, deployment, and distribution. We present Cerebrum, an open-source platform that addresses this gap through three key components: (1) a comprehensive SDK…

2025

Exploring Concept Depth: How Large Language Models Acquire Knowledge and Concept at Different Layers?

COLING 2025main

Large language models (LLMs) have shown remarkable performances across a wide range of tasks. However, the mechanisms by which these models encode tasks of varying complexities remain poorly understood. In this paper, we explore the hypothesis that LLMs process concepts of varying complexities in di…

2025

From Commands to Prompts: LLM-based Semantic File System for AIOS

ICLR 2025poster

Large language models (LLMs) have demonstrated significant potential in the development of intelligent LLM-based agents. However, when users use these agent applications to perform file operations, their interaction with the file system still remains the traditional paradigm: reliant on manual navig…

2025

Massive Values in Self-Attention Modules are the Key to Contextual Knowledge Understanding

ICML 2025poster

Large language models (LLMs) have achieved remarkable success in contextual knowledge understanding. In this paper, we show for the first time that these concentrated massive values consistently emerge in specific regions of attention queries (Q) and keys (K) while not having such patterns in values…

2025

iAgent: LLM Agent as a Shield between User and Recommender Systems

ACL 2025finding

Traditional recommender systems usually take the user-platform paradigm, where users are directly exposed under the control of the platform’s recommendation algorithms. However, the defect of recommendation algorithms may put users in very vulnerable positions under this paradigm. First, many sophis…

2024

Learning Autonomous Driving Tasks via Human Feedbacks with Large Language Models

EMNLP 2024finding

Traditional autonomous driving systems have mainly focused on making driving decisions without human interaction, overlooking human-like decision-making and human preference required in complex traffic scenarios. To bridge this gap, we introduce a novel framework leveraging Large Language Models (LL…

Cited by 2SourcePDFScholar
2023

NOTABLE: Transferable Backdoor Attacks Against Prompt-based NLP Models

ACL 2023long

Prompt-based learning is vulnerable to backdoor attacks. Existing backdoor attacks against prompt-based models consider injecting backdoors into the entire embedding layers or word embedding vectors. Such attacks can be easily affected by retraining on downstream tasks and with different prompting s…

2022

Rethinking the Reverse-engineering of Trojan Triggers

NeurIPS 2022accept

Deep Neural Networks are vulnerable to Trojan (or backdoor) attacks. Reverse-engineering methods can reconstruct the trigger and thus identify affected models. Existing reverse-engineering methods only consider input space constraints, e.g., trigger size in the input space. Expressly, they assume th…