论文

HIDE方法解释黑盒模型

Hide and Seek: HIDE Concepts to SEEK explanations for black box models

精选理由

HnS方法能解释黑盒视觉模型,不需要模型权重或标记数据,还能做图像级归因分析。

研究人员提出Hide and Seek (HnS)新方法,用于黑盒视觉模型的概念级归因分析。该方法无需访问模型权重或标记概念数据集,通过视觉语言模型检索概念对齐图像,定位并遮罩区域,测量模型输出变化。在CelebA和COCO-Stuff基准上,HnS与真实归因的一致性优于现有方法。该方法还适用于图像标注和卫星图像分割等现有概念归因方法无法处理的任务。

原文 · arXiv cs.LG

Hide and Seek: HIDE Concepts to SEEK explanations for black box models

Interpreting machine learning model predictions is a central challenge, particularly for black-box vision models. Concept-based attribution methods interpret models using human-understandable concepts. Existing concept-based methods can only interpret classification and regression models. They require access to model weights or large concept-annotated datasets, and most methods cannot provide image-level attribution. We propose Hide and Seek (HnS), a new method for concept-level attribution that does not require access to model weights or labeled concept datasets. HnS can interpret a wide range of vision models. Given a concept and image set, HnS retrieves concept-aligned images using a vision-language model, localizes the concept within each image, masks the localized region, and measures attribution from the change in the black-box model's output between the original and masked images. This enables faithful attribution by directly measuring the model's behavior and supporting local attributions. We validated HnS on CelebA and COCO-Stuff. Compared to existing methods, HnS showed better agreement with ground-truth attributions. We demonstrated the applicability of HnS on diverse tasks, including image captioning and satellite image segmentation, where no existing concept attribution method can be used. We provided our anonymous code in the supplementary material.