tahamajs/RR / Docs /visualizer /components /a3_compressor.json
tahamajs's picture
download
raw
2.79 kB
{
"id": "a3_compressor",
"name": "A3 Compressor: Activation-Aware Factorized SVD Compression with Fisher/Acti",
"file": "Codes/framework/compression_engine/a3_compressor.py",
"type": "Compression Engine",
"category": "Low-Rank SVD & Factorization",
"paradigm": "svd",
"icon": "\u26a1",
"description": "A3 Compressor: Activation-Aware Factorized SVD Compression with Fisher/Activation Energy Weighting and LayerNorm Recalibration. Registered in COMPRESSOR_REGISTRY as 'a3' and 'a3_compressor'. Mathematical Formulation: 1. Online Streaming Activation Covariance & Empirical Fisher Diagonal Proxy (Activation Energy): $$C_X = \\frac{1}{N} \\sum_{t=1}^N x_t x_t^\\top, \\quad F = \\mathbb{E}_{x \\sim \\mathc",
"classes": [
"A3FactorizedLinear",
"A3Compressor"
],
"methods_count": 18,
"math_latex": "\\text{FLOPs}_{\\text{dense}} = 2 \\cdot m \\cdot n \\quad \\longrightarrow \\quad \\text{FLOPs}_{\\text{factorized}} = 2 \\cdot r \\cdot (m + n)",
"components": [
{
"id": "v_factor",
"name": "Factor V (Low-Rank Subspace Projection)",
"role": "Projection to Rank r",
"shape_transform": "[r \u00d7 d_in]",
"color": "#3acbb8",
"math_latex": "z_t = x_t V_r^\\top, \\quad V_r \\in \\mathbb{R}^{r \\times d_{in}}"
},
{
"id": "u_factor",
"name": "Factor U (Full-Rank Expansion Projection)",
"role": "Expansion to Output Space",
"shape_transform": "[d_out \u00d7 r]",
"color": "#2dd4bf",
"math_latex": "y_t = z_t (\\Sigma_r U_r^\\top) + b, \\quad U_r \\in \\mathbb{R}^{d_{out} \\times r}"
},
{
"id": "adapter",
"name": "Residual LoRA / Norm Calibration",
"role": "Fidelity Compensation",
"shape_transform": "[d_out \u00d7 d_in] Residual",
"color": "#5eead4",
"math_latex": "\\Delta y_t = x_t (B A)^\\top \\cdot \\frac{\\alpha}{r}, \\quad A \\in \\mathbb{R}^{r \\times d_{in}}, \\; B \\in \\mathbb{R}^{d_{out} \\times r}"
}
],
"model_projections": {
"distilbert": {
"hidden_size": 768,
"intermediate_size": 3072,
"baseline_layer_params": 6488064,
"compressed_layer_params": 786432,
"savings_ratio": 8.25
},
"roberta": {
"hidden_size": 768,
"intermediate_size": 3072,
"baseline_layer_params": 6488064,
"compressed_layer_params": 786432,
"savings_ratio": 8.25
},
"gpt2": {
"hidden_size": 768,
"intermediate_size": 3072,
"baseline_layer_params": 6488064,
"compressed_layer_params": 786432,
"savings_ratio": 8.25
},
"qwen": {
"hidden_size": 896,
"intermediate_size": 4864,
"baseline_layer_params": 11124736,
"compressed_layer_params": 1081344,
"savings_ratio": 10.29
}
}
}

Xet Storage Details

Size:
2.79 kB
·
Xet hash:
74d5a5b5954b15e57d34023344e08538e2a7454c43c6df40e59c07814e8045b1

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.