Explainable embeddings with Distance Explainer

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, extended

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

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

Topics

Code references

Best for: AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.