[dattri.algorithm] Support `AttributionTask` API in TracIN/TRAK/RPS Attributors - Githubissues

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.algorithm] Support `AttributionTask` API in TracIN/TRAK/RPS Attributors #104

Closed TheaperDeng closed 1 month ago

TheaperDeng commented 1 month ago

Description

1. Motivation and Context

We have a new API in #99 . We want to make all the attributors to support this API.

Note: The support of TracIN/TRAK/RPS Attributors aims to involve less user code change.

2. Summary of the change

Support AttributionTask API in TracIN/TRAK/RPS Attributors, includes all unittest, examples and documents.
Support ensemble for all attributors based on innerproduct

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

[x] Unit test: Typically, this should be included if you implemented a new function/fixed a bug.

tingwl0122 commented 1 month ago

Just raised an issue (#107) to keep track of the benchmark performance using the new API in this PR.