English

Constructing sensible baselines for Integrated Gradients

Machine Learning 2024-12-19 v1 High Energy Physics - Experiment

Abstract

Machine learning methods have seen a meteoric rise in their applications in the scientific community. However, little effort has been put into understanding these "black box" models. We show how one can apply integrated gradients (IGs) to understand these models by designing different baselines, by taking an example case study in particle physics. We find that the zero-vector baseline does not provide good feature attributions and that an averaged baseline sampled from the background events provides consistently more reasonable attributions.

Keywords

Cite

@article{arxiv.2412.13864,
  title  = {Constructing sensible baselines for Integrated Gradients},
  author = {Jai Bardhan and Cyrin Neeraj and Mihir Rawat and Subhadip Mitra},
  journal= {arXiv preprint arXiv:2412.13864},
  year   = {2024}
}

Comments

7 pages, 5 figures. Accepted to 4th Annual AAAI Workshop on AI to Accelerate Science and Engineering (AI2ASE)

R2 v1 2026-06-28T20:40:30.041Z