Skip to main content

Fingerprint any transformer's compression potential in 30 minutes.

Project description

fraQtl

fraQtl Diagnostic

Fingerprint any transformer's compression potential — fast.

For public installs, use fraqtl-diagnostic >= 0.2.0 for the inference-readiness scanner.

Measures per-layer:

  • γ (stretched-exponential decay shape of the Hessian spectrum)
  • knee (spectrum cutoff index)
  • k95 (directions needed for 95% of eigenvalue energy)
  • depth-law (how decay shape evolves across layers)
  • compression potential + suggested bit budgets (Shannon-based)

Works on any HuggingFace-compatible transformer. ~3 min for a 0.5B model on A100, ~5 min for 1B, ~10 min for 7B.


Install

pip install fraqtl-diagnostic           # >=0.2.0 on PyPI
pip install -e /path/to/diagnostic-public  # editable install from source

Use

fraqtl analyze meta-llama/Llama-3.2-1B-Instruct

Inference readiness scan

diagnose-inference checks whether a model's config appears ready for a target serving context. It uses HuggingFace config.json plus textbook memory math: no GPU, no model load, no compression run.

fraqtl diagnose-inference Qwen/Qwen2.5-7B-Instruct --context 65536

Example summary:

model:      Qwen/Qwen2.5-7B-Instruct
arch:       qwen2 (GQA, layers=28, kv_heads=4)
context:    32768 native / 65536 requested
KV memory:  3.758 GB @ 64K (fp16)
flags:      CONTEXT_EXCEEDS_NATIVE ROPE_SCALING_REQUIRED YARN_REQUIRED YARN_MISSING

The command writes a *_diagnose-inference_*.json machine-readable readiness receipt to disk, and prints a short colored summary to stdout.

from fraqtl_diagnostic import analyze

report = analyze("meta-llama/Llama-3.2-1B-Instruct")
print(report.summary())
report.to_html("llama-1b_fingerprint.html")
report.to_png("llama-1b_fingerprint.png")

Try it on your GPU in one command (Modal)

If you don't want to fight Python-env dependencies locally, the fastest way to try the tool on a real model is via Modal (free tier gives you an A100):

# one-time: `pip install modal && modal setup`
# assumes a Modal secret named `huggingface` with an HF token

cd diagnostic-public/
modal run tests/modal_try.py --model-id Qwen/Qwen2.5-0.5B
modal run tests/modal_try.py --model-id TinyLlama/TinyLlama-1.1B-Chat-v1.0
modal run tests/modal_try.py --model-id mistralai/Mistral-7B-v0.1 --n-seqs 32 --seq-len 512

# pull the report back:
modal volume get fraqtl-hf-cache fraqtl-results/diagnostic-smoke ./reports/

What you get

Three outputs, same data, different framings:

  • *.json — machine-readable per-layer fingerprint (feed into other tools)
  • *.html — human-readable report with tables + embedded figure
  • *.png — 4-panel figure: spectrum overlay, γ depth-law, k95/layer, summary

How to read the output

γ (stretched-exponential shape parameter)

The Hessian input-covariance spectrum λ_i of each linear projection is fit against λ_i ≈ exp(−β · i^γ + c). γ is the shape of the decay:

γ range interpretation
γ ≈ 0.3 Stretched: fast head decay, long tail → compressible
γ ≈ 0.5 Typical for attention o_proj on Llama/Qwen/Mistral
γ ≈ 0.8 Typical for MLP down_proj on Llama/Qwen/Mistral
γ ≈ 1.0 Pure exponential decay — harder to compress aggressively
γ > 1.0 Super-exponential (flat head, sharp crash) — limited

Lower γ = more compression headroom.

k95 / dim

Fraction of eigendirections needed to capture 95% of eigenvalue energy. A value of 0.1 means "95% of the Hessian mass lives in the top 10% of directions" — prime territory for rank-preserving compression. Values typical on production transformers:

k95/dim range implication
< 10% very compressible, low-rank friendly
10–30% common; most dense transformers fall here
30–50% harder to compress without structured loss
> 50% spectrum is near-uniform, limited headroom

Depth-law

Linear fit of γ across layer depth. A negative slope is the common case (shallow layers exponential, deep layers more stretched). The magnitude of the slope × R² tells you whether the shape is a stable architecture property or noisy per-layer.

Suggested bit budget

Shannon-derived bits-per-weight that the information-theoretic ceiling can tolerate at three conservatism tiers. This is a ceiling, not a prediction. Real PPL loss from compression depends on the implementation. The diagnostic tells you how much room the math leaves; the actual compression run tells you how close to the ceiling you got.


Status

v0.2 (current): diagnostic metrics + suggested bit budgets, plus diagnose-inference for config-level serving-readiness checks.

v1.0 (coming with Paper 3, ~4 weeks): adds Shannon-efficiency grading — "your model is at X% of the theoretical ceiling vs competitors at Y%."

Same pip install fraqtl-diagnostic — grading is a layer on top of the existing diagnostic, not a separate tool.


How it works (one-paragraph summary)

For each target projection W : ℝ^d_in → ℝ^d_out, we capture the input covariance H = E[x^T x] on wikitext-2 calibration, then eigendecompose it. The spectrum λ_i encodes how much of the layer's Jacobian mass lives along each eigendirection of the input distribution. Tight universal shape (fixed γ across layers) implies compressible redundancy; a fat-tailed spectrum (high k95/dim) implies less. Shannon rate-distortion gives the information-theoretic floor D*(R) = geomean(λ) · 2^(−2R) at any bit budget R, which the diagnostic reports.

Full derivation + universality data across 8 architectures is in the forthcoming Paper 3.


Want to actually compress your model?

The diagnostic tells you the ceiling. The compression engine is the closed part of the product:

fraqtl.ai/compress


License

Apache 2.0.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

fraqtl_diagnostic-0.2.3.tar.gz (45.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

fraqtl_diagnostic-0.2.3-py3-none-any.whl (43.0 kB view details)

Uploaded Python 3

File details

Details for the file fraqtl_diagnostic-0.2.3.tar.gz.

File metadata

  • Download URL: fraqtl_diagnostic-0.2.3.tar.gz
  • Upload date:
  • Size: 45.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.13

File hashes

Hashes for fraqtl_diagnostic-0.2.3.tar.gz
Algorithm Hash digest
SHA256 2ca964af323bada05d54905d29dc14d0ce729c2ca5a407fc9560641ba7eaa781
MD5 5275548c88a3a4d9609931d4a6a163c8
BLAKE2b-256 cedab4c05b830e1f4396f5cb81d9495ce0f0ad6d294395d596c0a5fff5ea1c9e

See more details on using hashes here.

File details

Details for the file fraqtl_diagnostic-0.2.3-py3-none-any.whl.

File metadata

File hashes

Hashes for fraqtl_diagnostic-0.2.3-py3-none-any.whl
Algorithm Hash digest
SHA256 f57729715c33b507e484e53b295561e867e33f99e92a41069c32697c17d4ed74
MD5 549b32de4991447213c6e306d24c3ea7
BLAKE2b-256 1bc398c4c33d074446d41bccd02257d4704d9814b866b14f823193e7a2de7da7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page