Using Stratified Sampling to Improve LIME Image Explanations

Abstract

We investigate the use of a stratified sampling approach for LIME Image, a popular model-agnostic explainable AI method for computer vision tasks, in order to reduce the artifacts generated by typical Monte Carlo sampling. Such artifacts are due to the undersampling of the dependent variable in the synthetic neighborhood around the image being explained, which may result in inadequate explanations due to the impossibility of fitting a linear regressor on the sampled data. We then highlight a connection with the Shapley theory, where similar arguments about undersampling and sample relevance were suggested in the past. We derive all the formulas and adjustment factors required for an unbiased stratified sampling estimator. Experiments show the efficacy of the proposed approach.

Cite

Text

Rashid et al. "Using Stratified Sampling to Improve LIME Image Explanations." AAAI Conference on Artificial Intelligence, 2024. doi:10.1609/AAAI.V38I13.29397

Markdown

[Rashid et al. "Using Stratified Sampling to Improve LIME Image Explanations." AAAI Conference on Artificial Intelligence, 2024.](https://mlanthology.org/aaai/2024/rashid2024aaai-using/) doi:10.1609/AAAI.V38I13.29397

BibTeX

@inproceedings{rashid2024aaai-using,
  title     = {{Using Stratified Sampling to Improve LIME Image Explanations}},
  author    = {Rashid, Muhammad and Amparore, Elvio G. and Ferrari, Enrico and Verda, Damiano},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {14785-14792},
  doi       = {10.1609/AAAI.V38I13.29397},
  url       = {https://mlanthology.org/aaai/2024/rashid2024aaai-using/}
}