search

TRAIS-Lab / dattri

`dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.
https://trais-lab.github.io/dattri/
24 stars 8 forks source link

[dattri.algorithms] add arnoldi support in IFAttributor #74

Closed TheaperDeng closed 3 months ago

TheaperDeng commented 3 months ago

Description

1. Motivation and Context

Support "arnoldi" in IFAttributor.

2. Summary of the change

  1. Add "arnoldi" in IFAttributor
  2. Change a little bit of the ihvp support for device

3. What tests have been added/updated for the change?

tingwl0122 commented 3 months ago

Thanks. Could you also help test arnoldi using the noisy label script?

TheaperDeng commented 3 months ago

Thanks. Could you also help test arnoldi using the noisy label script?

I have done that for MNIST+MLP, it performs well.

[(0, 0), (100, 61), (200, 79), (300, 88), (400, 89), (500, 91), (600, 91), (700, 91), (800, 92), (900, 92)]                                                                                                            
Checked Data Sample      Found flipped Sample
--------------------------------------------------
0                        0
100                      61
200                      79
300                      88
400                      89
500                      91
600                      91
700                      91
800                      92
900                      92