Channel Similarity

Hi there, I read your paper and was intrigued by Figure 6b. I was wondering, how did you visualise this and is there any code to do this? Thanks

Please refer to some publicly available repos for CKA calculation like this: https://github.com/yuanli2333/CKA-Centered-Kernel-Alignment

Thanks for your response, I have taken a look at the repository, however, am confused in terms of implementing this for share_feat to get the plot. One way is to perhaps do share_feat[i] and conduct the process against share_feat[j]. However, this seems to be fairly long and I'm not sure if the desired output can be obtained this way

Any help on this would be appreciated