Solve the prototype assignment problem of UNI feature extraction

[ShanLing-cool]

Hello, I used UNI in CLAM to extract features. When reproducing the BRCA dataset, almost all patches were assigned to a single prototype, resulting in a certain prototype accounting for up to 95% in the visualization. I saw that you mentioned this shortcoming at the end of the paper. How do you solve this problem?