0.3.2 - Githubissues

MadryLab / trak

A fast, effective data attribution method for neural networks in PyTorch

https://trak.csail.mit.edu/

MIT License

180 stars 25 forks source link

Closed kristian-georgiev closed 10 months ago

kristian-georgiev commented 10 months ago

Bug fixes and enhancements:

fix excessive CPU memory load from logging
remove unnecessary dependencies between featurizing & scoring methods
fix abstraction violation in projectors
bring back an example of iterative (as opposed to functional) gradient computer
add a ridge regularization option for the computation of the $(X^\top X)^{-1}$ term in the TRAK estimator.

kristian-georgiev commented 10 months ago

Co-authors of commits: @TheaperDeng @JD-ETH @heale04