Paper Reading AI Learner

Feature Space Analysis by Guided Diffusion Model

2025-09-09 17:18:39
Kimiaki Shirahama, Miki Yanobu, Kaduki Yamashita, Miho Ohsaki

Abstract

One of the key issues in Deep Neural Networks (DNNs) is the black-box nature of their internal feature extraction process. Targeting vision-related domains, this paper focuses on analysing the feature space of a DNN by proposing a decoder that can generate images whose features are guaranteed to closely match a user-specified feature. Owing to this guarantee that is missed in past studies, our decoder allows us to evidence which of various attributes in an image are encoded into a feature by the DNN, by generating images whose features are in proximity to that feature. Our decoder is implemented as a guided diffusion model that guides the reverse image generation of a pre-trained diffusion model to minimise the Euclidean distance between the feature of a clean image estimated at each step and the user-specified feature. One practical advantage of our decoder is that it can analyse feature spaces of different DNNs with no additional training and run on a single COTS GPU. The experimental results targeting CLIP's image encoder, ResNet-50 and vision transformer demonstrate that images generated by our decoder have features remarkably similar to the user-specified ones and reveal valuable insights into these DNNs' feature spaces.

Abstract (translated)

深度神经网络(DNN)的一个关键问题是其内部特征提取过程的黑盒性质。针对视觉相关领域,本文通过提出一种解码器来分析DNN的特征空间,该解码器可以生成具有与用户指定特性紧密匹配的图像特征的图像。由于以往的研究中忽略了这一保证,我们的解码器能够生成接近特定特性的图像,从而揭示各种属性在图像中的编码情况。我们的解码器实现为一种引导扩散模型,它指导预训练的扩散模型进行逆向图像生成过程,以最小化每一步估计的清洁图像特征与用户指定特征之间的欧氏距离。 我们解码器的一个实际优势是它可以无需额外训练地分析不同DNN的特征空间,并且可以在单个商用现成(COTS)GPU上运行。针对CLIP的图像编码器、ResNet-50和视觉变压器进行的实验结果表明,由我们的解码器生成的图像具有与用户指定特性非常相似的特征,并揭示了这些DNN特征空间中的宝贵见解。

URL

https://arxiv.org/abs/2509.07936

PDF

https://arxiv.org/pdf/2509.07936.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot