EnsembleSHAP: Faithful and Certifiably Robust Attribution for Random Subspace Method
About
Random subspace method has wide security applications such as providing certified defenses against adversarial and backdoor attacks, and building robustly aligned LLM against jailbreaking attacks. However, the explanation of random subspace method lacks sufficient exploration. Existing state-of-the-art feature attribution methods, such as Shapley value and LIME, are computationally impractical and lacks security guarantee when applied to random subspace method. In this work, we propose EnsembleSHAP, an intrinsically faithful and secure feature attribution for random subspace method that reuses its computational byproducts. Specifically, our feature attribution method is 1) computationally efficient, 2) maintains essential properties of effective feature attribution (such as local accuracy), and 3) offers guaranteed protection against privacy-preserving attacks on feature attribution methods. To the best of our knowledge, this is the first work to establish provable robustness against explanation-preserving attacks. We also perform comprehensive evaluations for our explanation's effectiveness when faced with different empirical attacks, including backdoor attacks, adversarial attacks, and jailbreak attacks. The code is at https://github.com/Wang-Yanting/EnsembleSHAP. WARNING: This document may include content that could be considered harmful.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Faithfulness Evaluation | AG News (test) | Rate of Label Changes17.5 | 24 | |
| Faithfulness Evaluation | SST-2 (test) | Rate of Label Changes36.5 | 24 | |
| Faithfulness Evaluation | IMDB (test) | Rate of Label Changes60 | 24 | |
| Faithfulness Evaluation | SST-2 | Rate of Label Changes32 | 12 | |
| Faithfulness Evaluation | AG-News | Rate of Label Changes24 | 12 | |
| Faithfulness Evaluation | IMDB | Rate of Label Changes60 | 12 | |
| Keyword Prediction | SST-2 | Precision58.5 | 8 | |
| Keyword Prediction | IMDB | Precision53.5 | 8 | |
| Keyword Prediction | AG-News | Precision55.7 | 8 | |
| Explanation Faithfulness | Jailbreaking GCG (test) | Rate of Label Changes15 | 8 |