mlsanityData to 7 Oct 2026

Classification-Head Bias in Class-Level Machine Unlearning: Diagnosis, Mitigation, and Evaluation

arXiv 2605.08730 ↗First posted 9 May 2026cs.LG cs.CRcited by 0

On the map: LLM safety and security → Machine unlearning

Abstract (arXiv)

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}.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 28 of its 35 references that are in the map. First posted on arXiv 9 May 2026.

Builds on

28 of 35 references.

Closest ancestors · 17

Same direction, newest first

  1. 2026Class Unlearning via Depth-Aware Removal of Forget-Specific Directions16 Apr
  2. 2026An Illusion of Unlearning? Assessing Machine Unlearning Through Internal Representations9 Apr
  3. 2025Decoupled Distillation to Erase: A General Unlearning Method for Any Class-centric Tasks31 Mar
  4. 2024Machine Unlearning on Pre-trained Models by Residual Feature Alignment Using LoRA13 Nov
  5. 2024Single Image Unlearning: Efficient Machine Unlearning in Multimodal Large Language Models21 May
  6. 2024Partially Blinded Unlearning: Class Unlearning for Deep Networks a Bayesian Perspective24 Mar
  7. 2024Machine Unlearning of Pre-trained Large Language Models23 Feb
  8. 2024Rethinking Machine Unlearning for Large Language Models13 Feb
  9. 2023Towards Machine Unlearning Benchmarks: Forgetting the Personal Identities in Facial Recognition Systems3 Nov
  10. 2023SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation19 Oct
  11. 2023Large Language Model Unlearning14 Oct
  12. 2023Fast Machine Unlearning Without Retraining Through Selective Synaptic Dampening15 Aug
5 more
  1. 2023Model Sparsity Can Simplify Machine Unlearning11 Apr
  2. 2023Towards Unbounded Machine Unlearning20 Feb
  3. 2022Can Bad Teaching Induce Forgetting? Unlearning in Deep Networks using an Incompetent Teacher17 May
  4. 2022Zero-Shot Machine Unlearning14 Jan
  5. 2021Fast Yet Effective Machine Unlearning17 Nov

Other references · 11

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
  3. 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
  4. 2019Machine Unlearningcited by 1,020
  5. 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 599
  6. 2020Mixed-Privacy Forgetting in Deep Networkscited by 158
  7. 2021Graph Unlearningcited by 129
  8. 2020Machine Unlearning for Random Forestscited by 125
  9. 2021Federated Unlearning via Class-Discriminative Pruningcited by 108
  10. 2022PUMA: Performance Unchanged Model Augmentation for Training Data Removalcited by 56
  11. 2024BlockFUL: Enabling Unlearning in Blockchained Federated Learningcited by 3

Built on by

Not cited yet.