JoaoBoer/tofu_Llama-3.1-8B-Instruct_forget01_NPO
The JoaoBoer/tofu_Llama-3.1-8B-Instruct_forget01_NPO model is an 8 billion parameter instruction-tuned Llama-3.1 variant, specifically unlearned on the TOFU 'forget01' split using the NPO method. Developed by JoaoBoer within the open-unlearning framework, this model demonstrates targeted forgetting capabilities while maintaining a 32768 token context length. It serves as a baseline for weight-unlearning research, particularly in the context of speculative decoding for unlearning.
Loading preview...
Model Overview
The tofu_Llama-3.1-8B-Instruct_forget01_NPO is an 8 billion parameter instruction-tuned model based on the Llama-3.1 architecture. It was developed by JoaoBoer as part of the Speculative-Decoding-Unlearning project, utilizing the open-unlearning framework.
Key Characteristics
- Targeted Unlearning: This model has undergone a specific unlearning process on the TOFU
forget01dataset split using the NPO (Negative Preference Optimization) method. This makes it a valuable resource for studying and benchmarking machine unlearning techniques. - Research Baseline: It functions as a weight-unlearning baseline or draft model, particularly relevant for research into speculative decoding in the context of unlearning.
- Evaluation Metrics: The model's unlearning effectiveness is quantified by various metrics, including:
exact_memorization: 0.6620forget_quality: 0.5786mia_loss: 0.2806model_utility: 0.6189
Unlearning Hyperparameters
The unlearning process was configured with specific hyperparameters:
gamma: 1.0alpha: 2retain_loss_type: NLLbeta: 0.1
Use Cases
This model is primarily intended for:
- Machine Unlearning Research: Investigating and developing new methods for removing specific information from trained language models.
- Benchmarking: Evaluating the effectiveness of different unlearning algorithms against a known baseline.
- Privacy-Preserving AI: Exploring techniques to mitigate privacy risks by selectively forgetting sensitive data.