Awesome
ImageNetVC
Codes and datasets for our paper: ImageNetVC: Zero- and Few-Shot Visual Commonsense Evaluation on 1000 ImageNet Categories.
You can also download our dataset through Huggingface Datasets: hemingkx/ImageNetVC.
Overview
Recently, Large Language Models (LLMs) have been serving as general-purpose interfaces, posing a significant demand for comprehensive visual knowledge. However, it remains unclear how well current LLMs and their visually augmented counterparts (VaLMs) can master visual commonsense knowledge. To investigate this, we propose ImageNetVC, a human-annotated dataset specifically designed for zero- and few-shot visual commonsense evaluation across 1,000 ImageNet categories. Utilizing ImageNetVC, we benchmark the fundamental visual commonsense knowledge of both unimodal LLMs and VaLMs. Furthermore, we analyze the factors affecting the visual commonsense knowledge of large-scale models, providing insights into the development of language models enriched with visual commonsense knowledge.
Takeaways
The main evaluation results of LLMs and VaLMs on ImageNetVC are shown in the following. Here, we highlight several interesting findings.
- Falcon and LLaMA excel in all four presented LLM model families, especially on the color and component sub-tasks.
- In-context learning (ICL) not only improves the visual commonsense performance of LLMs but also reduces their variance across different prompts.
- VaLMs improve the visual commonsense ability of their LLM backbones, despite small performance gains on the shape subset.
- ICL capability of VaLMs should be further valued.
How to Use
There are two folders in this repository, LLM
and VaLM
, representing two types of models considered in the paper: LLM and Visually-augmented LM.
For LLM, cd LLM
and install the environment by running pip install -r requirements.txt
. Run ImageNetVC.py
to obtain the experimantal results on ImageNetVC.
For VaLM, cd VaLM/BLIP-2
then follow BLIP-2 to install the environment and download necessary models. The code for ImagetNetVC is in ImageNetVC.py
.
Citation
Please kindly cite our paper if you find our datasets or code useful:
@inproceedings{xia-etal-2023-imagenetvc,
title = "ImageNetVC: Zero- and Few-Shot Visual Commonsense Evaluation on 1000 ImageNet Categories",
author = "Xia, Heming and
Dong, Qingxiu and
Li, Lei and
Xu, Jingjing and
Liu, Tianyu and
Qin, Ziwei and
Sui, Zhifang",
editor = "Bouamor, Houda and
Pino, Juan and
Bali, Kalika",
booktitle = "Findings of the Association for Computational Linguistics: EMNLP 2023",
month = dec,
year = "2023",
address = "Singapore",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2023.findings-emnlp.133",
pages = "2009--2026",
}