Can you make a 3:1 SWA version or linear attention version?

#6
by Gavin-chen - opened

This model uses 42 KiB of KV cache per token. While this is smaller than the 27B model (64 KiB), it's larger than the 9B model (32 KiB), and this model is only 2B parameters. Can your team make it's KV cache even smaller?

Also, if your team is using SWA, I know of a relevant paper; please see: https://arxiv.org/abs/2506.15545

Thanks for the careful read — you're right that the per-token KV cache is large relative to the parameter count, and comparing against the 9B is a fair point.

To be clear, this isn't an area we've left unaddressed: MiniCPM-SALA already implements exactly this direction — a hybrid architecture combining 25% sparse attention (InfLLM-v2) with 75% linear attention (Lightning Attention), which brings substantially lower KV-cache overhead than a dense baseline. We simply haven't applied that architecture to this particular model; MiniCPM5-2B is dense full attention, so there's no SWA or linear-attention variant of this checkpoint.

Can we see any other model that use SALA's architecture in future?

Sign up or log in to comment