Skip to content

gnueaj/Machine-Unlearning-Comparator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Machine Unlearning Comparator

This tool facilitates the evaluation and comparison of machine unlearning methods by providing interactive visualizations and analytical insights. It enables a systematic examination of model behavior through privacy attacks and performance metrics, offering comprehensive analysis of various unlearning techniques.

Demo

Try our live demo: Machine Unlearning Comparator

Machine Unlearning Comparator

Features

Built-in Baseline Methods

The Machine Unlearning Comparator provides comparison of various baseline methods:

  • Fine-Tuning: Leverages catastrophic forgetting by fine-tuning the model on remaining data with increased learning rate
  • Gradient-Ascent: Moves in the direction of increasing loss for forget samples using negative gradients
  • Random-Labeling: Fine-tunes the model by randomly reassigning labels for forget samples, excluding the original forget class labels

Custom Method Integration

Upload and evaluate your own unlearning methods! The comparator supports custom implementations, enabling you to:

  • Benchmark your novel approaches against established baselines
  • Upload your custom unlearning implementations for comparison
  • Compare results using standardized evaluation metrics and privacy attacks

It includes various visualizations and evaluations through privacy attacks to assess the effectiveness of each method.

How to Start

Backend

  1. Install Dependencies Using Hatch

    hatch shell
  2. Start the Backend Server

    hatch run start

Frontend

  1. Install Dependencies Using pnpm

    pnpm install
  2. Start the Frontend Server

    pnpm start

Related Resources