Model Stitching and Visualization How GAN Generators can Invert Networks in Real-Time

Rudolf Herdt,Maximilian Schmidt,Daniel Otero Baguer,Jean Le'Clerc Arrastia,Peter Maass
2024-02-20
Abstract:In this work, we propose a fast and accurate method to reconstruct activations of classification and semantic segmentation networks by stitching them with a GAN generator utilizing a 1x1 convolution. We test our approach on images of animals from the AFHQ wild dataset, ImageNet1K, and real-world digital pathology scans of stained tissue samples. Our results show comparable performance to established gradient descent methods but with a processing time that is two orders of magnitude faster, making this approach promising for practical applications.
Computer Vision and Pattern Recognition,Artificial Intelligence,Machine Learning,Image and Video Processing
What problem does this paper attempt to address?