Section 6 of 6
Conclusions
Zheng Zhang and Liang Zhao · about 1 minutes
In this study, we introduce a novel framework called Transferable Deep Clustering Model (TDCM) to tackle the challenge of limited generalization ability in previous end-to-end deep clustering techniques when faced with unseen domain data. instead of optimizing a fixed set of centroids specific to the training source domain, our proposed TDCM employs an adapted centroids updating module, enabling automatic adaptation of centroids based on the input domain data. As a result, our framework exhibits enhanced generalization capabilities to handle unseen domain data. To capture the intrinsic structure and patterns of clusters, we propose an attention-based learnable module, which learns a data-driven score function for measuring the underlying similarity among samples. Theoretical analysis guarantees the effectiveness of our proposed module in extracting underlying similarity relationships, surpassing conventional clustering techniques such as k-means or Gaussian Mixture Models (GMM) in terms of expressiveness. Extensive experiments conducted on synthetic and real-world datasets validate the effectiveness of our proposed model in addressing distributional drift during the transfer of clustering knowledge from trained source domains to unseen target domains.