arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.08466cs.LGcs.AIcs.DSstat.ML

Objective-Based Hierarchical Clustering of Deep Embedding Vectors

ITMO University(圣彼得堡国立信息技术机械与光学大学);Indiana University, Bloomington(印第安纳大学伯明顿分校)

Stanislav Naumov, Grigory Yaroslavtsev, Dmitrii Avdiukhin

英文摘要:

We initiate a comprehensive experimental study of objective-based hierarchical clustering methods on massive datasets consisting of deep embedding vectors from computer vision and NLP applications. This includes a large variety of image embedding (ImageNet, ImageNetV2, NaBirds), word embedding (Twitter, Wikipedia), and sentence embedding (SST-2) vectors from several popular recent models (e.g. ResNet, ResNext, Inception V3, SBERT). Our study includes datasets with up to $4.5$ million entries with embedding dimensions up to $2048$. In order to address the challenge of scaling up hierarchical clustering to such large datasets we propose a new practical hierarchical clustering algorithm B++&C. It gives a 5%/20% improvement on average for the popular Moseley-Wang (MW) / Cohen-Addad et al. (CKMM) objectives (normalized) compared to a wide range of classic methods and recent heuristics. We also introduce a theoretical algorithm B2SAT&C which achieves a $0.74$-approximation for the CKMM objective in polynomial time. This is the first substantial improvement over the trivial $2/3$-approximation achieved by a random binary tree. Prior to this work, the best poly-time approximation of $\approx 2/3 + 0.0004$ was due to Charikar et al. (SODA'19).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑