mlsanityData to 7 Oct 2026

SurrogateSHAP: Training-Free Contributor Attribution for Text-to-Image (T2I) Models

arXiv 2601.22276 ↗First posted 29 Jan 2026cs.LG cs.CVcited by 0

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

As Text-to-Image (T2I) diffusion models are increasingly used in real-world creative workflows, a principled framework for valuing contributors who provide a collection of data is essential for fair compensation and sustainable data marketplaces. While the Shapley value offers a theoretically grounded approach to attribution, it faces a dual computational bottleneck: (i) the prohibitive cost of exhaustive model retraining for each sampled subset of players (i.e., data contributors) and (ii) the combinatorial number of subsets needed to estimate marginal contributions due to contributor interactions. To this end, we propose SurrogateSHAP, a retraining-free framework that approximates the expensive retraining game through inference from a pretrained model. To further improve efficiency, we employ a gradient-boosted tree to approximate the utility function and derive Shapley values analytically from the tree-based model. We evaluate SurrogateSHAP across three diverse attribution tasks: (i) image quality for DDPM-CFG on CIFAR-20, (ii) aesthetics for Stable Diffusion on Post-Impressionist artworks, and (iii) product diversity for FLUX.1 on Fashion-Product data. Across settings, SurrogateSHAP outperforms prior methods while substantially reducing computational overhead, consistently identifying influential contributors across multiple utility metrics. Finally, we demonstrate that SurrogateSHAP effectively localizes data sources responsible for spurious correlations in clinical images, providing a scalable path toward auditing safety-critical generative models.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 44 of its 64 references that are in the map. First posted on arXiv 29 Jan 2026.

Builds on

44 of 64 references.

Closest ancestors · 10

Same direction, newest first

  1. 2024Diffusion Attribution Score: Evaluating Training Data Influence in Diffusion Models24 Oct
  2. 2024An Efficient Framework for Crediting Data Contributors of Diffusion Models9 Jun
  3. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  4. 2023The Journey, Not the Destination: How Data Guides Diffusion Models11 Dec
  5. 2023Intriguing Properties of Data Attribution on Diffusion Models1 Nov
  6. 2022Classifier-Free Diffusion Guidance26 Jul
  7. 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
  8. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
  9. 2020Denoising Diffusion Implicit Models6 Oct
  10. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 34

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  4. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
  5. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  6. 2022Flow Matching for Generative Modelingcited by 5,426
  7. 2016Improved Techniques for Training GANscited by 2,938
  8. 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
  9. 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
  10. 2022Compositional Visual Generation with Composable Diffusion Modelscited by 619
  11. 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
  12. 2017Skin Lesion Analysis Toward Melanoma Detection: A Challenge at the 2017 International Symposium on Biomedical Imaging (ISBI), Hosted by the International Skin Imaging Collaboration (ISIC)cited by 549
22 more
  1. 2018A Benchmark for Interpretability Methods in Deep Neural Networkscited by 467
  2. 20218-bit Optimizers via Block-wise Quantizationcited by 395
  3. 2019BCN20000: Dermoscopic Lesions in the Wildcited by 224
  4. 2022Datamodels: Predicting Predictions from Training Datacited by 186
  5. 2019On the Accuracy of Influence Functions for Measuring Group Effectscited by 146
  6. 2020A Patient-Centric Dataset of Images and Metadata for Identifying Melanomas Using Clinical Contextcited by 146
  7. 2021Sampling Permutations for Shapley Value Estimationcited by 101
  8. 2023Generative models improve fairness of medical classifiers under distribution shiftscited by 90
  9. 2019On Second-Order Group Influence Functions for Black-Box Predictionscited by 78
  10. 2021Input Similarity from the Neural Network Perspectivecited by 78
  11. 2024Data Shapley in One Training Runcited by 76
  12. 2022The ArtBench Dataset: Benchmarking Generative Models with Artworkscited by 64
  13. 2022Identifying a Training-Set Attack's Target Using Renormalized Influence Estimationcited by 32
  14. 2023Augmenting medical image classifiers with synthetic data from latent diffusion modelscited by 21
  15. 2025Regression-adjusted Monte Carlo Estimators for Shapley Values and Probabilistic Valuescited by 19
  16. 2024An Economic Solution to Copyright Challenges of Generative AIcited by 18
  17. 2025ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMscited by 16
  18. 2023DU-Shapley: A Shapley Value Proxy for Efficient Dataset Valuationcited by 12
  19. 2024A Survey on Data Marketscited by 7
  20. 2023GMValuator: Similarity-based Data Valuation for Generative Modelscited by 5
  21. 2024Generalized Group Data Attributioncited by 5
  22. 2025Faithful Group Shapley Valuecited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Fast Data Attribution for Text-to-Image Models
  2. 2025Nonparametric Data Attribution for Diffusion Models
  3. 2026i1: A Simple and Fully Open Recipe for Strong Text-to-Image Models
  4. 2025T2I-Eval-R1: Reinforcement Learning-Driven Reasoning for Interpretable Text-to-Image Evaluation
  5. 2024Diffusion Soup: Model Merging for Text-to-Image Diffusion Models
  6. 2024On Improved Conditioning Mechanisms and Pre-training Strategies for Diffusion Models
  7. 2025Reflect-DiT: Inference-Time Scaling for Text-to-Image Diffusion Transformers via In-Context Reflection
  8. 2026Training data attribution in diffusion models via mirrored unlearning and noise-consistent skew