CoWindow Attention

Two new attention models slash redundancy to stretch context further

Attention research has a habit of making the same trade-off: sacrifice coverage for speed.

4 min readMachine Learning

Two new papers from a research team, CoWindow Attention and MassAlloc Attention, take aim at a problem many of us have felt but few have named: the sheer redundancy in how attention computes over long contexts. The authors, one of whom shared the work on Reddit, show that by systematically cutting wasted computation, they can speed up the attention operator by up to 7.4x in forward passes and 8.6x in backward passes at 128K tokens, all while maintaining comparable model capabilities at 14B parameters. This is not a claim of universal lossless equivalence to dense attention, they are upfront about that, but it is a concrete, measured step toward making long-context models more practical. For anyone who has watched their training budget balloon as context windows grow, this is the kind of news that deserves a close look.

What makes these approaches worth paying attention to is how they sidestep the complexity trap. CoWindow Attention distributes distant context across KV heads using complementary windows, so each head sees only a sparse slice but the union covers the full causal history, no learned router, no indexer, just a position-defined pattern. MassAlloc Attention keeps full QK scoring but uses the softmax statistics themselves to decide whether the rest of the computation for a tile is worth doing. Both support training forward and backward, and both have been tested at scales from 0.6B to 14B with continued training at 32B. This contrasts with approaches like the one described in [Monodratic: learned product-hash routing for sparse causal attention [R]](/post/monodratic-learned-product-hash-routing-for-sparse-causal-at-cmsge9oa604ajmi9z6jyvij13), which relies on a learned routing mechanism to achieve sparsity. The CoWA and MALA methods are simpler by design, they lean on structure and statistics rather than learned gating, which could make them easier to integrate into existing training pipelines without introducing new failure modes. And in a world where Kimi K3's full weights are here, but they're 'open' with a caveat: What enterprises should know reminds us that openness alone doesn't solve efficiency, these reductions in FLOPs, 28.5% for CoWA and 23.1% for MALA at 14B with 32K context, are the kind of real-world savings that can make or break a deployment decision.

Here is the takeaway worth quoting: **At 14B with 32K context, CoWA reduces total training FLOPs by 28.5% while maintaining comparable capabilities, that is a meaningful efficiency gain, not a theoretical promise.** The authors are honest about the caveats: collective coverage does not mean identical head-wise outputs, and MALA still pays for full QK scoring. They are also inviting feedback on workloads that might stress their assumptions, which is the right posture for research that is still maturing. The open question for practitioners is whether these speedups hold up in end-to-end training runs, not just in isolated attention operators, and whether the tolerance for adaptive post-score allocation remains stable across diverse data distributions. If the team can answer those questions with the same clarity they have brought to the operator benchmarks, they will have given the field something rarer than another architecture, they will have given it a practical lever to pull.

From Machine Learning

I'm one of the authors of two recent papers exploring different sources of redundant computation in attention. I'd like to share the ideas and hear feedback from people working on long-context models and attention kernels.

CoWindow Attention (CoWA) distributes distant context across KV heads using complementary windows, while sharing local and prefix-sink windows. Each head attends sparsely, but the union of their visible positions covers the full causal history. The pattern is position-defined and requires no learned router or indexer.

Read the original at Machine Learning