CHASE: Channel-Aligned Structure Exploitation for Geometry-Aware Model Engineering
cs.LG, cs.CL
Submitted: 2026-10-07
Updated: 2026-10-07
Comments: 26 pages, 9 tables
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: Geometric and Spectral Alignment (GSA) characterizes trained networks through spectral concentration, physical-channel alignment, support structure, and changes in singular bases.
Abstract
Geometric and Spectral Alignment (GSA) characterizes trained networks through spectral concentration, physical-channel alignment, support structure, and changes in singular bases. In this paper, we propose CHASE (Channel-Aligned Structure Exploitation) to use these structures in practical model design. CHASE covers six applications across model modification, reconfiguration, and compression. CORA, COEC, and CORAM apply GSA to parameter-efficient finetuning, structured-pruning compensation, and model merging. We further develop three new methods. CAGA uses GSA to identify multi-head attention heads that can share a KV representation and constructs the shared key and value heads through geometric alignment and low-rank subspace extraction. SAKV uses GSA to determine which adjacent layers can share a low-rank KV-cache representation and the retained rank for each layer group. CAPS uses GSA spectral structure to group output neurons and selects retained input channels separately for each group. Results from CORA, COEC, and CORAM establish the effectiveness of GSA for adaptation, pruning compensation, and model merging. Experiments on CAGA show that geometric shared-head construction substantially improves MHA-to-GQA conversion, and SAKV and CAPS improve over representative baselines for KV-cache compression and structured pruning. These results show that the structures identified by GSA can be used directly to design methods for a range of model operations.
Sources
- Optimised Grouped-Query Attention Mechanism for Transformers
- The Llama 3 Herd of Models
- RCPU: Rotation-Constrained Error Compensation for Structured Pruning of Large Language Models
- RULER: What's the Real Context Size of Your Long-Context Language Models?
- Geometric and Spectral Alignment for Deep Neural Network I
- Geometric and Spectral Alignment for Deep Neural Network II
- ShortGPT: Layers in Large Language Models are More Redundant Than You Expect
- Pointer Sentinel Mixture Models
- Mistral 7B
- Fast Transformer Decoding: One Write-Head is All You Need
- Qwen2.5 Technical Report
- Llama 2: Open Foundation and Fine-Tuned Chat Models
- CORA: Per-Slice Coherent Orthogonal Rotation for SVD-based Low-Rank Adaptation
- Qwen2.5-1M Technical Report
- Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunch
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks