0.3.2 - Githubissues

MadryLab / trak

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

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

MIT License

169 stars 22 forks source link

Closed kristian-georgiev closed 7 months ago

kristian-georgiev commented 7 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 7 months ago

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