MMGenBench: Fully Automatically Evaluating LMMs from the Text-to-Image Generation Perspective
1Beihang University, 2Alibaba Group, 3Tsinghua University
*Work done during an internship at Alibaba Group †Project Leader ‡Corresponding Author
📖Paper | 🏠Homepage | 🤗Huggingface
Large Multimodal Models (LMMs) demonstrate impressive capabilities. However, current benchmarks predominantly focus on image comprehension in specific domains, and these benchmarks are labor-intensive to construct. Moreover, their answers tend to be brief, making it difficult to assess the ability of LMMs to generate detailed descriptions of images. To address these limitations, we propose the MMGenBench-Pipeline, a straightforward and fully automated evaluation pipeline. This involves generating textual descriptions from input images, using these descriptions to create auxiliary images via text-to-image generative models, and then comparing the original and generated images. Furthermore, to ensure the effectiveness of MMGenBench-Pipeline, we design MMGenBench-Test, evaluating LMMs across 13 distinct image patterns, and MMGenBench-Domain, focusing on generative image performance. A thorough evaluation involving over 50 popular LMMs demonstrates the effectiveness and reliability of both the pipeline and benchmark. Our observations indicate that numerous LMMs excelling in existing benchmarks fail to adequately complete the basic tasks related to image understanding and description. This finding highlights the substantial potential for performance improvement in current LMMs and suggests avenues for future model optimization. Concurrently, MMGenBench-Pipeline can efficiently assess the performance of LMMs across diverse domains using only image inputs. All code and data will be released.
Usage
Getting Started
Environment Installation
Clone this repository
git clone git@github.com:lerogo/MMGenBench.git
cd MMGenBenchDownload dataset
huggingface-cli download --repo-type dataset lerogo/MMGenBench --local-dir MMGenBench-data
Install the relevant environment, including torch, transformers, diffusers and unicom (used to extract image representation).
Preliminary
We use the InternVL2-2B as an example. The structure of the code and data is as follows.
. ├── MMGenBench-data # The MMGenBench-Test/Domain dataset we downloaded from huggingface │ ├── MMGenBench-Domain.json │ ├── MMGenBench-Domain.tsv │ ├── MMGenBench-Test-label-count.json │ ├── MMGenBench-Test-label-index.json │ ├── MMGenBench-Test.json │ ├── MMGenBench-Test.tsv │ ├── README.md │ └── check.py ├── README.md # This file ├── evalimg # For extracting features and calculating metrics using the image representation model │ ├── metric_fid.py │ ├── output │ │ ├── InternVL2-2B_MMGenBench-Domain.json │ │ └── InternVL2-2B_MMGenBench-Test.json │ ├── requirements.txt │ ├── run.py │ └── run.sh ├── generate # For processing LMMs' output with the text-to-image models │ ├── flux.py │ ├── input │ │ ├── InternVL2-2B_MMGenBench-Domain.xlsx │ │ └── InternVL2-2B_MMGenBench-Test.xlsx │ ├── kolors.py │ ├── lumina.py │ ├── output │ │ ├── InternVL2-2B_MMGenBench-Domain.tsv │ │ └── InternVL2-2B_MMGenBench-Test.tsv │ ├── requirements.txt │ ├── run.py │ ├── run.sh │ ├── sd.py │ └── tools.py └── visual # For visualization ├── outputs │ ├── InternVL2-2B_MMGenBench-Domain.json │ ├── InternVL2-2B_MMGenBench-Domain.xlsx │ ├── InternVL2-2B_MMGenBench-Test.json │ └── InternVL2-2B_MMGenBench-Test.xlsx ├── run.py └── run.sh
Evaluation Pipeline
Stage 1
Adapt your model in VLMEvalKit and use MMGenBench for inference.
Run command:
torchrun --nproc-per-node=4 run.py --model <YOUR LMM> --data MMGenBench-Test MMGenBench-Domain --mode infer --verbose
We use the InternVL2-2B as an example. Then you can get two files: InternVL2-2B_MMGenBench-Test.xlsx, InternVL2-2B_MMGenBench-Domain.xlsx. Put them in folder ./generate/input
Stage 2
Modify ./generate/run.sh to select the text-to-image model and to select the number of GPUs you need to use.
And run:
Then you can get two files: ./generate/output/InternVL2-2B_MMGenBench-Test.tsv, ./generate/output/InternVL2-2B_MMGenBench-Domain.tsv
Stage 3
We will use the unicom model to extract features from the original images and generated images, you need to install unicom (https://github.com/deepglint/unicom).
Modify ./evalimg/run.sh to evaluate the performance on MMGenBench-Test and MMGenBench-Domain respectively.
And run:
Then you can get two files: evalimg/output/InternVL2-2B_MMGenBench-Test.json, ./evalimg/output/InternVL2-2B_MMGenBench-Domain.json.
Visual
Run command:
You can see the relevant results in the output folder, including metrics and visualization results.
Q&A
If you have any questions, please submit an issue or contact lerogohl<AT>gmail.com.
Citation
If you find MMGenBench or code useful, please cite
@misc{huang2024MMGenBench, title={MMGenBench: Fully Automatically Evaluating LMMs from the Text-to-Image Generation Perspective}, author={Hailang Huang and Yong Wang and Zixuan Huang and Huaqiu Li and Tongwen Huang and Xiangxiang Chu and Richong Zhang}, year={2024}, eprint={2411.14062}, archivePrefix={arXiv}, primaryClass={cs.CV}, url={https://arxiv.org/abs/2411.14062}, }