- MCS dataset was removed from the benchmark
- For not complying with the benchmark's criteria (see the paper for more information). It can still be run by using the
--d MCS
flag.
- For not complying with the benchmark's criteria (see the paper for more information). It can still be run by using the
- People dataset added to the benchmark!
- This dataset dates back to 2009 and belongs to the paper Ehinger, K. A., Hidalgo-Sotelo, B., Torralba, A., & Oliva, A, "Modeling Search for People in 900 Scenes: A combined source model of eye guidance", Visual cognition, 17(6-7), 945–978, 2009.
- eccNET added to the benchmark!
- This visual search model belongs to the paper Gupta SK., Zhang M., Wu C., Wolfe JM., Kreiman G., "Visual Search Asymmetry: Deep Nets and Humans Share Similar Inherent Biases", NeurIPS 2021.
Visual search is an essential part of almost any everyday human interaction with the visual environment. Nowadays, several algorithms are able to predict gaze positions during simple observation, but few models attempt to simulate human behavior during visual search in natural scenes. Furthermore, these models vary widely in their design and exhibit differences in the datasets and metrics with which they were evaluated.
To attend to this problem, we have selected publicly available state-of-the-art visual search models and datasets in natural scenes, and provide a common framework for their evaluation. We apply a unified format and criteria, bridging the gaps between them, and we estimate the models’ efficiency and similarity with humans using a specific set of metrics.
Current models scores, relative to human subjects, averaged across all datasets (*):
AUCperf | AvgMM | Corr | AUChsp | NSShsp | IGhsp | LLhsp | Score | |
---|---|---|---|---|---|---|---|---|
Humans | 0.62 | 0.87 | - | - | - | - | - | - |
Gold Standard | - | - | - | 0.91 | 2.92 | 2.75 | 2.25 | 0.00 |
nnIBS | 0.51 | 0.84 | 0.23 | 0.80 | 1.44 | 1.22 | 0.50 | -0.12 |
eccNET | 0.46 | 0.83 | 0.17 | 0.70 | 1.01 | -0.87 | -1.31 | -0.40 |
IVSN | 0.57 | 0.80 | 0.12 | 0.62 | 0.85 | -2.05 | -2.53 | -0.56 |
IRL | 0.30 | 0.60 | 0.06 | 0.49 | 0.90 | -3.05 | -3.37 | -0.76 |
Uniform | - | - | - | 0.50 | 0.00 | 0.51 | 0.00 | -0.82 |
Center bias | - | - | - | 0.66 | 0.54 | 0.00 | -0.51 | -0.83 |
AUCperf
measures efficiency, while AvgMM
is the Multi-Match average between models and subjects. Corr
is the correlation between within-humans Multi-Match (whMM) and human-model Multi-Match (hmMM). Lastly, AUChsp
, NSShsp
, IGhsp
and LLhsp
focus on human scanpath prediction (HSP). The precise definition of each can be found at Metrics
.
(*) IRL was not run in People dataset as it is not able to search for people.
Scores for individual datasets:
Python 3.8 or newer is required. To install, run:
git clone [email protected]:FerminT/VisualSearchBenchmark.git
pip3 install -r ./VisualSearchBenchmark/requirements.txt
Detectron2
needs to be installed separately (see detectron2/installation). It is only needed if you're going to run the IRL model on new images.
Runs all models in every dataset and compute all metrics (by default, precomputed results are loaded):
python3 run_benchmark.py
The code was tested in Ubuntu 16.04 and later.
--d dataset1_name dataset2_name ..
: runs the visual search models in the datasets specified. Each dataset name corresponds to its folder's name inDatasets
.--m model1_name model2_name ..
: runs the specified visual search models. Each model name corresponds to its folder's name inModels
.--mts [mm] [perf] [hsp]
: computes the specified metrics. Values can bemm
(Multi-Match),perf
(Cumulative Performance), and/orhsp
(Human Scanpath Prediction). SeeMetrics
for more information. Leave blank to not compute any metric.--noplot
: Do not plot the results. Useful for running the benchmark and going AFK.--f
: Forces execution. Precomputed results are deleted and the specified models and metrics are run from scratch. WARNING: it will take a long time!
Runs nnIBS
on the COCOSearch18
and Interiors
datasets, and then computes Multi-Match and Cumulative Performance:
python3 run_benchmark.py --d COCOSearch18 Interiors --m nnIBS --mts mm perf
If you use our work, please cite us:
@inproceedings{travi2022visions,
author = {Travi, Ferm\'{\i}n and Ruarte, Gonzalo and Bujia, Gaston and Kamienkowski, Juan Esteban},
booktitle = {Advances in Neural Information Processing Systems},
editor = {S. Koyejo and S. Mohamed and A. Agarwal and D. Belgrave and K. Cho and A. Oh},
pages = {11987--12000},
publisher = {Curran Associates, Inc.},
title = {ViSioNS: Visual Search in Natural Scenes Benchmark},
url = {https://proceedings.neurips.cc/paper_files/paper/2022/file/4ea14e6090343523ddcd5d3ca449695f-Paper-Datasets_and_Benchmarks.pdf},
volume = {35},
year = {2022}
}
@article{Bujia2022vsearch,
author = {Bujia, Gaston and Sclar, Melanie and Vita, Sebastian and Solovey, Guillermo and Kamienkowski, Juan Esteban},
doi = {10.3389/fnsys.2022.882315},
issn = {1662-5137},
journal = {Frontiers in Systems Neuroscience},
title = {Modeling Human Visual Search in Natural Scenes: A Combined Bayesian Searcher and Saliency Map Approach},
url = {https://www.frontiersin.org/article/10.3389/fnsys.2022.882315},
volume = {16},
year = {2022}
}
The exact materials used for the paper can be found at the 1. NeurIPS 2022 branch, 2. SVRHM 2021 branch, 3. Frontiers in Systems Neuroscience repository.
- eccNET: Gupta SK., Zhang M., Wu C., Wolfe JM., Kreiman G., Visual Search Asymmetry: Deep Nets and Humans Share Similar Inherent Biases, NeurIPS 2021.
- IVSN: Zhang M. et al., Finding any Waldo with Zero-shot Invariant and Efficient Visual Search Model, Nature Comun. 9, 2018.
- nnIBS: Bujia G., Sclar M., Vita S., Solovey G., Kamienkowski JE., Modeling Human Visual Search in Natural Scenes: A Combined Bayesian Searcher and Saliency Map Approach, Frontiers in Systems Neuroscience, Volume 16, 2022.
- IRL: Yang, Z. et. al., Predicting Goal-directed Human Attention Using Inverse Reinforcement Learning, IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2020
- Interiors: Bujia G., Sclar M., Vita S., Solovey G., Kamienkowski JE., Modeling Human Visual Search in Natural Scenes: A Combined Bayesian Searcher and Saliency Map Approach, Frontiers in Systems Neuroscience, Volume 16, 2022.
- Unrestricted: Zhang M. et al., Finding any Waldo with Zero-shot Invariant and Efficient Visual Search Model, Nature Comun. 9, 2018.
- COCOSearch18: Chen, Y., Yang, Z., Ahn, S., Samaras, D., Hoai, M., & Zelinsky, G., COCO-Search18 Fixation Dataset for Predicting Goal-directed Attention Control, Scientific Reports, 11 (1), 1-11, 2021.
- MCS: G.J. Zelinsky et al., Benchmarking Gaze Prediction for Categorical Visual Search, CVPR Workshops 2019, 2019.
- People: Ehinger, K. A., Hidalgo-Sotelo, B., Torralba, A., & Oliva, A, Modeling Search for People in 900 Scenes: A combined source model of eye guidance, Visual cognition, 17(6-7), 945–978, 2009.