SEA-NN
Improving Attribution Methods by Learning Submodular Functions
Accepted at the 25th International Conference on Artificial Intelligence and Statistics (AISTATS ‘22).
Piyushi Manupriya, Tarun Ram Menta, J. Saketha Nath, Vineeth N Balasubramanian
| Code in PyTorch | Paper | Poster | Talk | Link to the version accepted at ICML ‘20 workshops (XXAI, WHI) |
This work explores the novel idea of learning a submodular scoring function to improve the specificity of existing feature attribution methods. A novel formulation for learning a deep submodular set function that is consistent with the real-valued attribution maps obtained by existing attribution methods is proposed. The final attribution value of a feature is then defined as the marginal gain in the induced submodular score of the feature in the context of other highly attributed features, thus decreasing the attribution of redundant yet discriminatory features. Experiments on multiple datasets illustrate that the proposed attribution method achieves higher specificity along with good discriminative power.
Acknowledgements
The first author is especially grateful to the mentorship of Dr Bilal Alsallakh and for being supported by the Google PhD Fellowship.