加载中…
加载中…
This repository provides a physically compressed and LoRA-healed checkpoint of google-bert/bert-base-uncased fine-tuned on GLUE SST-2, compressed using the rotational_slicegpt_int4 (RotationalSliceGPTINT4Compressor) engine from the NeuralOps Library at Pareto-selected rank k=256 (4.46x compression ratio, -77.6% parameter reduction).
RotationalSliceGPTINT4Compressor: Rotational-SliceGPT-INT4 hybrid compression engine.
| Property | Specification |
|---|---|
Registry Key (COMPRESSOR_REGISTRY) |
rotational_slicegpt_int4 |
| NeuralOps Class & Family | RotationalSliceGPTINT4Compressor (quantization) |
| Source Implementation | neuralops/compression/quantization/rotational_slicegpt_int4.py |
| Base Architecture | google-bert/bert-base-uncased (109,483,778 parameters, 12 layers, $D=768$) |
Showcase Rank (k) |
256 (LoRA Recovery Applied: True) |
| Compressed Parameter Count | 24,549,122 (-77.58% reduction, 4.460x physical compression ratio) |
| Teacher Baseline SST-2 Acc | 91.63% |
| Showcase Final SST-2 Acc | 56.08% (Zero-Shot: 50.92%, Parity Delta: -35.55%) |
k=256, 128, 64, 32)Evaluated physically on NVIDIA GeForce RTX 3090 (24GB) on the GLUE SST-2 validation split (872 samples). Whenever zero-shot compression drops accuracy by > 2.0% alongside > 1.10x reduction, NeuralOps applies targeted r=8 LoRA healing (50 steps, lr=2e-4, 45s budget):
Rank (k) |
Physical Ratio | Compressed Params (Reduction) | Zero-Shot Acc | Healed Acc | Parity vs Teacher (91.63%) |
|---|---|---|---|---|---|
256 ⭐ |
4.46x |
24,549,122 (-77.6%) |
50.92% |
56.08% (LoRA) |
-35.55% |
128 |
4.46x |
24,549,122 (-77.6%) |
49.08% |
50.92% (LoRA) |
-40.71% |
64 |
4.46x |
24,549,122 (-77.6%) |
49.08% |
50.92% (LoRA) |
-40.71% |
32 |
4.46x |
24,549,122 (-77.6%) |
49.08% |
50.92% (LoRA) |
-40.71% |
(⭐ indicates the serialized checkpoint packaged in this Hugging Face repository.)
pytorch_model.bin: Full serialized compressed BERT state dictionary (with low-rank factors / quantization buffers / LoRA healing adapters baked in).config.json: Model configuration tagged with compression_method: "rotational_slicegpt_int4" and compression_rank: 256.compression_summary.json: Complete machine-readable telemetry across all evaluated ranks [256, 128, 64, 32].bert-base-uncased vocabulary (vocab.txt, tokenizer.json, tokenizer_config.json).import torch
from huggingface_hub import hf_hub_download
from transformers import AutoTokenizer, BertForSequenceClassification
from neuralops.core.compression_pipeline import compress_model
repo_id = "tahamajs/bert-comp-rotational-slicegpt-int4-sst2"
tokenizer = AutoTokenizer.from_pretrained(repo_id)
# Reconstruct compressed NeuralOps topology and load serialized compressed + healed weights
model = BertForSequenceClassification.from_pretrained("google-bert/bert-base-uncased", num_labels=2)
model = compress_model(model, {"compressors": [{"method": "rotational_slicegpt_int4", "rank": 256}]})
ckpt_path = hf_hub_download(repo_id=repo_id, filename="pytorch_model.bin")
state_dict = torch.load(ckpt_path, map_location="cpu")
model.load_state_dict(state_dict, strict=False)
model.eval()
inputs = tokenizer("NeuralOps achieves strong compression without sacrificing downstream accuracy.", return_tensors="pt")
with torch.no_grad():
logits = model(**inputs).logits
pred = logits.argmax(dim=-1).item()
print("Predicted SST-2 Sentiment (0=Negative, 1=Positive):", pred)
@software{neuralops2026compressorzoo,
title = {NeuralOps: Multi-Rank Compressed BERT-SST2 Zoo with Pareto LoRA Recovery},
author = {Majlesi, Taha},
year = {2026},
url = {https://github.com/tahamajs/NeuralOps}
}