IsoQuant: Hardware-Aligned SO(4) Isoclinic Rotations for LLM KV Cache Compression
- URL: http://arxiv.org/abs/2603.28430v1
- Date: Mon, 30 Mar 2026 13:37:45 GMT
- Title: IsoQuant: Hardware-Aligned SO(4) Isoclinic Rotations for LLM KV Cache Compression
- Abstract summary: We propose a blockwise rotation framework based on quaternion algebra and the isoclinic decomposition of $SO(4)$.<n>IsoQuant achieves mean kernel-level speedups of $4.5times$--$4.7times$ over RotorQuant with peak speedups above $6times$.
- Score: 0.4496256885343706
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Orthogonal feature decorrelation is effective for low-bit online vector quantization, but dense random orthogonal transforms incur prohibitive $O(d^2)$ storage and compute. RotorQuant reduces this cost with blockwise $3$D Clifford rotors, yet the resulting $3$D partition is poorly aligned with modern hardware and offers limited local mixing. We propose \textbf{IsoQuant}, a blockwise rotation framework based on quaternion algebra and the isoclinic decomposition of $SO(4)$. It represents each $4$D block as a quaternion and applies a closed-form transform $T(v)=q_L v \overline{q_R}$. This yields two main variants: \emph{IsoQuant-Full}, which realizes the full $SO(4)$ rotation, and \emph{IsoQuant-Fast}, which keeps only one isoclinic factor for lower cost; the framework also admits a lightweight $2$D special case. At $d=128$, IsoQuant-Full reduces forward rotation cost from about $2{,}408$ FMAs in RotorQuant to $1{,}024$, while IsoQuant-Fast further reduces it to $512$. Across $18$ fused CUDA settings with $d \in {128,256,512}$, bit widths ${2,3,4}$, and FP16/FP32 execution, IsoQuant achieves mean kernel-level speedups of about $4.5\times$--$4.7\times$ over RotorQuant while maintaining comparable reconstruction MSE, with peak speedups above $6\times$. Current validation is limited to the stage-1 quantize--dequantize path on synthetic normalized vectors; end-to-end KV-cache evaluation remains future work.
Related papers
- Quantum Multi-Armed Bandits and Linear Bandits: Lower Bounds and Algorithms [42.57939149964391]
We study quantum multi-armed bandits (QMAB) and quantum linear bandits (QLB)<n>We prove the first minimax lower bounds of $(Klog(T/K))$ for QMAB and $(dlog(T/d))$ for finite-action QLB.
arXiv Detail & Related papers (2026-08-14T14:04:21Z) - Quantum Speedups for Stochastic Optimization with Heavy-Tailed Noise [49.730496294398726]
We develop a novel quantum mean estimator for heavy-tailed random variables.<n>We derive stronger lower bounds for tail index>4/3$, showing that a nontrivial dependence on a dimension is unavoidable.<n>For a convex objective function, we propose a quantum projected gradient descent method.
arXiv Detail & Related papers (2026-07-28T09:29:37Z) - KroQuant: Kronecker-Structured Block Transforms for Efficient Post-Training Quantization of Diffusion Transformers [42.11461757540461]
Post-training quantization of diffusion transformers (DiTs) to W4A4 severely degrades output quality.<n>We propose KroQuant, a PTQ method that applies a learned Kronecker-structured invertible transform to each 32-element block of the activation.
arXiv Detail & Related papers (2026-07-23T15:52:20Z) - Asymptotically Optimal Regret for Reinforcement Learning without Horizon Dependence [54.59847568544922]
We study horizon-free regret for finite-horizon time-homogeneous Markov decision processes with $A$ states, $A$ actions, horizon $H$, and per-trajectory total reward bounded by $1$.<n>We prove a regret upper bound [tilde O(sqrtSAK+S8A3)] with failure probability $$, where $K$ is the number of episodes and $tilde O(sqrtSAK+S3K)$ hides $mathsfpoly
arXiv Detail & Related papers (2026-07-22T07:42:19Z) - Hurwitz Quaternion Multiplicative Quantization for KV Cache Compression [19.924689669010117]
HQMQ treats each 4-element chunk of K or V as a quaternion and quantizes its unit direction to the emphproduct $q_p cdot q_s$.<n> HQMQ delivers up to $5.05times$ KV compression, shrinking a Llama-3-70B 128k-context cache from 43 GB to 8.5 GB.
arXiv Detail & Related papers (2026-05-26T20:09:17Z) - When Quantization Is Free: An int4 KV Cache That Outruns fp16 on Apple Silicon [0.0]
KV-cache quantization is framed as a quality--latency trade-off.<n>We show it is emphinverted on Apple Silicon's unified memory.
arXiv Detail & Related papers (2026-05-07T05:44:39Z) - Block-Wise Differentiable Sinkhorn Attention: Tail-Refinement Gradients with a Gap-Aware Dustbin Bridge [0.0]
We study long-context balanced optimal transport attention on TPU hardware through a stopped-base, fixed-depth tail-refinement surrogate.<n>We provide a local surrogate-bias bound, an a posteriori bias certificate, and a projective contraction certificate for strictly positive active blocks.<n>On synthetic masked problems, the optimized matches exact autodiff of the same centered surrogate to within $10-5$--$10-10$.
arXiv Detail & Related papers (2026-04-28T19:01:11Z) - A four-player potential game for barren-plateau-aware quantum ansatz design [0.0]
We cast the design of parameterized quantum circuits as a four-player potential game whose state is a circuit directed acyclic graph (DAG)<n>Per-player restricted action sets factorize the move space into append, remove, retype, and rewire operations; a block-coordinate $varepsilon-Nash residual $_textNash$ certifies that no single player can improve unilaterally.
arXiv Detail & Related papers (2026-04-23T07:58:46Z) - Toward Magnetic-Field-Free Quantum Computing and Quantum Reservoir Computing in Engineered Organic Materials: A Unified Framework from the 3-Layer Quantum Brain Hypothesis [0.0]
We extend the spin-vortex-induced loop-current (SVILC) qubit to engineered organic materials operated without any applied magnetic field.<n>Four paths are proposed: (P1) a flavin--nitroxide radical-pair reservoir, (P2) a perchlorotriphenylmethyl (PTM) radical array in a covalent organic framework, and (P4) a Su-Schrieffer--Heeger soliton on trans-polyacetylene.
arXiv Detail & Related papers (2026-04-22T09:12:12Z) - Universal Quantum Suppression in Frustrated Ising Magnets across the Quasi-1D to 2D Crossover via Quantum Annealing [0.0]
competing ferromagnetic and antiferromagnetic couplings generate a sign problem provably intractable for quantum Monte Carlo at any system size.<n>We measure quantum-driven transitions at $g_cmathrmQPUin0.286,,0.210,,0.156,,0.093$ for $in1.0,,0.7,,0.5,,0.3$.
arXiv Detail & Related papers (2026-03-25T13:50:52Z) - Optimal Scalar Quantization for Matrix Multiplication: Closed-Form Density and Phase Transition [50.36362492608702]
We study entrywise scalar quantization of two matrices prior to multiplication.<n>We obtain a closed-form optimal point density [ star(u) propto exp!left(-fracu26right)bigl( (1-2)+2u22bigr), qquad u=fracx_X,, and prove a correlation-driven phase transition.
arXiv Detail & Related papers (2026-03-20T01:53:44Z) - ATLAS: Efficient Atom Rearrangement for Defect-Free Neutral-Atom Quantum Arrays Under Transport Loss [46.043413607980845]
Neutral-atom quantum computers encode qubits in individually trapped atoms arranged in optical lattices.<n>Algorithm converts a randomly loaded $W times W$ lattice into a defect-free $L times L$ subarray.<n>Algorithm achieves sublinear move scaling and linear growth of required initial size with target dimension.
arXiv Detail & Related papers (2025-11-20T12:32:35Z) - SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models [61.474101404805545]
Diffusion models can generate high-quality images, but as they scale, rising memory demands and higher latency pose deployment challenges.<n>We propose SVDQuant, a new 4-bit quantization paradigm to overcome this limitation.<n>We reduce the memory usage for the 12B FLUX.1 models by 3.5$times$, achieving 3.0$times$ speedup over the 4-bit weight-only quantization (W4A16) baseline.
arXiv Detail & Related papers (2024-11-07T18:59:58Z) - FlatQuant: Flatness Matters for LLM Quantization [58.28221892035609]
We propose FlatQuant, a new post-training quantization approach that enhances the flatness of weights and activations.<n>Our approach identifies optimal affine transformations for each linear layer, calibrated in hours via a lightweight objective.<n>It achieves less than 1% accuracy drop for W4A4 quantization on the LLaMA-3-70B model, surpassing SpinQuant by 7.5%.
arXiv Detail & Related papers (2024-10-12T08:10:28Z) - Measuring quantum relative entropy with finite-size effect [53.64687146666141]
We study the estimation of relative entropy $D(rho|sigma)$ when $sigma$ is known.<n>Our estimator attains the Cram'er-Rao type bound when the dimension $d$ is fixed.
arXiv Detail & Related papers (2024-06-25T06:07:20Z) - Globally optimal interferometry with lossy twin Fock probes [0.0]
We show that a method of moments readout of two quadratic spin observables $J_z2$ and $J_+2+J_-2$ is globally optimal for Dicke state probes.
In the lossy setting, we derive the time-inhomogeneous Markov process describing the effect of particle loss on twin Fock states.
arXiv Detail & Related papers (2023-08-10T22:56:12Z) - Cost Function Dependent Barren Plateaus in Shallow Parametrized Quantum
Circuits [0.755972004983746]
Variational quantum algorithms (VQAs) optimize the parameters $vectheta$ of a parametrized quantum circuit.
We prove two results, assuming $V(vectheta)$ is an alternating layered ansatz composed of blocks forming local 2-designs.
We illustrate these ideas with large-scale simulations, up to 100 qubits, of a quantum autoencoder implementation.
arXiv Detail & Related papers (2020-01-02T18:18:25Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.