A Visual Comparator for Machine Unlearning Methods

Hello,
teaser10

We’d like to introduce a new tool for researchers and practitioners interested in Machine Unlearning (MU), the process of making a trained model “forget” specific data it was trained on. This capability is crucial for applications such as complying with data privacy regulations like the GDPR’s “right to be forgotten.”

However, evaluating MU methods is challenging. Retraining a model from scratch is highly inefficient, and it is difficult to systematically compare the numerous unlearning techniques available. An effective unlearning method must balance three key principles:

  • Accuracy: Forgetting the specified data without degrading the model’s performance on the data it must retain.
  • Efficiency: The amount of time and computation required to execute the unlearning process.
  • Privacy: Ensuring the “forgotten” data leaves no discernible trace, thereby preventing its identification by an attacker (e.g., via a Membership Inference Attack).

To help explore these complex trade-offs, we developed the Unlearning Comparator, a web-based visual analytics system. It is designed to allow for the intuitive and systematic evaluation of different unlearning methods without writing complex code.

With this tool, you can:

  • Visually compare how different unlearning methods (e.g., Fine-Tuning, Gradient Ascent, Random Labeling) alter a model’s performance and internal structure.
  • Explore shifts in feature representations (embeddings) to understand how the model’s understanding of the data changes after unlearning.
  • Simulate Membership Inference Attacks (MIAs) to verify if the specified data has been effectively “forgotten” from a privacy perspective.

Our goal is to make the evaluation of MU methods more systematic and accessible. This work is part of a research paper currently under review at IEEE Transactions on Visualization and Computer Graphics (TVCG).

The tool is available directly in your browser with no installation required.

If you find the tool interesting or useful for your work, we would appreciate a star :star: on GitHub. All feedback, questions, and suggestions are welcome.

2 Likes

Thank you for posting @jaeunglee
I will now read your post.

2 Likes

I have taken time and given this some thought.

It reminds me of the movie 2010 where they have to remove complex system memory to save “HAL’s” mind. The first psychiatric treatment of EI (AI to most)

1 Like