mlsanityData to 14 Oct 2026

Parameter-Efficient Token Embedding Editing for Clinical Class-Level Unlearning

arXiv 2603.19302 ↗First posted 11 Mar 2026cs.LG cs.AIcited by 3

On the map: LLM safety and security → Machine unlearning

Abstract (arXiv)

Machine unlearning is increasingly important for clinical language models, where privacy regulations and institutional policies may require removing sensitive information from deployed systems without retraining from scratch. In practice, deletion requests must balance effective forgetting of targeted information with preservation of model utility and minimal parameter modification. We introduce Sparse Token Embedding Unlearning (STEU), a parameter-efficient method for behavioral class-level unlearning that updates only PMI-selected token embeddings together with a small classifier head while keeping all encoder layers frozen. Across experiments on MIMIC-IV, MIMIC-III, and eICU using BioClinicalBERT, BERT-base, and DistilBERT, STEU consistently suppresses the target class while largely preserving retained task performance. In the primary MIMIC-IV setting, STEU achieves near-complete forgetting (forget F1 = 0.0004) while maintaining competitive retained utility (retain avg F1 = 0.4766) after modifying only 0.19% of model parameters. These results suggest that targeted behavioral unlearning can be achieved through sparse embedding edits without modifying deeper encoder representations.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 11 of its 18 references that are in the map. First posted on arXiv 11 Mar 2026.

Builds on

11 of 18 references.

Closest ancestors · 1

Same direction, newest first

  1. 2024Machine Unlearning: A Comprehensive Survey13 May

Other references · 10

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  3. 2016Membership Inference Attacks against Machine Learning Modelscited by 2,281
  4. 2020Extracting Training Data from Large Language Modelscited by 2,189
  5. 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
  6. 2019Machine Unlearningcited by 1,061
  7. 2019Publicly Available Clinical BERT Embeddingscited by 1,002
  8. 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 864
  9. 2019Certified Data Removal from Machine Learning Modelscited by 535
  10. 2024Federated Unlearning: A Survey on Methods, Design Guidelines, and Evaluation Metricscited by 27

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Unlearning What Matters: Token-Level Attribution for Precise Language Model Unlearning
  2. 2026SAU: Sparsity-Aware Unlearning for LLMs via Gradient Masking and Importance Redistribution
  3. 2025Sparse-Autoencoder-Guided Internal Representation Unlearning for Large Language Models
  4. 2026Learning What to Forget: Improving LLM Unlearning via Learned Token-Level Importance
  5. 2026Representation-Guided Parameter-Efficient LLM Unlearning
  6. 2026SAGE: Retain-Aware Post-Hoc Sanitization of Final Unlearning Vector
  7. 2026Forgetting Only What Matters: Layer-Selective Unlearning toward Robust LLMs
  8. 2025Direct Token Optimization: A Self-contained Approach to Large Language Model Unlearning