Skip to content

Latest commit

 

History

History
210 lines (148 loc) · 9.4 KB

README.md

File metadata and controls

210 lines (148 loc) · 9.4 KB
Raw

VisOnlyQA

This repository contains the code and data for the paper "VisOnlyQA: Large Vision Language Models Still Struggle with Visual Perception of Geometric Information".

VisOnlyQA is designed to evaluate the visual perception capability of large vision language models (LVLMs) on geometric information of scientific figures. The evaluation set includes 1,200 mlutiple choice questions in 12 visual perception tasks on 4 categories of scientific figures. We also provide a training dataset consisting of 70k instances.

@misc{kamoi2024visonlyqa,
    title={VisOnlyQA: Large Vision Language Models Still Struggle with Visual Perception of Geometric Information}, 
    author={Ryo Kamoi and Yusen Zhang and Sarkar Snigdha Sarathi Das and Ranran Haoran Zhang and Rui Zhang},
    year={2024},
    journal={arXiv preprint arXiv:2412.00947}
}

Dataset

VisOnlyQA is provided in two formats: VLMEvalKit and Hugging Face Dataset. You can use either of them to evaluate your models and report the results in your papers. However, when you report the results, please explicitly mention which version of the dataset you used because the two versions are different.

Examples

VLMEvalKit

VLMEvalKit provides one-command evaluation. However, VLMEvalKit is not designed to reproduce the results in the paper. We welcome using it to report the results on VisOnlyQA in your papers, but please explicitly mention that you used VLMEvalKit.

The major differences are:

  • VisOnlyQA on VLMEvalKit does not include the chemistry__shape_multi split
  • VLMEvalKit uses different prompts and postprocessing.

Refer to this document for the installation and setup of VLMEvalKit. After setting up the environment, you can evaluate any supported models on VisOnlyQA with the following command (this example is for InternVL2-26B).

python run.py --data VisOnlyQA-VLMEvalKit --model InternVL2-26B

Hugging Face Dataset

The original VisOnlyQA dataset is provided in Hugging Face Dataset. If you want to reproduce the results in our paper, please use this version and code in the GitHub repository.

dataset folder of the GitHub repository includes identical datasets, except for the training data.

from datasets import load_dataset

real_eval = load_dataset("ryokamoi/VisOnlyQA_Eval_Real")
real_synthetic = load_dataset("ryokamoi/VisOnlyQA_Eval_Synthetic")

# Splits
print(real_eval.keys())
# dict_keys(['geometry__triangle', 'geometry__quadrilateral', 'geometry__length', 'geometry__angle', 'geometry__area', 'geometry__diameter_radius', 'chemistry__shape_single', 'chemistry__shape_multi', 'charts__extraction', 'charts__intersection'])

print(real_synthetic.keys())
# dict_keys(['syntheticgeometry__triangle', 'syntheticgeometry__quadrilateral', 'syntheticgeometry__length', 'syntheticgeometry__angle', 'syntheticgeometry__area', '3d__size', '3d__angle'])

# Prompt
print(real_eval['geometry__triangle'][0]['prompt_no_reasoning'])
# There is no triangle ADP in the figure. True or False?

# A triangle is a polygon with three edges and three vertices, which are explicitly connected in the figure.

# Your response should only include the final answer (True, False). Do not include any reasoning or explanation in your response.

# Image
print(real_eval['geometry__triangle'][0]['decoded_image'])
# <PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=103x165 at 0x7FB4F83236A0>

# Answer
print(real_eval['geometry__triangle'][0]['answer'])
# False

Data Format

Each instance of VisOnlyQA dataset has the following attributes:

Features

  • decoded_image: [PIL.Image] Input image
  • question: [string] Question (without instruction)
  • prompt_reasoning: [string] Prompt with intstruction to use chain-of-thought
  • prompt_no_reasoning: [string] Prompt with intstruction not to use chain-of-thought
  • answer: [string] Correct answer (e.g., True, a)

Metadata

  • image_path: [string] Path to the image file
  • image_category: [string] Category of the image (e.g., geometry, chemistry)
  • question_type: [string] single_answer or multiple answers
  • task_category: [string] Category of the task (e.g., triangle)
  • response_options: [List[string]] Multiple choice options (e.g., ['True', 'False'], ['a', 'b', 'c', 'd', 'e'])
  • source: [string] Source dataset
  • id: [string] Unique ID

Statistics

Directory Structure

Core directories and files of this repository:

.
├── dataset  # VisOnlyQA dataset (Eval-Real, Eval-Synthetic)
├── results
│   ├── model_resposnes  # Responses from LVLMs on VisOnlyQA
│   ├── evaluation_metrics  # Accuracy
│   ├── tables  # Tables in the paper
│   ├── figures  # Figures in the paper
│   └── analysis  # Analysis of the results
├── setup
│   └── setup.sh  # Run this script to setup the environment
├── shell  # Shell scripts for reproducing our experiments
├── src  # Source code
├── config  # Main config file is in src/config.py
└── finetuning_results  # Log files of the fine-tuning experiments

Setup

bash setup.sh

We run our experiments on the following environment. You might need to modify configulations if you run our code on a different environment.

  • Eight NVIDIA A100 SXM4 80GB GPUs
  • Driver Version: 550.54.15
  • CUDA Version: 12.4

Evaluate LVLMs on VisOnlyQA

Please refer to the shell scripts in the shell/4_evaluation folder.

# for small open LVLMs
bash shell/4_evaluation/evaluation_open_small.sh

Reproduce Fine-tuning

We fine-tuned the following LVLMs on VisOnlyQA-Train.

Our fine-tuning code is based on the code provided by the authors of the models. Please refer to the shell scripts in the shell/3_training folder for details.

bash shell/3_training/train_internvl2_4B.sh

Reproduce Dataset Creation

Datasets are provided in the dataset folder and at Hugging Face Datasets. You do not need to run the dataset creation code to use the datasets.

If you are interested in reproducing the dataset creation process, follow the instructions below.

Setup

If you are interested in reproducing the annotation interface: We use Google Spreadsheet for annotation. You need to set up Google API Credentials.

conda activate visonlyqa
export HF_ACCOUNT="your_hugging_face_account"  # dataset will be created in your HF account as private datasets
export CONDA_SH="~/anaconda3/etc/profile.d/conda.sh"  # set your anaconda path

Run

Refer to the shell files in shell/1_train_dataset_creation and shell/2_evaluation_dataset_creation.

License

Please refer to LICENSE.md.

Contact

If you have any questions, feel free to open an issue or reach out directly to Ryo Kamoi ([email protected]).