Unofficial Python3 Edition
This repository is for "Deep Joint-Semantics Reconstructing Hashing for Large-Scale Unsupervised Cross-Modal Retrieval"
(ICCV 2019, Oral)
By Shupeng Su*, Zhisheng Zhong*, Chao Zhang (* Authors contributed equally).
Cross-modal hashing encodes the multimedia data into a common binary hash space in which the correlations among the samples from different modalities can be effectively measured. Deep cross-modal hashing further improves the retrieval performance as the deep neural networks can generate more semantic relevant features and hash codes. In this paper, we study the unsupervised deep cross-modal hash coding and propose Deep JointSemantics Reconstructing Hashing (DJSRH), which has the following two main advantages. First, to learn binary codes that preserve the neighborhood structure of the original data, DJSRH constructs a novel joint-semantics affinity matrix which elaborately integrates the original neighborhood information from different modalities and accordingly is capable to capture the latent intrinsic semantic affinity for the input multi-modal instances. Second, DJSRH later trains the networks to generate binary codes that maximally reconstruct above joint-semantics relations via the proposed reconstructing framework, which is more competent for the batch-wise training as it reconstructs the specific similarity value unlike the common Laplacian constraint merely preserving the similarity order. Extensive experiments demonstrate the significant improvement by DJSRH in various cross-modal retrieval tasks.
see env.yaml
For datasets, we follow Deep Cross-Modal Hashing's Github (Jiang, CVPR 2017). You can download these datasets from:
- Wikipedia articles, Link
- MIRFLICKR25K, [OneDrive], [Baidu Pan, password: 8dub]
- NUS-WIDE (top-10 concept), [OneDrive], [Baidu Pan, password: ml4y]
The following experiment results are the average values, if you demand for better results, please run the experiment a few more times (2~5).
- Clone this repo:
git clone https://github.com/idejie/DJSRH3.git
. - Change the 'DATASET_DIR' in
settings.py
to where you place the datasets. - An example to train a model:
python train.py
- Modify the parameter
EVAL = True
insettings.py
for validation. - Ablation studies (optional): if you want to evaluate other components of our DJSRH, please refer to our paper and
settings.py
.
Table 1. The mAP@50 results on NUS-WIDE to evaluate the effectiveness of each component in DJSRH.
Model | Configuration | 64bits (I→T) | 64bits (T→I) | 128bits (I→T) | 128bits (T→I) |
---|---|---|---|---|---|
DJSRH-1 | S=SI | 0.717 | 0.712 | 0.741 | 0.735 |
DJSRH-2 | S=ST | 0.702 | 0.606 | 0.734 | 0.581 |
DJSRH-3 | βSI+(1−β)ST | 0.724 | 0.720 | 0.747 | 0.738 |
DJSRH-4 | +(η=0.4) | 0.790 | 0.745 | 0.803 | 0.757 |
DJSRH-5 | +(μ=1.5) | 0.793 | 0.747 | 0.812 | 0.768 |
DJSRH | +(λ1=λ2=0.1) | 0.798 | 0.771 | 0.817 | 0.789 |
DJSRH-6 | −(α=1) | 0.786 | 0.770 | 0.811 | 0.782 |
From the table we can observe that each of our proposed components plays a certain role for our final results.
Table 2. The mAP@50 results on image query text (I→T) and text query image (T→I) retrieval tasks at various encoding lengths and datasets. The best performances are shown as Red while the suboptimal as Blue.
Figure 1. The precision@top-R curves on different datasets at 128 encoding length.
If you find this code useful, please cite our paper:
@inproceedings{su2019deep,
title={Deep Joint-Semantics Reconstructing Hashing for Large-Scale Unsupervised Cross-Modal Retrieval},
author={Shupeng Su, Zhisheng Zhong, Chao Zhang},
booktitle={International Conference on Computer Vision},
year={2019}
}
All rights are reserved by the authors.