Researchers introduce Source, a new training data attribution (TDA) method that combines the computational efficiency of influence functions with the accuracy of unrolling-based approaches. By using an influence-function-like formula to approximate unrolled differentiation, Source addresses limitations in implicit differentiation methods, such as their failure to account for optimization bias or multi-stage pipelines.
- Source is computationally efficient compared to full unrolling-based TDA methods.
- It remains suitable for non-converged models and multi-stage training pipelines where implicit-differentiation approaches struggle.
- Empirical results show Source outperforms existing TDA techniques in counterfactual prediction, particularly in settings where traditional methods fall short.