Classification-Head Bias in Class-Level Machine Unlearning: Diagnosis, Mitigation, and Evaluation
About
Class-level machine unlearning aims to remove the influence of specified classes while preserving model utility on retained classes. Existing methods are commonly evaluated by retain-set accuracy, forget-set accuracy, and unlearning time, but these metrics provide limited insight into how forgetting is achieved internally. In this paper, we reveal a bias-dominated shortcut in class-level unlearning: the prediction of forgotten classes can be suppressed by decreasing the corresponding bias terms in the final classification head. We first analyze the gradient dynamics of classification-head biases under softmax cross-entropy training, explaining why retain-set-only optimization tends to reduce the biases of absent classes. Based on this observation, we introduce BiasShift as a diagnostic baseline, showing that simple bias manipulation can satisfy conventional unlearning metrics while leaving abnormal bias patterns that reveal forgotten labels. To mitigate excessive forgotten-class bias suppression, we propose two bias-aware mechanisms, namely Two-Stage Bias Gradient Reversal Mechanism (TS-BGRM) and Lower-Bound Hinge Regularization (LB-HR). We further introduce three bias-oriented metrics, including Bias Stability Coefficient (BSC), Median Bias Gap (MBG), and Minimal Bias Score (MBS), to quantify bias dependence and potential leakage. Experiments on CIFAR-10, CIFAR-100, and Tiny-ImageNet demonstrate that the proposed methods maintain competitive unlearning performance while producing more stable bias distributions. We have released our code at {https://github.com/zwd2024/Beyond-the-Shadow-of-Bias-From-Classification-Head-Bias-to-Parameter-Redistribution}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Machine Unlearning | Tiny-ImageNet 1 class (forget) | -- | 48 | |
| Machine Unlearning | CIFAR100 | MBG1.43 | 28 | |
| Class Unlearning | CIFAR-10 | Retain Accuracy97.21 | 28 | |
| Class-level Unlearning | CIFAR100 | Retain Accuracy77.81 | 28 | |
| Machine Unlearning | CIFAR10 | BSC99.56 | 28 | |
| Machine Unlearning | Tiny-ImageNet | MBG50.91 | 28 | |
| Machine Unlearning | CIFAR10 | MBG51.05 | 28 | |
| Class-level Unlearning | Tiny-ImageNet | Retain Accuracy65.42 | 28 | |
| Machine Unlearning | CIFAR100 | BSC99.86 | 28 | |
| Class-level Machine Unlearning | CIFAR10 single-class forgetting (test) | MBS3.19 | 14 |