Learning Credible Deep Neural Networks with Rationale Regularization

Abstract

Recent explainability related studies have shown that state-of-the-art DNNs do not always adopt correct evidences to make decisions. It not only hampers their generalization but also makes them less likely to be trusted by end-users. In pursuit of developing more credible DNNs, in this paper we propose CREX, which encourages DNN models to focus more on evidences that actually matter for the task at hand, and to avoid overfitting to data-dependent bias and artifacts. Specifically, CREX regularizes the training process of DNNs with rationales, i.e., a subset of features highlighted by domain experts as justifications for predictions, to enforce DNNs to generate local explanations that conform with expert rationales. Even when rationales are not available, CREX still could be useful by requiring the generated explanations to be sparse. Experimental results on two text classification datasets demonstrate the increased credibility of DNNs trained with CREX. Comprehensive analysis further shows that while CREX does not always improve prediction accuracy on the held-out test set, it significantly increases DNN accuracy on new and previously unseen data beyond test set, highlighting the advantage of the increased credibility.

ICDM 2019

References in corpus (11)

• Adam: A Method for Stochastic Optimization
• Distributed Representations of Words and Phrases and their Compositionality
• Distilling the Knowledge in a Neural Network
• Towards A Rigorous Science of Interpretable Machine Learning
• Methods for Interpreting and Understanding Deep Neural Networks
• A Structured Self-attentive Sentence Embedding
• Man is to Computer Programmer as Woman is to Homemaker? Debiasing Word Embeddings
• Understanding Neural Networks through Representation Erasure
• Investigating Backtranslation in Neural Machine Translation
• Women also Snowboard: Overcoming Bias in Captioning Models
• Evaluating Explanation Without Ground Truth in Interpretable Machine Learning

Cited by in corpus (3)

• Local Interpretations for Explainable Natural Language Processing: A Survey
• Interpretations are useful: penalizing explanations to align neural networks with prior knowledge
• iCapsNets: Towards Interpretable Capsule Networks for Text Classification