new

Get trending papers in your email inbox!

Subscribe

Daily Papers

byAK and the research community

Oct 29

OCTOPUS: Optimized KV Cache for Transformers via Octahedral Parametrization Under optimal Squared error quantization

The key-value (KV) cache dominates memory bandwidth and footprint in long-context autoregressive inference. Recent rotation-preconditioned codecs (TurboQuant, PolarQuant) show that a structured random rotation followed by a per-coordinate scalar quantizer matched to an analytically tractable marginal is a near-optimal recipe for KV compression. OCTOPUS advances this paradigm through joint quantization of rotated coordinate triplets. Each triplet's direction is mapped to a square via an octahedral parameterization, and the two resulting coordinates and the triplet norm are Lloyd-Max quantized against implementation-matched marginals. Optimizing the per-triplet squared error gives a strictly non-uniform bit allocation depending only on the total dimensionality of the keys. We find the finite-dimensional quality optimum with sweeps to be constant on every real decoder we test. The codec is data-oblivious, online, and deterministic given a seed. Across text, video, and audio, OCTOPUS matches or beats every prior rotation codec at every reported bit width and metric, with a lead that grows as bits drop for extreme compression. Furthermore, a fused Triton implementation reconstructs keys on the fly without materializing the uncompressed key, so the codec adds no decode-time bandwidth or latency over the existing dequantization. Project Page: https://octopus-quant.github.io/

stabilityai Stability AI
·
May 19 1

CIS(2): a state-specific, size-intensive perturbative correction to configuration interaction singles

We present a state-specific second-order perturbation theory based on a configuration interaction singles (CIS) reference and its generalized Fock operator. Starting from a fully internally contracted construction, we retain the complete doubles space and internally contracted triples in the first-order wave function. We consider two partitions, Canonical and Block-diagonal, with reference excitation energies given by the Fock excitation gap ω^{(0)} and the CIS excitation energy ω_{rm CIS}, respectively. Our analysis establishes that strict size-intensivity is preserved when the contracted-triples denominators use bare Fock gaps and the occupied--virtual Fock block is projected to remove residual spectator coupling. Based on this observation, combining Canonical triples with Block-diagonal doubles yields a size-intensive Hybrid partition. These properties are verified for water with non-interacting helium atoms. Our formulation enables O(o^3v^2) cost per matrix--vector product in a semicanonical basis. On the QUEST\#1 benchmark, Canonical and Block-diagonal show opposite systematic biases, whereas Hybrid gives the best overall accuracy among the three partitions. Its mean absolute errors for singlets and triplets are 0.24 and 0.15~eV, compared with 0.28 and 0.22~eV for CIS(D). The improvement is largest for Rydberg excitations, although CIS(D) remains more accurate for valence singlets.

  • 1 authors
·
Oct 1

Three-Phase Transformer

We present Three-Phase Transformer (3PT), a residual-stream structural prior for decoder-only Transformers on a standard SwiGLU + RMSNorm + RoPE + GQA backbone. The hidden vector is partitioned into N equally-sized cyclic channels, each maintained by phase-respecting ops: a per-channel RMSNorm, a 2D Givens rotation between attention and FFN that rotates each channel by theta + i*(2*pi/N), and a head-count constraint aligning GQA heads with the partition. The architecture is a self-stabilizing equilibrium between scrambling and re-imposition, not a bolted-on module. The partition carves out a one-dimensional DC subspace orthogonal to the channels, into which we inject a fixed Gabriel's horn profile r(p) = 1/(p+1) as an absolute-position side-channel composing orthogonally with RoPE's relative-position rotation. The canonical N=3 borrows its metaphor from balanced three-phase AC, where three sinusoids 120 degrees apart sum to zero with no anti-correlated pair. At 123M parameters on WikiText-103, 3PT achieves -7.20% perplexity (-2.62% bits-per-byte) over a matched RoPE-Only baseline at +1,536 parameters (0.00124% of total), with 1.93x step-count convergence speedup (1.64x wall-clock). N behaves as a parameter-sharing knob rather than a unique optimum: at 5.5M an N-sweep over {1,2,3,4,6,8,12} is near-monotone with N=1 winning; at 123M a three-seed sweep finds N=3 and N=1 statistically indistinguishable. The load-bearing mechanism is the channel-partitioned residual stream, per-block rotation, per-phase normalization, and horn DC injection. We characterize (a) self-stabilization of the geometry without explicit enforcement, a novel instance of the conservation-law framework for neural networks; (b) a U-shaped depth profile of rotation-angle drift at 12 layers; (c) orthogonal composition with RoPE, attention, and FFN.

BrainsBuild BrainsBuild
·
Apr 14 5