arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2111.04086cs.LGcs.AIcs.DBcs.MA
收藏

基于长尾数据的Meta跨模态哈希算法

Meta Cross-Modal Hashing on Long-Tailed Data

Southwest University(西南大学);Shandong University(山东大学);George Mason University(乔治梅森大学);King Abdullah University of Science and Technology(阿卜杜拉国王科技大学)

Runmin Wang, Guoxian Yu, Carlotta Domeniconi, Xiangliang Zhang

摘要:

由于跨模态哈希算法在减少存储空间的同时加快了异构数据的查询速度,因此在多模态数据的近似最近邻搜索中得到了广泛的研究。大多数散列方法假设训练数据是类平衡的,然而,在实践中,真实世界的数据往往具有长尾分布。在本文中,我们介绍了一种基于元学习的跨模态哈希方法(MetaCMH)来处理长尾数据。由于尾类中训练样本的缺乏,MetaCMH首先从不同模态的数据中学习直接特征,然后引入联想记忆模块来学习尾类样本的记忆特征。然后,它结合直接和记忆功能,以获得每个样本的Meta功能。对于长尾分布的头类样本,直接特征的权重较大,因为有足够的训练数据来学习它们;而对于稀有类,记忆特征的权重较大。最后,MetaCMH使用似然损失函数来保持不同模态的相似性,并以端到端的方式学习哈希函数。在长尾数据集上的实验表明,MetaCMH的性能明显优于现有的方法,特别是在尾部类上。

英文摘要:

Due to the advantage of reducing storage while speeding up query time on big heterogeneous data, cross-modal hashing has been extensively studied for approximate nearest neighbor search of multi-modal data. Most hashing methods assume that training data is class-balanced.However, in practice, real world data often have a long-tailed distribution. In this paper, we introduce a meta-learning based cross-modal hashing method (MetaCMH) to handle long-tailed data. Due to the lack of training samples in the tail classes, MetaCMH first learns direct features from data in different modalities, and then introduces an associative memory module to learn the memory features of samples of the tail classes. It then combines the direct and memory features to obtain meta features for each sample. For samples of the head classes of the long tail distribution, the weight of the direct features is larger, because there are enough training data to learn them well; while for rare classes, the weight of the memory features is larger. Finally, MetaCMH uses a likelihood loss function to preserve the similarity in different modalities and learns hash functions in an end-to-end fashion. Experiments on long-tailed datasets show that MetaCMH performs significantly better than state-of-the-art methods, especially on the tail classes.

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

↑