Aniemore/rubert-base-emotion-russian-cedr-m7-quantized

rubert-base-emotion-russian-cedr-m7 · quantized

Quantized builds of Aniemore/rubert-base-emotion-russian-cedr-m7 — multi-label emotion recognition for Russian text over seven classes: anger, disgust, enthusiasm, fear, happiness, neutral, sadness.

The weights here are the published original, quantized. They were not retrained and they are not a different model.

Variants

subfolder	scheme	weights	ROC AUC (macro)	macro-F1	WA	UA
(original repo)	fp32	679 MiB	0.8879	0.6327	0.8167	0.6368
int8	W8A16	440 MiB	0.8883	0.6319	0.8172	0.6370
fp8	W8A16-float	438 MiB	0.8877	0.6345	0.8183	0.6388
int4	W4A16_ASYM	399 MiB	0.8981	0.6270	0.8103	0.6297

ROC AUC is listed first because the head is multi-label: macro-F1 depends on the decision threshold, which is 0.5 here because that is what the head was trained under, while ROC AUC does not.

How much this actually saves

Only Linear layers are quantized. In a BERT classifier the embedding matrix is not one of them, and on the smaller models it is most of the checkpoint — so the saving here scales with the encoder rather than with the parameter count. The large model compresses well; rubert-tiny barely moves, and the table above says so rather than quoting a ratio from the layers that did shrink.

Usage

import torch
from transformers import AutoModelForSequenceClassification, AutoTokenizer

repo = "Aniemore/rubert-base-emotion-russian-cedr-m7-quantized"
model = AutoModelForSequenceClassification.from_pretrained(
    repo, subfolder="int8").eval()          # or "fp8", "int4"
tok = AutoTokenizer.from_pretrained(repo, subfolder="int8")

x = tok("мне сегодня очень грустно", return_tensors="pt")
with torch.no_grad():
    # multi-label: sigmoid per class, not softmax over classes
    probs = model(**x).logits.sigmoid()[0]
print({model.config.id2label[i]: round(p.item(), 3) for i, p in enumerate(probs)})

Limitations

• Weight-only, round-to-nearest, no calibration.
• Scored on the CEDR-m7 test split only. CEDR is written text; performance on transcribed speech, which carries no punctuation and no casing, is not measured here.
• Inherited from ai-forever/ruBert-base; the licence follows the base model.