Human-in-the-loop Extraction of Interpretable Concepts in Deep Learning Models
arXiv:2108.03738
Abstract
The interpretation of deep neural networks (DNNs) has become a key topic as more and more people apply them to solve various problems and making critical decisions. Concept-based explanations have recently become a popular approach for post-hoc interpretation of DNNs. However, identifying human-understandable visual concepts that affect model decisions is a challenging task that is not easily addressed with automatic approaches. We present a novel human-in-the-loop approach to generate user-defined concepts for model interpretation and diagnostics. Central to our proposal is the use of active learning, where human knowledge and feedback are combined to train a concept extractor with very little human labeling effort. We integrate this process into an interactive system, ConceptExtract. Through two case studies, we show how our approach helps analyze model behavior and extract human-friendly concepts for different machine learning tasks and datasets and how to use these concepts to understand the predictions, compare model performance and make suggestions for model refinement. Quantitative experiments show that our active learning approach can accurately extract meaningful visual concepts. More importantly, by identifying visual concepts that negatively affect model performance, we develop the corresponding data augmentation strategy that consistently improves model performance.
11 pages, 10 figures
References in corpus (10)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- A Unified Approach to Interpreting Model Predictions
- Rethinking Atrous Convolution for Semantic Image Segmentation
- Learning a Probabilistic Latent Space of Object Shapes via 3D Generative-Adversarial Modeling
- Do Deep Nets Really Need to be Deep?
- Cost-Effective Active Learning for Deep Image Classification
- Sanity Checks for Saliency Maps
- Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)
- Latent Space Purification via Neural Density Operators