Explainable embeddings with Distance Explainer
Summary
The paper introduces Distance Explainer, a novel local, post-hoc eXplainable AI (XAI) method for interpreting embedded vector spaces. Adapting saliency-based techniques from RISE, it explains the distance between two embedded data points by assigning attribution values through selective masking and distance-ranked mask filtering. Evaluated on cross-modal embeddings (image-image and image-caption pairs) using ImageNet (1000-dimensional classification vectors) and CLIP (512-dimensional semantic vectors) models, the method effectively identifies features contributing to similarity or dissimilarity. Experiments with ResNet50 and VGG16 models demonstrate high robustness and consistency, assessed via Faithfulness, Sensitivity/Robustness, and Randomization metrics. Parameter tuning, including mask quantity and selection strategy, significantly affects explanation quality.
Key takeaway
For Machine Learning Engineers developing or deploying models with embedded spaces, understanding feature contributions to embedding distances is critical for trustworthiness. You should consider integrating Distance Explainer to generate local, post-hoc explanations for similarity or dissimilarity between data points. Experiment with mask quantity and selection strategies to optimize explanation quality for your specific cross-modal or single-modal applications, enhancing model transparency.
Key insights
Distance Explainer attributes feature contributions to similarity or dissimilarity between embedded data points.
Principles
- XAI for embedded spaces requires local, post-hoc explanations.
- Distance-ranked mask filtering improves saliency map generation.
- Parameter tuning is crucial for explanation quality.
Method
Distance Explainer masks an input item, calculates its cosine distance to a reference item's embedding, ranks masks by distance, filters them (top/bottom X%), and sums remaining masks to create an attribution map.
In practice
- Explain image-image similarity using ImageNet models.
- Interpret image-caption semantic relationships with CLIP.
- Tune mask quantity and selection for specific use cases.
Topics
- Explainable AI
- Embedded Spaces
- Saliency Maps
- Multi-modal Embeddings
- CLIP Model
- Attribution Methods
Code references
Best for: AI Scientist, Machine Learning Engineer, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.