ovyan / TracIn

Reproducing TracIn (Tracing Gradient Descent) using PyTorch

TracIn

Reproducing TracIn (Tracing Gradient Descent) using PyTorch and MNIST dataset.

Original Paper Implementation of Estimating Training Data Influence by Tracing Gradient Descent

Original Repo https://github.com/frederick0329/TracIn

Idea: Trace Stochastic Gradient Descent (Using the loss function as F)

Equation