RepetitionCurse: Measuring and Understanding Router Imbalance in Mixture-of-Experts LLMs under DoS Stress
This paper introduces RepetitionCurse, a low-cost black-box attack that exploits the lack of load balancing constraints in Mixture-of-Experts LLMs by using simple repetitive token patterns to force severe routing concentration, thereby creating computational bottlenecks and significantly degrading inference latency and service availability.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
Imagine a massive, high-tech library where thousands of books (the "experts") are stored on different shelves (the "GPUs"). When a librarian (the "router") gets a request, they don't just grab one book; they pick the top few best books to answer the question. To make things fast, the library splits the work: some shelves handle the first few books, others handle the next batch, and they all work at the same time. This is how modern AI models called Mixture-of-Experts (MoE) work. They are designed to be super efficient by spreading the work out evenly.
However, a new paper called "RepetitionCurse" reveals a sneaky flaw in how this library operates.
The Problem: The "Copy-Paste" Trick
The researchers discovered that if you ask the librarian a question using a very strange, repetitive pattern (like typing "AI AI AI AI..." over and over), the librarian gets confused.
Normally, the librarian looks at the meaning of your words to decide which books to pick. But when you use a repetitive pattern, the librarian stops looking for meaning and starts acting like a broken record. Instead of spreading the work out across all the shelves, the librarian suddenly decides that only one specific shelf has the right books for this repetitive pattern.
The Result: A Traffic Jam
Here is what happens next:
- The Bottleneck: All the work gets dumped onto that single shelf. The librarians on that shelf are running around frantically, trying to find all the books.
- The Idle Workers: Meanwhile, the librarians on all the other shelves are standing around doing nothing, waiting for the busy shelf to finish.
- The Delay: Because the whole system has to wait for that one slow shelf to catch up before it can move on to the next step, the entire library grinds to a halt.
In the world of AI, this delay is called Time-to-First-Token (TTFT). It's the time it takes for the AI to say its first word. Under normal conditions, this is fast. Under this "RepetitionCurse" attack, the AI might take 2 to 3 times longer to start speaking.
Why This Matters
The paper calls this a Denial-of-Service (DoS) attack. It's like someone walking into a fast-food restaurant and ordering 1,000 identical, complicated burgers at once. The kitchen gets clogged trying to make those specific burgers, and suddenly, the person who just wanted a simple coffee has to wait an hour.
- The Attack is Simple: You don't need to be a hacker genius. You just need to type repetitive words. The paper calls this a "black-box" attack, meaning you don't need to know how the AI's brain works inside; you just need to know that repeating words breaks the balance.
- It Works Everywhere: The researchers tested this on 139 different AI models (including popular ones like Mixtral and Qwen). They found that almost all of them are vulnerable.
- The More You Scale, The Worse It Gets: Ironically, the more powerful the AI system is (using more computers to work in parallel), the easier it is to break. The researchers found that using more computers to split the work actually makes the traffic jam worse when this attack is used.
The Bottom Line
The paper concludes that while splitting the work among many computers makes AI faster, it also creates a weak spot. If someone uses a simple, repetitive trick, they can force the system to ignore its own efficiency rules, causing it to slow down dramatically and potentially break its promises to users (like "we will reply within 2 seconds").
The authors suggest that until AI companies figure out a better way to balance the work automatically, they might need to be careful about how many computers they use at once, or they risk their systems being held hostage by a simple "AI AI AI" loop.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.