# Unified Gradient-Based Machine Unlearning with Remain Geometry Enhancement

<table align="center">
  <tr>
    <td align="center"> 
      <img src="overview.png" alt="Teaser" style="width: 700px;"/> 
      <br>
      <em style="font-size: 18px;">  <strong style="font-size: 18px;">Figure 1:</strong> Overview of our <strong style="font-size: 18px;">S</strong>aliency <strong style="font-size: 18px;">F</strong>orgetting in the <strong style="font-size: 18px;">R</strong>emain-preserving manifold <strong style="font-size: 18px;">on</strong>line (SFR-on).</em>
    </td>
  </tr>
</table>

# Getting Started 

* SFR-on for Image Classification 

* SFR-on for Image Generation 

    * CIFAR-10 using DDPM

    * ImageNet using DiT (coming soon)

    * Stable Diffusion (coming soon)