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.