- [arXiv]
- [CVPRW2023]
- accepted at CVPR2023 workshop on CVFAD
Set-to-set matching is the problem of matching two different sets of items based on some criteria. Especially when each item in the set is high-dimensional, such as an image, set-to-set matching is treated as one of the applied problems to be solved by utilizing neural networks. Most machine learning-based set-to-set matching generally assumes that the training and test data follow the same distribution. However, such assumptions are often violated in real-world machine learning problems. In this paper, we propose SHIFT15M, a dataset that can be used to properly evaluate set-to-set matching models in situations where the distribution of data changes between training and testing. Some benchmark experiments show that the performance of naive methods drops due to the effects of the distribution shift. In addition, we provide software to handle the SHIFT15M dataset in a very simple way. The URL for the software will appear after this manuscript is published.
We provide the Datasheet for SHIFT15M. This datasheet is based on the Datasheets for Datasets [1] template.
System | Python 3.6 | Python 3.7 | Python 3.8 |
---|---|---|---|
Linux CPU | |||
Linux GPU | |||
Windows CPU / GPU | |||
Mac OS CPU |
SHIFT15M is a large-scale dataset based on approximately 15 million items accumulated by the fashion search service IQON.
$ pip install shift15m
$ git clone https://github.com/st-tech/zozo-shift15m.git
$ cd zozo-shift15m
$ poetry build
$ pip install dist/shift15m-xxxx-py3-none-any.whl
You can download SHIFT15M dataset as follows:
from shift15m.datasets import NumLikesRegression
dataset = NumLikesRegression(root="./data", download=True)
(x_train, y_train), (x_test, y_test) = dataset.load_dataset(target_shift=True)
Please download the dataset as follows:
$ bash scripts/download_all.sh
The following tasks are now available:
Tasks | Task type | Shift type | # of input dim | # of output dim |
---|---|---|---|---|
NumLikesRegression | regression | target shift | (N, 25) | (N, 1) |
SumPricesRegression | regression | covariate shift, target shift | (N, 1) | (N, 1) |
ItemPriceRegression | regression | target shift | (N, 4096) | (N, 1) |
ItemCategoryClassification | classification | target shift | (N, 4096) | (N, 7) |
Set2SetMatching | set-to-set matching | covariate shift | (N, 4096)x(M, 4096) | (1) |
As templates for numerical experiments on the SHIFT15M dataset, we have published experimental results for each task with several models.
The original dataset is maintained in json format, and a row consists of the following:
{
"user":{"user_id":"xxxx", "fav_brand_ids":"xxxx,xx,..."},
"like_num":"xx",
"set_id":"xxx",
"items":[
{"price":"xxxx","item_id":"xxxxxx","category_id1":"xx","category_id2":"xxxxx"},
...
],
"publish_date":"yyyy-mm-dd",
"tags": "tag_a, tag_b, tag_c, ..."
}
To learn more about making a contribution to SHIFT15M, please see the following materials:
The dataset itself is provided under a CC BY-NC 4.0 license. On the other hand, the software in this repository is provided under the MIT license.
The following table is necessary for this dataset to be indexed by search engines such as Google Dataset Search.
property | value | ||||||
---|---|---|---|---|---|---|---|
name | SHIFT15M Dataset |
||||||
alternateName | SHIFT15M |
||||||
alternateName | shift15m-dataset |
||||||
url | https://github.com/st-tech/zozo-shift15m |
||||||
sameAs | https://github.com/st-tech/zozo-shift15m |
||||||
description | SHIFT15M is a multi-objective, multi-domain dataset which includes multiple dataset shifts. |
||||||
provider |
|
||||||
license |
|
- 01/08/2022, added tags info (#187)
- Papadopoulos, Stefanos I., et al. "Multimodal Quasi-AutoRegression: Forecasting the visual popularity of new fashion products." arXiv preprint arXiv:2204.04014 (2022).
- Papadopoulos, Stefanos, et al. Fashion Trend Analysis and Prediction Model. 1, Zenodo, 2021, doi:10.5281/zenodo.5795089.
- [1] Gebru, Timnit, et al. "Datasheets for datasets." arXiv preprint arXiv:1803.09010 (2018).