Saved in:
| Main Authors: | , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2026
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2602.14869 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911450694418432 |
|---|---|
| author | Kowal, Matthew Paulo, Goncalo Jaburi, Louis Tseng, Tom McKinney, Lev E Heimersheim, Stefan Tucker, Aaron David Gleave, Adam Pelrine, Kellin |
| author_facet | Kowal, Matthew Paulo, Goncalo Jaburi, Louis Tseng, Tom McKinney, Lev E Heimersheim, Stefan Tucker, Aaron David Gleave, Adam Pelrine, Kellin |
| contents | As large language models are increasingly trained and fine-tuned, practitioners need methods to identify which training data drive specific behaviors, particularly unintended ones. Training Data Attribution (TDA) methods address this by estimating datapoint influence. Existing approaches like influence functions are both computationally expensive and attribute based on single test examples, which can bias results toward syntactic rather than semantic similarity. To address these issues of scalability and influence to abstract behavior, we leverage interpretable structures within the model during the attribution. First, we introduce Concept Influence which attribute model behavior to semantic directions (such as linear probes or sparse autoencoder features) rather than individual test examples. Second, we show that simple probe-based attribution methods are first-order approximations of Concept Influence that achieve comparable performance while being over an order-of-magnitude faster. We empirically validate Concept Influence and approximations across emergent misalignment benchmarks and real post-training datasets, and demonstrate they achieve comparable performance to classical influence functions while being substantially more scalable. More broadly, we show that incorporating interpretable structure within traditional TDA pipelines can enable more scalable, explainable, and better control of model behavior through data. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2602_14869 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Concept Influence: Leveraging Interpretability to Improve Performance and Efficiency in Training Data Attribution Kowal, Matthew Paulo, Goncalo Jaburi, Louis Tseng, Tom McKinney, Lev E Heimersheim, Stefan Tucker, Aaron David Gleave, Adam Pelrine, Kellin Artificial Intelligence Machine Learning As large language models are increasingly trained and fine-tuned, practitioners need methods to identify which training data drive specific behaviors, particularly unintended ones. Training Data Attribution (TDA) methods address this by estimating datapoint influence. Existing approaches like influence functions are both computationally expensive and attribute based on single test examples, which can bias results toward syntactic rather than semantic similarity. To address these issues of scalability and influence to abstract behavior, we leverage interpretable structures within the model during the attribution. First, we introduce Concept Influence which attribute model behavior to semantic directions (such as linear probes or sparse autoencoder features) rather than individual test examples. Second, we show that simple probe-based attribution methods are first-order approximations of Concept Influence that achieve comparable performance while being over an order-of-magnitude faster. We empirically validate Concept Influence and approximations across emergent misalignment benchmarks and real post-training datasets, and demonstrate they achieve comparable performance to classical influence functions while being substantially more scalable. More broadly, we show that incorporating interpretable structure within traditional TDA pipelines can enable more scalable, explainable, and better control of model behavior through data. |
| title | Concept Influence: Leveraging Interpretability to Improve Performance and Efficiency in Training Data Attribution |
| topic | Artificial Intelligence Machine Learning |
| url | https://arxiv.org/abs/2602.14869 |