When you ask a large language model a question, the system splits work into two stages: understanding and answering. The first stage creates a huge key-value (KV) cache that must be sent to the second stage before it starts. These transfers are extremely large, taking hundreds of milliseconds and dominating network links. The network's routing system (ECMP) can accidentally send two of these huge transfers through the same cable, causing delays.
ARK coordinates the senders so each chooses a different network path for its transfer. It adjusts the source port number, which changes the path selected by ECMP. This avoids collisions without new hardware.
In simulations, ARK reduced the average transfer time by up to 27.3% and the time for the slowest 5% by 34.0%. This gets the first token to users faster, improving the experience under load.
