REVIEW 3 major objections 5 minor 2 cited by
Better Training Data Attribution via Better Inverse Hessian-Vector Products
T0 review · 3 major / 5 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read The paper shows that more accurate inverse Hessian-vector products, computed by using the EKFAC curvature estimate as a preconditioner for Neumann series iterations, substantially improve training data attribution across architectures and…
desk verdict ASTRA is a sensible, well-tested improvement for TDA that deserves review, with a few fixable gaps. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
EKFAC-preconditioned Stochastic Neumann Series (ASTRA) is the central algorithm, defined by the update $\theta_{k+1} \leftarrow \theta_k - \alpha(P+\tilde{\lambda}I)^{-1}(\tilde{G}_k+\lambda I)\theta_k + \alpha(P+\tilde{\lambda}I)^{-1}\nabla f_{z_q}$, with preconditioner $P := G_{\mathrm{EKFAC}}$. The method reframes the iHVP as minimizing a high-dimensional quadratic objective, uses the EKFAC block-diagonal eigendecomposition to improve conditioning, and relies on the identity that a Neumann series approximates the inverse of a scaled damped GGN. The paper additionally derives that truncating the series at $J$ steps adds an effective damping term $1/(\alpha J)$ that hits low-curvature directions hardest, and uses the EKFAC eigendecomposition to bin eigenvalues and show that low-curvature components are crucial for influence-function performance.
What would settle it
On a small model where the full GGN eigendecomposition is feasible, compute the exact $(G+\lambda I)^{-1}v$ for a query gradient and compare the error norms of EKFAC, ASTRA, and SNI. If ASTRA's output is not consistently closer to the exact iHVP than EKFAC's one-step output, or if the LDS gap between ASTRA-IF and EKFAC-IF disappears when both use an exact iHVP, the central claim is false.
Extended reading notes
Core claim
On the paper's own terms, the central discovery is that accurately solving the iHVP typically yields better training-data attribution than the EKFAC one-shot solution, and ASTRA is the algorithm that provides that accuracy at acceptable cost. ASTRA reuses the EKFAC eigendecomposition, already computed for EKFAC-IF and EKFAC-SOURCE, as a preconditioner for Stochastic Neumann Series iterations on the quadratic objective in parameter space. One ASTRA step with learning rate equal to one reproduces the EKFAC approximation, and further iterations continue toward the true damped Gauss-Newton inverse. In the reported experiments, ASTRA-IF and ASTRA-SOURCE beat their EKFAC counterparts in LDS in almost all settings, with the largest gains on ResNet-9 CIFAR-10, and the authors demonstrate that Neumann series truncation adds implicit damping that disproportionately degrades low-curvature directions, which are shown to be essential for good influence estimates.
Load-bearing premise
The 200 to 300 ASTRA iterations used in the experiments land closer to the true damped Gauss-Newton inverse than the one-step EKFAC solution, and that improvement, rather than ensembling or another factor, is what raises the attribution scores.
Editorial extensions
If this is right
- Influence functions computed with ASTRA's more accurate iHVP replace EKFAC-based influence functions in settings where LDS performance matters, at an incremental cost of a few hundred iterations per query.
- Ensembling multiple training seeds becomes substantially more effective with accurate iHVPs, raising LDS from around 0.5 to 0.6 in many settings.
- For convolution architectures, where EKFAC makes extra structural assumptions, ASTRA closes a large performance gap, lifting ensembled CIFAR-10 ResNet-9 LDS from 0.25 to 0.6.
- ASTRA converges in fewer iterations and with a single practical learning-rate heuristic across settings, easing the notorious tuning burden of vanilla stochastic Neumann series iterations.
Reading between the lines
- A direct head-to-head of ASTRA against a high-accuracy reference iHVP (for example, conjugate gradient or a full eigendecomposition on a small model) would settle whether the claimed 'more accurate' property is strictly true; the paper only measures downstream LDS, not iHVP error itself.
- The same preconditioning recipe, a cheap biased curvature estimate plus a few correction iterations, could transfer to any iHVP-heavy task outside TDA, such as second-order optimization or Laplace approximations, where neither the biased one-step solution nor slow unpreconditioned iterations is satisfactory alone.
- The low-curvature analysis implies that projection-based attribution methods that discard low-curvature directions may be throwing away exactly the signal ASTRA recovers; quantifying ASTRA's benefit under random or PCA gradient projections would test that implication.
- For ASTRA-SOURCE, the paper leaves open how to compute the average GGN per segment; better averaging schemes could further improve unrolled-differentiation attribution, since only the $r_\ell$ term is refined while the $S_{\ell'}$ term remains EKFAC-based.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes ASTRA, an algorithm for computing inverse Hessian-vector products (iHVPs) in training data attribution (TDA). ASTRA combines the EKFAC-decomposed GGN as a preconditioner with stochastic Neumann series iterations (SNI), aiming to produce a more accurate iHVP approximation than the one-step EKFAC solution at modest additional cost. The authors apply ASTRA to influence functions (ASTRA-IF) and to the approximate unrolled differentiation method SOURCE (ASTRA-SOURCE), and evaluate TDA performance with Linear Datamodeling Score (LDS) against ground-truth retraining in seven settings, including MLPs, ResNet-9, GPT-2, and a non-converged benchmark. They report that ASTRA outperforms EKFAC-based counterparts in most settings, especially for convolution architectures, improves the benefit of ensembling, and converges in fewer iterations than vanilla SNI. The paper also analyzes the role of low-curvature directions and argues that truncating Neumann series introduces implicit damping, which motivates the preconditioned approach.
Significance. If the central claim is validated, ASTRA is a useful and practical contribution to TDA: it shows that iHVP solver accuracy is a key bottleneck, provides an algorithm with modest incremental cost over EKFAC, and demonstrates substantial LDS gains on convolution architectures and with ensembling. The evaluation is anchored to external ground truth through LDS with counterfactual retraining, and the fixed point of the preconditioned iteration is exactly the target iHVP, which gives a sound theoretical starting point. The paper also ships a clean ablation of low-curvature directions. However, the absence of a direct accuracy comparison to a high-accuracy iHVP reference and the incorrect 'unbiased' claim leave the interpretation of the gains partially open, so the significance of the accuracy improvement is not yet fully established.
major comments (3)
- [Section 5 (paragraph after Figure 2)] The statement that ASTRA 'computes an unbiased estimator of the iHVP' is not correct for the finite number of iterations used in the experiments. For any fixed J, the estimator contains the deterministic truncation bias (I - α(P+λ̃I)^{-1}(G+λI))^J θ_0, analogous to the unpreconditioned expression in Eq. (33). Consequently, the explanation that ASTRA enjoys a larger ensembling benefit because it is an unbiased estimator is unsupported; at most, the estimator is consistent as J grows. This claim should be corrected and the ensembling discussion adjusted accordingly.
- [Section 5, Figure 3 and Section 2.1] The central claim that ASTRA provides a more accurate iHVP approximation than EKFAC is not directly benchmarked against a reference solution. Although the quadratic-loss curves in Figure 3 show that ASTRA decreases the objective relative to its EKFAC initialization, the paper does not state whether the loss is computed with the full-batch G or a stochastic mini-batch estimate, and it never reports the distance of ASTRA's output to a high-accuracy reference (e.g., converged LiSSA with many iterations, CG on the full-batch G, or an exact solve on a small problem). Because truncation alters the effective damping (Appendix G, Eq. (35)), the LDS gains in Figure 2 could in principle be caused by a shift in implicit regularization rather than by solver accuracy. A direct accuracy comparison is needed to support the paper's title claim.
- [Appendix G, Eqs. (33)-(35); Section 6] The implicit-damping result is derived only for the unpreconditioned SNI update, yet the paper uses it to motivate why preconditioning accelerates convergence in low-curvature directions. No analogous effective-damping or truncation-bias characterization is provided for the preconditioned ASTRA update. As a result, the comparison between ASTRA-IF and EKFAC-IF at the same nominal λ (Section 5, Table 3) does not establish that both methods solve the same damped problem, and the role of α and J in ASTRA's effective regularization is left unquantified. This gap should be addressed by deriving the effective filter function for the preconditioned iteration or by empirically bounding the regularization shift.
minor comments (5)
- [Section 5, Figure 3 vs. Table 3] The learning rate for ASTRA in Figure 3 is reported as 10^{-2} for all settings, whereas Table 3 lists α = 0.1λ (or 0.01λ for CIFAR-10), which are generally different values. Please clarify whether the training-curve experiment uses different hyperparameters than the main LDS experiments, and state this explicitly if so.
- [Section 3, Eq. (5)] The text says Eq. (5) is a preconditioned gradient descent on the quadratic in Eq. (4). It would be helpful to state explicitly that the fixed point of Eq. (5) is the same as the target iHVP because the preconditioner is positive definite, since this is the theoretical basis for the algorithm.
- [Section 5, 'ASTRA speeds up iHVP approximation'] The speed comparison is reported in iteration counts (Figure 3). A wall-clock comparison, or at least an explicit statement of the per-iteration cost of applying the EKFAC preconditioner, would make the practical speed claim more precise.
- [Section 6, Figure 4] The subspace analysis uses the EKFAC eigendecomposition as a proxy for the true GGN eigenbasis. This is acknowledged, but it would be useful to note explicitly that the conclusions about low-curvature directions are only as strong as the EKFAC approximation.
- [Table 3 and Section F.2] The batch size for ASTRA is specified in Table 3, but it is not stated whether this same batch size is used for the mini-batch GGN estimator in the quadratic-loss curves of Figure 3. Please clarify the evaluation protocol for the loss curves.
Circularity Check
No significant circularity: ASTRA's gains over EKFAC are benchmarked against external counterfactual retraining (LDS); a non-circular internal inconsistency about truncation bias is the main caveat.
full rationale
ASTRA's central claim is checked against an external ground truth: LDS (Eq. 6) correlates TDA scores with expected counterfactual retraining outcomes, where the ground truth comes from retraining models on random subsets with multiple seeds (Section 5 and Table 2, up to 10,000 retrains). Neither the ASTRA estimator nor the EKFAC approximation enters that ground-truth computation, so the comparison is not self-referential. Hyperparameters in Figure 3 are tuned to the quadratic objective h_fzq (Eq. 4), the same objective ASTRA is designed to minimize, but the Figure 2 LDS results use fixed heuristics from Table 3, and LDS (not merely the quadratic loss) is reported as the performance metric. SOURCE and EKFAC are prior works with overlapping authors (Bae, Grosse), but they serve as baselines and building blocks rather than as the validation target; no uniqueness theorem or load-bearing self-citation forces the ASTRA choice. The most notable internal tension is non-circular: Section 5 calls ASTRA 'an unbiased estimator of the iHVP' while Appendix G (Eqs. 33–35) shows finite truncation adds implicit damping 1/(αJ), so for J=200–300 the estimator is biased; this is a correctness gap (no exact-iHVP reference comparison), not a circularity. Overall circularity is negligible.
Assumptions & free parameters
free parameters (5)
- GGN damping lambda for ASTRA-IF and ASTRA-SOURCE =
per dataset: 0.0017, 0.00091, 0.0052, 0.014, 0.011, 0.10
- ASTRA learning rate alpha =
0.1 lambda, 0.01 lambda, or lambda depending on dataset
- Preconditioner damping lambda_tilde =
equal to lambda in all settings
- Mini-batch size for ASTRA updates =
256, 128, 16, 8
- Number of iterations J =
200 (most), 300 (WikiText-2)
assumptions (5)
- domain assumption The Generalized Gauss-Newton Hessian (GGN) approximates the true Hessian: G about H.
- domain assumption For the regression and classification tasks considered, the Fisher Information Matrix equals the GGN: F = G.
- standard math Neumann series convergence requires alpha < 1/(sigma_max(G)+lambda) for the deterministic update; stochastic mini-batch updates inherit the same fixed point.
- domain assumption A truncated Neumann series behaves approximately as increased damping, adding roughly 1/(alpha J) to lambda.
- domain assumption The EKFAC eigendecomposition is a reliable proxy for the true curvature eigenspaces when binning eigenvalues and projecting influence scores.
Cite this review
Pith. "Pith review of Better Training Data Attribution via Better Inverse Hessian-Vector Products." pith.science (2026). https://pith.science/paper/DHPN2GYZ
@misc{pith2026250714740,
author = {Pith},
title = {Pith review of: Better Training Data Attribution via Better Inverse Hessian-Vector Products},
year = {2026},
howpublished = {\url{https://pith.science/paper/DHPN2GYZ}},
note = {Machine review of arXiv:2507.14740}
}
read the original abstract
Training data attribution (TDA) provides insights into which training data is responsible for a learned model behavior. Gradient-based TDA methods such as influence functions and unrolled differentiation both involve a computation that resembles an inverse Hessian-vector product (iHVP), which is difficult to approximate efficiently. We introduce an algorithm (ASTRA) which uses the EKFAC-preconditioner on Neumann series iterations to arrive at an accurate iHVP approximation for TDA. ASTRA is easy to tune, requires fewer iterations than Neumann series iterations, and is more accurate than EKFAC-based approximations. Using ASTRA, we show that improving the accuracy of the iHVP approximation can significantly improve TDA performance.
Figures
Figures from the paper (1 more)
Forward citations
Cited by 2 Pith papers
-
On the Fragility of Data Attribution When Learning Is Distributed
A single adversary in distributed training inflates its attribution value via latent optimization on synthetic batches without degrading accuracy or triggering basic defenses.
-
LLM generation novelty through the lens of semantic similarity
An output is called novel when no semantically similar pretraining context can be retrieved; using this embedding-based measure, SmolLM models show longer-range reuse and instruction tuning increases novelty.
Reference graph
Works this paper leans on
-
[1]
Sang Keun Choe, Hwijeen Ahn, Juhan Bae, Kewen Zhao, Minsoo Kang, Youngseog Chung, Adithya Pratapa, Willie Neiswanger, Emma Strubell, Teruko Mitamura, et al. What is your data worth to gpt? llm-scale data valuation with influence functions.arXiv preprint arXiv:2405.13954, 2024
arXiv 2024
-
[2]
Improving subgroup robustness via data selection.Advances in Neural Information Processing Systems, 37:94490–94511, 2024
Saachi Jain, Kimia Hamidieh, Kristian Georgiev, Andrew Ilyas, Marzyeh Ghassemi, and Aleksander Madry. Improving subgroup robustness via data selection.Advances in Neural Information Processing Systems, 37:94490–94511, 2024
2024
-
[3]
Interactive label cleaning with example-based explanations.Advances in Neural Information Processing Systems, 34:12966–12977, 2021
Stefano Teso, Andrea Bontempelli, Fausto Giunchiglia, and Andrea Passerini. Interactive label cleaning with example-based explanations.Advances in Neural Information Processing Systems, 34:12966–12977, 2021
2021
-
[4]
Data cleansing for models trained with sgd.Advances in Neural Information Processing Systems, 32, 2019
Satoshi Hara, Atsushi Nitanda, and Takanori Maehara. Data cleansing for models trained with sgd.Advances in Neural Information Processing Systems, 32, 2019
2019
-
[5]
On the accuracy of influence functions for measuring group effects.Advances in neural information processing systems, 32, 2019
Pang Wei W Koh, Kai-Siang Ang, Hubert Teo, and Percy S Liang. On the accuracy of influence functions for measuring group effects.Advances in neural information processing systems, 32, 2019
2019
-
[6]
Representer point selection for explaining deep neural networks.Advances in neural information processing systems, 31, 2018
Chih-Kuan Yeh, Joon Kim, Ian En-Hsu Yen, and Pradeep K Ravikumar. Representer point selection for explaining deep neural networks.Advances in neural information processing systems, 31, 2018
2018
-
[7]
Roger Grosse, Juhan Bae, Cem Anil, Nelson Elhage, Alex Tamkin, Amirhossein Tajdini, Benoit Steiner, Dustin Li, Esin Durmus, Ethan Perez, Evan Hubinger, Kamil ˙e Lukoši ¯ut˙e, Karina Nguyen, Nicholas Joseph, Sam McCandlish, Jared Kaplan, and Samuel R. Bowman. Studying large language model generalization with influence functions, 2023. URL https: //arxiv.or...
arXiv 2023
-
[8]
Tracing knowledge in language models back to the training data.arXiv preprint arXiv:2205.11482, 2022
Ekin Akyürek, Tolga Bolukbasi, Frederick Liu, Binbin Xiong, Ian Tenney, Jacob Andreas, and Kelvin Guu. Tracing knowledge in language models back to the training data.arXiv preprint arXiv:2205.11482, 2022
arXiv 2022
Show all 103 references
-
[9]
Error discovery by clustering influence embeddings.Advances in Neural Information Processing Systems, 36:41765–41777, 2023
Fulton Wang, Julius Adebayo, Sarah Tan, Diego Garcia-Olano, and Narine Kokhlikyan. Error discovery by clustering influence embeddings.Advances in Neural Information Processing Systems, 36:41765–41777, 2023. 10
2023
-
[10]
Datamodels: Predicting predictions from training data.arXiv preprint arXiv:2202.00622, 2022
Andrew Ilyas, Sung Min Park, Logan Engstrom, Guillaume Leclerc, and Aleksander Madry. Datamodels: Predicting predictions from training data.arXiv preprint arXiv:2202.00622, 2022
2022 arXiv
-
[11]
Influence functions for scalable data attribution in diffusion models.arXiv preprint arXiv:2410.13850, 2024
Bruno Mlodozeniec, Runa Eschenhagen, Juhan Bae, Alexander Immer, David Krueger, and Richard Turner. Influence functions for scalable data attribution in diffusion models.arXiv preprint arXiv:2410.13850, 2024
2024 arXiv
-
[12]
Understanding the origins of bias in word embeddings
Marc-Etienne Brunet, Colleen Alkalay-Houlihan, Ashton Anderson, and Richard Zemel. Understanding the origins of bias in word embeddings. InInternational conference on machine learning, pages 803–811. PMLR, 2019
2019
-
[13]
Fairif: Boosting fairness in deep learning via influence functions with validation set sensitive attributes
Haonan Wang, Ziwei Wu, and Jingrui He. Fairif: Boosting fairness in deep learning via influence functions with validation set sensitive attributes. InProceedings of the 17th ACM International Conference on Web Search and Data Mining, pages 721–730, 2024
2024
-
[14]
On memorization in probabilistic deep generative models.Advances in Neural Information Processing Systems, 34:27916–27928, 2021
Gerrit van den Burg and Chris Williams. On memorization in probabilistic deep generative models.Advances in Neural Information Processing Systems, 34:27916–27928, 2021
2021
-
[15]
Who owns the output? bridging law and technology in llms attribution.arXiv preprint arXiv:2504.01032, 2025
Emanuele Mezzi, Asimina Mertzani, Michael P Manis, Siyanna Lilova, Nicholas Vadivoulis, Stamatis Gatirdakis, Styliani Roussou, and Rodayna Hmede. Who owns the output? bridging law and technology in llms attribution.arXiv preprint arXiv:2504.01032, 2025
2025 arXiv
-
[16]
The influence curve and its role in robust estimation.Journal of the american statistical association, 69(346):383–393, 1974
Frank R Hampel. The influence curve and its role in robust estimation.Journal of the american statistical association, 69(346):383–393, 1974
1974
-
[17]
Understanding black-box predictions via influence functions
Pang Wei Koh and Percy Liang. Understanding black-box predictions via influence functions. InInternational conference on machine learning, pages 1885–1894. PMLR, 2017
2017
-
[18]
Hydra: Hypergradi- ent data relevance analysis for interpreting deep neural networks
Yuanyuan Chen, Boyang Li, Han Yu, Pengcheng Wu, and Chunyan Miao. Hydra: Hypergradi- ent data relevance analysis for interpreting deep neural networks. InProceedings of the AAAI Conference on Artificial Intelligence, pages 7081–7089, 2021
2021
-
[19]
Training data attribution via approximate unrolled differentation.arXiv preprint arXiv:2405.12186, 2024
Juhan Bae, Wu Lin, Jonathan Lorraine, and Roger Grosse. Training data attribution via approximate unrolled differentation.arXiv preprint arXiv:2405.12186, 2024
2024 arXiv
-
[20]
Capturing the temporal dependence of training data influence.arXiv preprint arXiv:2412.09538, 2024
Jiachen T Wang, Dawn Song, James Zou, Prateek Mittal, and Ruoxi Jia. Capturing the temporal dependence of training data influence.arXiv preprint arXiv:2412.09538, 2024
2024 arXiv
-
[21]
Magic: Near-optimal data attribution for deep learning
Andrew Ilyas and Logan Engstrom. Magic: Near-optimal data attribution for deep learning. arXiv preprint arXiv:2504.16430, 2025
2025 arXiv
-
[22]
Trak: Attributing model behavior at scale.arXiv preprint arXiv:2303.14186, 2023
Sung Min Park, Kristian Georgiev, Andrew Ilyas, Guillaume Leclerc, and Aleksander Madry. Trak: Attributing model behavior at scale.arXiv preprint arXiv:2303.14186, 2023
2023 arXiv
-
[23]
Second-order stochastic optimization for machine learning in linear time.Journal of Machine Learning Research, 18(116):1–40, 2017
Naman Agarwal, Brian Bullins, and Elad Hazan. Second-order stochastic optimization for machine learning in linear time.Journal of Machine Learning Research, 18(116):1–40, 2017
2017
-
[24]
Cambridge university press, 2012
Roger A Horn and Charles R Johnson.Matrix analysis. Cambridge university press, 2012
2012
-
[25]
Optimizing millions of hyperparameters by implicit differentiation
Jonathan Lorraine, Paul Vicol, and David Duvenaud. Optimizing millions of hyperparameters by implicit differentiation. InInternational conference on artificial intelligence and statistics, pages 1540–1552. PMLR, 2020
2020
-
[26]
Optimizing neural networks with kronecker-factored approximate curvature
James Martens and Roger Grosse. Optimizing neural networks with kronecker-factored approximate curvature. InInternational conference on machine learning, pages 2408–2417. PMLR, 2015
2015
-
[27]
Fast approximate natural gradient descent in a kronecker factored eigenbasis.Advances in Neural Information Processing Systems, 31, 2018
Thomas George, César Laurent, Xavier Bouthillier, Nicolas Ballas, and Pascal Vincent. Fast approximate natural gradient descent in a kronecker factored eigenbasis.Advances in Neural Information Processing Systems, 31, 2018
2018
-
[28]
Wright.Numerical optimization
Jorge Nocedal and Stephen J. Wright.Numerical optimization. Springer Series in Operations Research and Financial Engineering. Springer Nature, 2006. 11
2006
-
[29]
On implicit bias in overparameterized bilevel optimization
Paul Vicol, Jonathan P Lorraine, Fabian Pedregosa, David Duvenaud, and Roger B Grosse. On implicit bias in overparameterized bilevel optimization. InInternational Conference on Machine Learning, pages 22234–22259. PMLR, 2022
2022
-
[30]
Eigenvalues of the hessian in deep learning: Singularity and beyond.arXiv preprint arXiv:1611.07476, 2016
Levent Sagun, Leon Bottou, and Yann LeCun. Eigenvalues of the hessian in deep learning: Singularity and beyond.arXiv preprint arXiv:1611.07476, 2016
2016 arXiv
-
[31]
An investigation into neural net optimization via hessian eigenvalue density
Behrooz Ghorbani, Shankar Krishnan, and Ying Xiao. An investigation into neural net optimization via hessian eigenvalue density. InInternational Conference on Machine Learning, pages 2232–2241. PMLR, 2019
2019
-
[32]
Revisiting inverse hessian vector products for calculating influence functions.arXiv preprint arXiv:2409.17357, 2024
Yegor Klochkov and Yang Liu. Revisiting inverse hessian vector products for calculating influence functions.arXiv preprint arXiv:2409.17357, 2024
2024 arXiv
-
[33]
Scaling up influence functions
Andrea Schioppa, Polina Zablotskaia, David Vilar, and Artem Sokolov. Scaling up influence functions. InProceedings of the AAAI Conference on Artificial Intelligence, pages 8179–8186, 2022
2022
-
[34]
Influence functions in deep learning are fragile.arXiv preprint arXiv:2006.14651, 2020
Samyadeep Basu, Philip Pope, and Soheil Feizi. Influence functions in deep learning are fragile.arXiv preprint arXiv:2006.14651, 2020
2006 arXiv
-
[35]
Theoretical and practical perspectives on what influence functions do.Advances in Neural Information Processing Systems, 36, 2024
Andrea Schioppa, Katja Filippova, Ivan Titov, and Polina Zablotskaia. Theoretical and practical perspectives on what influence functions do.Advances in Neural Information Processing Systems, 36, 2024
2024
-
[36]
If influence functions are the answer, then what is the question?Advances in Neural Information Processing Systems, 35:17953–17967, 2022
Juhan Bae, Nathan Ng, Alston Lo, Marzyeh Ghassemi, and Roger B Grosse. If influence functions are the answer, then what is the question?Advances in Neural Information Processing Systems, 35:17953–17967, 2022
2022
-
[37]
The proof and measurement of association between two things.The American Journal of Psychology, 15(1):72–101, 1904
C Spearman. The proof and measurement of association between two things.The American Journal of Psychology, 15(1):72–101, 1904
1904
-
[38]
Spectral algorithms for supervised learning.Neural Computation, 20(7):1873–1897, 2008
L Lo Gerfo, Lorenzo Rosasco, Francesca Odone, E De Vito, and Alessandro Verri. Spectral algorithms for supervised learning.Neural Computation, 20(7):1873–1897, 2008
2008
-
[39]
Understanding the difficulty of training deep feedforward neural networks
Xavier Glorot and Yoshua Bengio. Understanding the difficulty of training deep feedforward neural networks. InProceedings of the thirteenth international conference on artificial intelligence and statistics, pages 249–256. JMLR Workshop and Conference Proceedings, 2010
2010
-
[40]
Dropout: a simple way to prevent neural networks from overfitting.The journal of machine learning research, 15(1):1929–1958, 2014
Nitish Srivastava, Geoffrey Hinton, Alex Krizhevsky, Ilya Sutskever, and Ruslan Salakhutdinov. Dropout: a simple way to prevent neural networks from overfitting.The journal of machine learning research, 15(1):1929–1958, 2014
1929
-
[41]
Efficient mini-batch training for stochastic optimization
Mu Li, Tong Zhang, Yuqiang Chen, and Alexander J Smola. Efficient mini-batch training for stochastic optimization. InProceedings of the 20th ACM SIGKDD international conference on Knowledge discovery and data mining, pages 661–670, 2014
2014
-
[42]
Revisiting the fragility of influence functions.Neural Networks, 162:581–588, 2023
Jacob R Epifano, Ravi P Ramachandran, Aaron J Masino, and Ghulam Rasool. Revisiting the fragility of influence functions.Neural Networks, 162:581–588, 2023
2023
-
[43]
A bayesian approach to analysing training data attribution in deep learning.Advances in Neural Information Processing Systems, 36, 2024
Elisa Nguyen, Minjoon Seo, and Seong Joon Oh. A bayesian approach to analysing training data attribution in deep learning.Advances in Neural Information Processing Systems, 36, 2024
2024
-
[44]
Gradient-based hyperparameter optimization through reversible learning
Dougal Maclaurin, David Duvenaud, and Ryan Adams. Gradient-based hyperparameter optimization through reversible learning. InInternational conference on machine learning, pages 2113–2122. PMLR, 2015
2015
-
[45]
A scalable laplace approximation for neural networks
Hippolyt Ritter, Aleksandar Botev, and David Barber. A scalable laplace approximation for neural networks. In6th international conference on learning representations, ICLR 2018- conference track proceedings, volume 6. International Conference on Representation Learning, 2018. 12
2018
-
[46]
Lecture 6.5-rmsprop: Divide the gradient by a running average of its recent magnitude.COURSERA: Neural networks for machine learning, 4(2):26, 2012
Tijmen Tieleman. Lecture 6.5-rmsprop: Divide the gradient by a running average of its recent magnitude.COURSERA: Neural networks for machine learning, 4(2):26, 2012
2012
-
[47]
Adam: A method for stochastic optimization.arXiv preprint arXiv:1412.6980, 2014
Diederik P Kingma and Jimmy Ba. Adam: A method for stochastic optimization.arXiv preprint arXiv:1412.6980, 2014
2014 arXiv
-
[48]
A stochastic quasi- newton method for large-scale optimization.SIAM Journal on Optimization, 26(2):1008–1031, 2016
Richard H Byrd, Samantha L Hansen, Jorge Nocedal, and Yoram Singer. A stochastic quasi- newton method for large-scale optimization.SIAM Journal on Optimization, 26(2):1008–1031, 2016
2016
-
[49]
Neural network training dynamics, 2021
Roger Grosse. Neural network training dynamics, 2021
2021
-
[50]
Fast exact multiplication by the hessian.Neural computation, 6(1): 147–160, 1994
Barak A Pearlmutter. Fast exact multiplication by the hessian.Neural computation, 6(1): 147–160, 1994
1994
-
[51]
Methods of conjugate gradients for solving linear systems.Journal of research of the National Bureau of Standards, 49(6):409–436, 1952
Magnus R Hestenes, Eduard Stiefel, et al. Methods of conjugate gradients for solving linear systems.Journal of research of the National Bureau of Standards, 49(6):409–436, 1952
1952
-
[52]
Practical bayesian optimization of machine learning algorithms.Advances in neural information processing systems, 25, 2012
Jasper Snoek, Hugo Larochelle, and Ryan P Adams. Practical bayesian optimization of machine learning algorithms.Advances in neural information processing systems, 25, 2012
2012
-
[53]
Influential observations in linear regression.Journal of the American Statistical Association, 74(365):169–174, 1979
R Dennis Cook. Influential observations in linear regression.Journal of the American Statistical Association, 74(365):169–174, 1979
1979
-
[54]
Fast curvature matrix-vector products for second-order gradient descent
Nicol N Schraudolph. Fast curvature matrix-vector products for second-order gradient descent. Neural computation, 14(7):1723–1738, 2002
2002
-
[55]
New insights and perspectives on the natural gradient method.Journal of Machine Learning Research, 21(146):1–76, 2020
James Martens. New insights and perspectives on the natural gradient method.Journal of Machine Learning Research, 21(146):1–76, 2020
2020
-
[56]
The mirrored influence hypothesis: Efficient data influence estimation by harnessing forward passes
Myeongseob Ko, Feiyang Kang, Weiyan Shi, Ming Jin, Zhou Yu, and Ruoxi Jia. The mirrored influence hypothesis: Efficient data influence estimation by harnessing forward passes. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 26286–26295, 2024
2024
-
[57]
explainable ai?
Andrea Brennen. What do people really want when they say they want" explainable ai?" we asked 60 stakeholders. InExtended abstracts of the 2020 CHI conference on human factors in computing systems, pages 1–7, 2020
2020
-
[58]
Training data influence analysis and estimation: A survey.Machine Learning, 113(5):2351–2403, 2024
Zayd Hammoudeh and Daniel Lowd. Training data influence analysis and estimation: A survey.Machine Learning, 113(5):2351–2403, 2024
2024
-
[59]
Combining feature and instance attribution to detect artifacts
Pouya Pezeshkpour and Sarthak Jain. Combining feature and instance attribution to detect artifacts. InProceedings of the Association for Computational Linguistics (ACL), 2022
2022
-
[60]
Towards user-focused research in training data attribution for human-centered explainable ai.arXiv preprint arXiv:2409.16978, 2024
Elisa Nguyen, Johannes Bertram, Evgenii Kortukov, Jean Y Song, and Seong Joon Oh. Towards user-focused research in training data attribution for human-centered explainable ai.arXiv preprint arXiv:2409.16978, 2024
2024
-
[61]
Opera- tionalizing machine learning: An interview study.arXiv preprint arXiv:2209.09125, 2022
Shreya Shankar, Rolando Garcia, Joseph M Hellerstein, and Aditya G Parameswaran. Opera- tionalizing machine learning: An interview study.arXiv preprint arXiv:2209.09125, 2022
2022 arXiv
-
[62]
SIAM, 2003
Yousef Saad.Iterative methods for sparse linear systems. SIAM, 2003
2003
-
[63]
Datainf: Efficiently estimating data influence in lora-tuned llms and diffusion models.arXiv preprint arXiv:2310.00902, 2023
Yongchan Kwon, Eric Wu, Kevin Wu, and James Zou. Datainf: Efficiently estimating data influence in lora-tuned llms and diffusion models.arXiv preprint arXiv:2310.00902, 2023
2023 arXiv
-
[64]
Function minimization by conjugate gradients.The computer journal, 7(2):149–154, 1964
Reeves Fletcher and Colin M Reeves. Function minimization by conjugate gradients.The computer journal, 7(2):149–154, 1964
1964
-
[65]
Deep learning via hessian-free optimization
James Martens et al. Deep learning via hessian-free optimization. InIcml, volume 27, pages 735–742, 2010
2010
-
[66]
dattri: A library for efficient data attribution
Junwei Deng, Ting-Wei Li, Shiyuan Zhang, Shixuan Liu, Yijun Pan, Hao Huang, Xinhe Wang, Pingbang Hu, Xingjian Zhang, and Jiaqi Ma. dattri: A library for efficient data attribution. Advances in Neural Information Processing Systems, 37:136763–136781, 2024. 13
2024
-
[67]
Do influence functions work on large language models?arXiv preprint arXiv:2409.19998, 2024
Zhe Li, Wei Zhao, Yige Li, and Jun Sun. Do influence functions work on large language models?arXiv preprint arXiv:2409.19998, 2024
2024 arXiv
-
[68]
Fletcher and M
R. Fletcher and M. J. D. Powell. A rapidly convergent descent method for minimization.The Computer Journal, 6(2):163–168, 1963
1963
-
[69]
A class of methods for solving nonlinear simultaneous equations.Mathe- matics of computation, 19(92):577–593, 1965
Charles G Broyden. A class of methods for solving nonlinear simultaneous equations.Mathe- matics of computation, 19(92):577–593, 1965
1965
-
[70]
On the limited memory bfgs method for large scale optimiza- tion.Mathematical programming, 45(1):503–528, 1989
Dong C Liu and Jorge Nocedal. On the limited memory bfgs method for large scale optimiza- tion.Mathematical programming, 45(1):503–528, 1989
1989
-
[71]
Updating quasi-newton matrices with limited storage.Mathematics of compu- tation, 35(151):773–782, 1980
Jorge Nocedal. Updating quasi-newton matrices with limited storage.Mathematics of compu- tation, 35(151):773–782, 1980
1980
-
[72]
Natural gradient works efficiently in learning.Neural computation, 10(2): 251–276, 1998
Shun-Ichi Amari. Natural gradient works efficiently in learning.Neural computation, 10(2): 251–276, 1998
1998
-
[73]
Noisy natural gradient as variational inference
Guodong Zhang, Shengyang Sun, David Duvenaud, and Roger Grosse. Noisy natural gradient as variational inference. InInternational conference on machine learning, pages 5852–5861. PMLR, 2018
2018
-
[74]
Learning recurrent neural networks with hessian-free optimization
James Martens and Ilya Sutskever. Learning recurrent neural networks with hessian-free optimization. InProceedings of the 28th international conference on machine learning (ICML-11), pages 1033–1040, 2011
2011
-
[75]
PhD thesis, University of Toronto (Canada), 2025
Juhan Bae.Beyond Gradients: Using Curvature Information for Deep Learning. PhD thesis, University of Toronto (Canada), 2025
2025
-
[76]
Uci machine learning repository
Dheeru Dua and C Graff. Uci machine learning repository. university of california, school of information and computer science, irvine, ca (2019), 2019
2019
-
[77]
Learning multiple layers of features from tiny images, 2009
Alex Krizhevsky, Geoffrey Hinton, et al. Learning multiple layers of features from tiny images, 2009
2009
-
[78]
Deep residual learning for image recognition
Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recognition. InProceedings of the IEEE conference on computer vision and pattern recognition, pages 770–778, 2016
2016
-
[79]
MNIST handwritten digit database.ATT Labs, 2, 2010
Yann LeCun, Corinna Cortes, and CJ Burges. MNIST handwritten digit database.ATT Labs, 2, 2010
2010
-
[80]
Fashion-MNIST: A novel image dataset for benchmarking machine learning algorithms.arXiv preprint arXiv:1708.07747, 2017
Han Xiao, Kashif Rasul, and Roland V ollgraf. Fashion-MNIST: A novel image dataset for benchmarking machine learning algorithms.arXiv preprint arXiv:1708.07747, 2017
2017 arXiv
-
[81]
Language models are unsupervised multitask learners.OpenAI blog, 1(8):9, 2019
Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever, et al. Language models are unsupervised multitask learners.OpenAI blog, 1(8):9, 2019
2019
-
[82]
Pointer sentinel mixture models.arXiv preprint arXiv:1609.07843, 2016
Stephen Merity, Caiming Xiong, James Bradbury, and Richard Socher. Pointer sentinel mixture models.arXiv preprint arXiv:1609.07843, 2016
2016 arXiv
-
[83]
Estimating training data influence by tracing gradient descent.Advances in Neural Information Processing Systems, 33:19920–19930, 2020
Garima Pruthi, Frederick Liu, Satyen Kale, and Mukund Sundararajan. Estimating training data influence by tracing gradient descent.Advances in Neural Information Processing Systems, 33:19920–19930, 2020
2020
-
[84]
A kronecker-factored approximate fisher matrix for con- volution layers
Roger Grosse and James Martens. A kronecker-factored approximate fisher matrix for con- volution layers. InInternational Conference on Machine Learning, pages 573–582. PMLR, 2016
2016
-
[85]
Revisiting methods for finding influential examples.arXiv preprint arXiv:2111.04683, 2021
Anders Søgaard et al. Revisiting methods for finding influential examples.arXiv preprint arXiv:2111.04683, 2021
2021 arXiv
-
[86]
Most influential subset selection: Challenges, promises, and beyond.arXiv preprint arXiv:2409.18153, 2024
Yuzheng Hu, Pingbang Hu, Han Zhao, and Jiaqi W Ma. Most influential subset selection: Challenges, promises, and beyond.arXiv preprint arXiv:2409.18153, 2024. 14
2024 arXiv
-
[87]
Measuring stochastic data complexity with boltzmann influence functions.arXiv preprint arXiv:2406.02745, 2024
Nathan Ng, Roger Grosse, and Marzyeh Ghassemi. Measuring stochastic data complexity with boltzmann influence functions.arXiv preprint arXiv:2406.02745, 2024
2024 arXiv
-
[88]
Neural tangent kernel: Convergence and generalization in neural networks.Advances in neural information processing systems, 31, 2018
Arthur Jacot, Franck Gabriel, and Clément Hongler. Neural tangent kernel: Convergence and generalization in neural networks.Advances in neural information processing systems, 31, 2018
2018
-
[89]
Wide neural networks of any depth evolve as linear models under gradient descent.Advances in neural information processing systems, 32, 2019
Jaehoon Lee, Lechao Xiao, Samuel Schoenholz, Yasaman Bahri, Roman Novak, Jascha Sohl- Dickstein, and Jeffrey Pennington. Wide neural networks of any depth evolve as linear models under gradient descent.Advances in neural information processing systems, 32, 2019
2019
-
[90]
More than a toy: Random matrix models predict how real-world neural representations generalize
Alexander Wei, Wei Hu, and Jacob Steinhardt. More than a toy: Random matrix models predict how real-world neural representations generalize. InInternational Conference on Machine Learning, pages 23549–23588. PMLR, 2022
2022
-
[91]
Limitations of the empirical fisher approximation for natural gradient descent.Advances in neural information processing systems, 32, 2019
Frederik Kunstner, Philipp Hennig, and Lukas Balles. Limitations of the empirical fisher approximation for natural gradient descent.Advances in neural information processing systems, 32, 2019
2019
-
[92]
Understanding influence functions and datamodels via harmonic analysis
Nikunj Saunshi, Arushi Gupta, Mark Braverman, and Sanjeev Arora. Understanding influence functions and datamodels via harmonic analysis. InThe Eleventh International Conference on Learning Representations, 2022
2022
-
[93]
On second-order group influence functions for black-box predictions
Samyadeep Basu, Xuchen You, and Soheil Feizi. On second-order group influence functions for black-box predictions. InInternational Conference on Machine Learning, pages 715–724. PMLR, 2020
2020
-
[94]
Simple and scalable predictive uncertainty estimation using deep ensembles.Advances in neural information processing systems, 30, 2017
Balaji Lakshminarayanan, Alexander Pritzel, and Charles Blundell. Simple and scalable predictive uncertainty estimation using deep ensembles.Advances in neural information processing systems, 30, 2017
2017
-
[95]
Averaging weights leads to wider optima and better generalization.arXiv preprint arXiv:1803.05407, 2018
Pavel Izmailov, Dmitrii Podoprikhin, Timur Garipov, Dmitry Vetrov, and Andrew Gordon Wilson. Averaging weights leads to wider optima and better generalization.arXiv preprint arXiv:1803.05407, 2018
2018 arXiv
-
[96]
A simple baseline for bayesian uncertainty in deep learning.Advances in neural information processing systems, 32, 2019
Wesley J Maddox, Pavel Izmailov, Timur Garipov, Dmitry P Vetrov, and Andrew Gordon Wilson. A simple baseline for bayesian uncertainty in deep learning.Advances in neural information processing systems, 32, 2019
2019
-
[97]
Bayesian deep learning and a probabilistic perspective of generalization.Advances in neural information processing systems, 33:4697–4708, 2020
Andrew G Wilson and Pavel Izmailov. Bayesian deep learning and a probabilistic perspective of generalization.Advances in neural information processing systems, 33:4697–4708, 2020
2020
-
[98]
Dropout as a bayesian approximation: Representing model uncertainty in deep learning
Yarin Gal and Zoubin Ghahramani. Dropout as a bayesian approximation: Representing model uncertainty in deep learning. Ininternational conference on machine learning, pages 1050–1059. PMLR, 2016
2016
-
[99]
Simfluence: Modeling the influence of individual training examples by simulating training runs.arXiv preprint arXiv:2303.08114, 2023
Kelvin Guu, Albert Webson, Ellie Pavlick, Lucas Dixon, Ian Tenney, and Tolga Bolukbasi. Simfluence: Modeling the influence of individual training examples by simulating training runs.arXiv preprint arXiv:2303.08114, 2023
2023 arXiv
-
[100]
Statistics and causal inference.Journal of the American statistical Association, 81(396):945–960, 1986
Paul W Holland. Statistics and causal inference.Journal of the American statistical Association, 81(396):945–960, 1986
1986
-
[101]
Explaining black box predictions and unveiling data artifacts through influence functions
Xiaochuang Han, Byron C Wallace, and Yulia Tsvetkov. Explaining black box predictions and unveiling data artifacts through influence functions. InProceedings of the 58th Annual Meeting of the Association for Computational Linguistics, pages 5553–5563, 2020
2020
-
[102]
How many and which training points would need to be removed to flip this prediction?arXiv preprint arXiv:2302.02169, 2023
Jinghan Yang, Sarthak Jain, and Byron C Wallace. How many and which training points would need to be removed to flip this prediction?arXiv preprint arXiv:2302.02169, 2023. 15 A Notation & Acronyms A.1 Notation Notation Description BBatch size in mini-batch gradient descent BBa...
2023 arXiv
-
[103]
conduct this evaluation by removing the top and bottom 10% of training samples ranked by influence functions and compare against the removal of the least influential (i.e., smallest influence scores by magnitude) and random samples to see if the resulting models change their p...
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.