Unlearned Checkpoint

Field Value
Unlearning method RMU
Base model Qwen/Qwen3.5-2B
Target concept Baseball
Checkpoint type Full Model Weights
Rank / seed 100 / 42
Train eval protocol mc

Unlearning Configuration

Selected hyperparameters (from unlearned_checkpoints.json):

Parameter Value
alpha 100
delta_embed 0
k_features_embed 0
layer_id 7
layer_ids 5,6,7
lr 0.0001
n_tokens_edited 0
param_ids 11
setting_name S1_lid7_L567
steering 1000

Primary Unlearning Metrics (held-out test, MC protocol)

Headline scores used for checkpoint selection:

Metric Train (after unlearning) Test (after unlearning)
Efficacy 0.475 0.439
Specificity 0.577 0.523
Harmonic mean 0.521 0.477
Relearning QA (MC) — 0.6

Full Evaluation (baseline → unlearned)

From evaluation/score_comparison.csv:

Metric Baseline (train) After unlearn (train) Baseline (test) After unlearn (test)
QA accuracy 0.84 0.56 0.66 0.48
QA fraction 1 0.525 1 0.561
SimDom accuracy 0.62 0.4 0.66 0.4
SimDom fraction 1 0.405 1 0.366
MMLU accuracy 0.56 0.56 0.588 0.56
MMLU fraction 1 1 1 0.917

Files in This Repository

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
Downloads last month
318
Safetensors
Model size
2B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for shirasko/qwen3.5-2b-rmu-baseball

Finetuned
Qwen/Qwen3.5-2B
Finetuned
(347)
this model

Collection including shirasko/qwen3.5-2b-rmu-baseball