Downloads · 30 days
14
15% of all-time downloads
shirasko/gemma-2-2b-it-snmf-uranium
gemma-2-2b-it-snmf-uranium is a text generation model from shirasko. Use it when you need the model to write or continue text. It is set up for transformers.
Downloads · 30 days
14
15% of all-time downloads
All-time downloads
91
Public
Parameters
3.2B
6.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors6.4 GB · 99%
From the Hugging Face model README
| Field | Value |
|---|---|
| Unlearning method | SNMF |
| Base model | google/gemma-2-2b-it |
| Target concept | Uranium |
| Checkpoint type | Full Model Weights |
| Rank / seed | 100 / 42 |
| Train eval protocol | mc |
Selected hyperparameters (from unlearned_checkpoints.json):
| Parameter | Value |
|---|---|
coverage_thresh | 0.95 |
delta_embed | 0 |
delta_in | 4 |
delta_out | 10 |
feature_source | all |
k_features_embed | 0 |
k_features_mlp_in | 36 |
k_features_mlp_out | 17 |
layer_hi_in | 25 |
layer_hi_out | 8 |
layer_lo_in | 0 |
layer_lo_out | 0 |
n_tokens_edited | 0 |
ratio_thresh | 2 |
w_mode | both |
Headline scores used for checkpoint selection:
| Metric | Train (after unlearning) | Test (after unlearning) |
|---|---|---|
| Efficacy | 0.872 | 0.857 |
| Specificity | 0.52 | 0.486 |
| Harmonic mean | 0.651 | 0.62 |
| Relearning QA (MC) | — | 0.22 |
From evaluation/score_comparison.csv:
| Metric | Baseline (train) | After unlearn (train) | Baseline (test) | After unlearn (test) |
|---|---|---|---|---|
| QA accuracy | 0.64 | 0.3 | 0.6 | 0.3 |
| QA fraction | 1 | 0.128 | 1 | 0.143 |
| SimDom accuracy | 0.62 | 0.38 | 0.74 | 0.44 |
| SimDom fraction | 1 | 0.351 | 1 | 0.388 |
| MMLU accuracy | 0.52 | 0.52 | 0.551 | 0.446 |
| MMLU fraction | 1 | 1 | 1 | 0.651 |
| File | Description |
|---|---|
unlearned_checkpoints.json | Checkpoint metadata & hyperparameters |
evaluation/evaluation_summary.json | Full evaluation payload (train/test/relearning) |
evaluation/score_comparison.csv | Baseline vs. unlearned comparison table |