arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2109.05205cs.CVcs.AIcs.IR

Contrastive Quantization with Code Memory for Unsupervised Image Retrieval

Tsinghua Shenzhen International Graduate School, Tsinghua University(清华大学清华-深圳国际研究生院);Harbin Institute of Technology, Shenzhen(哈尔滨工业大学(深圳));Shenzhen University(深圳大学);Peng Cheng Laboratory(鹏城实验室)

Jinpeng Wang, Ziyun Zeng, Bin Chen, Tao Dai, Shu-Tao Xia

英文摘要:

The high efficiency in computation and storage makes hashing (including binary hashing and quantization) a common strategy in large-scale retrieval systems. To alleviate the reliance on expensive annotations, unsupervised deep hashing becomes an important research problem. This paper provides a novel solution to unsupervised deep quantization, namely Contrastive Quantization with Code Memory (MeCoQ). Different from existing reconstruction-based strategies, we learn unsupervised binary descriptors by contrastive learning, which can better capture discriminative visual semantics. Besides, we uncover that codeword diversity regularization is critical to prevent contrastive learning-based quantization from model degeneration. Moreover, we introduce a novel quantization code memory module that boosts contrastive learning with lower feature drift than conventional feature memories. Extensive experiments on benchmark datasets show that MeCoQ outperforms state-of-the-art methods. Code and configurations are publicly available at https://github.com/gimpong/AAAI22-MeCoQ.

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

↑