Channel Similarity

[repers]

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

[zzh-tech]

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

[repers]

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