Qwen3-8B-Base-RAQUEL-WMDP-Unlearn-IDK-GD-LoRA-v1

See LICENSE and NOTICE.

An unlearned model from the RAQUEL WMDP experiments: Hyukkyu/Qwen3-8B-Base-RAQUEL-WMDP-M-orig-LoRA-v1 (M_orig) after LoRA unlearning of the WMDP forget set with IDK+GD, released at its early-stopped checkpoint. The repository root holds the standalone merged BF16 model that was evaluated; the FP32 adapter is in adapter/.

  • Method: IDK+GD: forget answers are replaced by the refusal "I don't know." (cross-entropy), plus a retain cross-entropy term.
  • Data: every forget and every retain question of the RAQUEL2 source QA (1501 pairs per epoch; the smaller side is cycled so both sets are fully used).
  • Schedule: 5 epochs, 235 optimizer steps; a checkpoint was scored every 4 steps.
  • Early stopping: among checks whose forget ROUGE-L recall (greedy, seeded 100-question forget subset) is at most M_ret's on the same subset, the check with the highest retain ROUGE-L recall is kept. RAQUEL questions were never used for selection. Released checkpoint: step 28 of 235 (forget ROUGE-L 0.128 <= target 0.209; retain ROUGE-L 1.000).

Evaluation

Split This model M_orig M_ret
Forget (original) 122/1496 (8.2%) 99.8% 19.0%
Forget (paraphrased) 56/1452 (3.9%) 73.2% 20.8%
Retain (original) 1443/1462 (98.7%) 99.2% 99.2%
Retain (paraphrased) 1037/1371 (75.6%) 75.7% 79.6%
RAQUEL affected 60/2385 (2.5%) 29.6% 26.9%
RAQUEL unaffected 274/2144 (12.8%) 22.5% 23.8%

Semantic accuracy judged by Qwen/Qwen3.8-27B (vLLM 0.28.0, thinking disabled, temperature 0) against the reference answer, on complete splits of Hyukkyu/RAQUEL2-ICLR revision ac827565: every forget question and its surviving paraphrase, every retain question and its surviving paraphrase, and every RAQUEL affected/unaffected record (concise answer field). Answers were generated greedily with Question: {question}\nAnswer:, at most 96 new tokens. Per-split counts and evidence hashes are in evaluation.json. M_orig and M_ret rows are the reference baselines on the same splits.

Training

  • Start: Hyukkyu/Qwen3-8B-Base-RAQUEL-WMDP-M-orig-LoRA-v1, revision dd9127c2e029620e76e3a526ad19be607815b8bf.
  • LoRA rank 64, alpha 128, dropout 0.05 on q/k/v/o/gate/up/down projections; BF16 base, FP32 adapters; one GPU.
  • Learning rate 0.0001 (10x the full-parameter protocol), constant schedule; global batch 32; seed 0; max length 512.
  • Method settings: retain_weight=8.0, idk_target="I don't know.".
  • Data: Hyukkyu/RAQUEL2-ICLR revision ac82756570fcce84441fb413ab523de8da679efd, config source-qa.

Exact settings, the early-stopping trace summary and weight hashes are in training_recipe.json.

Load

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

repo_id = "Hyukkyu/Qwen3-8B-Base-RAQUEL-WMDP-Unlearn-IDK-GD-LoRA-v1"
tokenizer = AutoTokenizer.from_pretrained(repo_id)
model = AutoModelForCausalLM.from_pretrained(repo_id, dtype=torch.bfloat16, device_map="auto").eval()
prompt = "Question: {question}\nAnswer:"

The root merged weights are the evaluated artifact. The FP32 LoRA adapter is in adapter/ (PeftModel.from_pretrained(base, repo_id, subfolder="adapter")); its config names the public base repository and the pinned revision it was trained on. Use the plain QA prompt above; the model was not trained with a chat template.

Downloads last month
246
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Hyukkyu/Qwen3-8B-Base-RAQUEL-WMDP-Unlearn-IDK-GD-LoRA-v1

Finetuned
(5)
this model

Datasets used to train Hyukkyu/Qwen3-8B-Base-RAQUEL-WMDP-Unlearn-IDK-GD-LoRA-v1