Transferable Deep Clustering Model
Zheng Zhang Liang Zhao
Deep learning has shown remarkable success in the field of clustering
recently. However, how to transfer a trained clustering model on a source
domain to a target domain by leveraging the acquired knowledge to guide the
clustering process remains challenging. Existing deep clustering methods often
lack generalizability to new domains because they typically learn a group of
fixed cluster centroids, which may not be optimal for the new domain
distributions. In this paper, we propose a novel transferable deep clustering
model that can automatically adapt the cluster centroids according to the
distribution of data samples. Rather than learning a fixed set of centroids,
our approach introduces a novel attention-based module that can adapt the
centroids by measuring their relationship with samples. In addition, we
theoretically show that our model is strictly more powerful than some classical
clustering algorithms such as k-means or Gaussian Mixture Model (GMM).
Experimental results on both synthetic and real-world datasets demonstrate the
effectiveness and efficiency of our proposed transfer learning framework, which
significantly improves the performance on target domain and reduces the
computational cost.