无需参数调优,用可靠局部结构提升聚类模型全局表现
You Can Trust Your Clustering Model: A Parameter-free Self-Boosting Plug-in for Deep Clustering
- 通过自适应k近邻筛选高置信样本,构建可信监督信号
- 在多个数据集上使现有聚类模型性能提升超3%,轮廓系数提升7倍以上
- 可即插即用,适合希望增强现有聚类模型的开发者
近期深度聚类模型取得了优异性能,但普遍存在全局与局部特征结构不一致的问题:局部特征在类内具强一致性与紧凑性,而全局特征常呈现边界交错、聚类分离度差。为此,本文提出DCBoost,一种无需参数调优的即插即用模块,旨在优化现有深度聚类模型的全局特征结构。首先,基于自适应k近邻的一致性过滤识别高置信样本,作为自监督的可信锚点;随后,利用这些样本计算判别性损失,促进类内紧凑与类间分离,指导网络优化。在多个基准数据集上的实验表明,该方法显著提升多种现有深度聚类模型的性能,尤其对当前最优基线(如ProPos)提升超过3%,轮廓系数增幅超过7倍。代码已开源。
原文摘要 · Abstract (English)
Recent deep clustering models have produced impressive clustering performance. However, a common issue with existing methods is the disparity between global and local feature structures. While local structures typically show strong consistency and compactness within class samples, global features often present intertwined boundaries and poorly separated clusters. Motivated by this observation, we propose DCBoost, a parameter-free plug-in designed to enhance the global feature structures of current deep clustering models. By harnessing reliable local structural cues, our method aims to elevate clustering performance effectively. Specifically, we first identify high-confidence samples through adaptive $k$-nearest neighbors-based consistency filtering, aiming to select a sufficient number of samples with high label reliability to serve as trustworthy anchors for self-supervision. Subsequently, these samples are utilized to compute a discriminative loss, which promotes both intra-class compactness and inter-class separability, to guide network optimization. Extensive experiments across various benchmark datasets showcase that our DCBoost significantly improves the clustering performance of diverse existing deep clustering models. Notably, our method improves the performance of current state-of-the-art baselines (e.g., ProPos) by more than 3% and amplifies the silhouette coefficient by over $7\times$. Code is available at <https://github.com/l-h-y168/DCBoost>.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。