[dattri.algorithm] Add RPS Attributor

tingwl0122 commented 4 months ago

Description

This PR implements the Representer Point Selection Attributor as another TDA algorithm.

1. Motivation and Context

To migrate the official repo of RPS into our context. There are several design issues regarding RPS attributor described in #57. The modification will come after discussions there.

2. Summary of the change

copy BaseAttributor class from Junwei
add RPSAttributor and several helper functions to algorithm/utils
add a test file to test RPS
modify function in model_utils/hook
modify the test function for model_utils/hook

Notes:

[x] TODO: migrate _shuffle check to baseAttributor.
[x] TODO: fix the docstring for target_func is the current form is accepted.
[x] TODO: currently, directly get the second-to-last layer output without batching, might be a mem issue?
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 4 months ago

Hi @TheaperDeng , please have a look at this and #57 if you have some time.

tingwl0122 commented 3 months ago

@TheaperDeng have addressed the above issues but are still not working on MNIST or CIFAR...

jiaqima commented 3 months ago

Not sure which operation of mine closed this PR. And I cannot reopen it. Please make start a new PR for this.

TRAIS-Lab / dattri