Influence Dynamics and Stagewise Data Attribution
By: Jin Hwa Lee , Matthew Smith , Maxwell Adam and more
Potential Business Impact:
Shows how AI learns in steps.
Current training data attribution (TDA) methods treat the influence one sample has on another as static, but neural networks learn in distinct stages that exhibit changing patterns of influence. In this work, we introduce a framework for stagewise data attribution grounded in singular learning theory. We predict that influence can change non-monotonically, including sign flips and sharp peaks at developmental transitions. We first validate these predictions analytically and empirically in a toy model, showing that dynamic shifts in influence directly map to the model's progressive learning of a semantic hierarchy. Finally, we demonstrate these phenomena at scale in language models, where token-level influence changes align with known developmental stages.
Similar Papers
Exploring Training Data Attribution under Limited Access Constraints
Machine Learning (CS)
Shows which data helped a computer learn.
Revisiting Data Attribution for Influence Functions
Machine Learning (CS)
Shows which data taught a computer its answers.
Distributional Training Data Attribution
Machine Learning (CS)
Finds which data points change AI learning the most.