2026-07-08 08:57:50 +00:00

3.1 KiB
Raw Permalink Blame History

AI2D

概述

AI2DAI2 Diagrams是一个用于评估人工智能系统理解与推理科学图表能力的基准数据集。该数据集包含来自科学教科书的5,000多个多样化图表涵盖水循环、食物网和生物过程等主题。

任务描述

  • 任务类型:图表理解与视觉推理
  • 输入:科学图表图像 + 多选题
  • 输出:正确选项
  • 领域:科学教育、视觉推理、图表理解

主要特点

  • 图表来源于真实的科学教科书
  • 需要结合理解视觉布局、符号和文本标签
  • 测试对图表元素之间关系的解读能力
  • 采用包含具有挑战性干扰项的多选题格式
  • 覆盖多样化的科学领域(生物学、物理学、地球科学)

评估说明

  • 默认评估使用 test 划分
  • 主要指标:多选题的 准确率Accuracy
  • 使用思维链Chain-of-Thought, CoT提示进行推理
  • 需同时理解文本标签和视觉元素

属性

属性
基准测试名称 ai2d
数据集ID lmms-lab/ai2d
论文 N/A
标签 Knowledge, MultiModal, QA
指标 acc
默认示例数 0-shot
评估划分 test

数据统计

指标
总样本数 3,088
提示词长度(平均) 324.64 字符
提示词长度(最小/最大) 256 / 1024 字符

图像统计:

指标
总图像数 3,088
每样本图像数 最小: 1, 最大: 1, 平均: 1
分辨率范围 177x131 - 1500x1500
格式 png

样例示例

子集: default

{
  "input": [
    {
      "id": "789e28fa",
      "content": [
        {
          "text": "Answer the following multiple choice question. The last line of your response should be of the following format: 'ANSWER: [LETTER]' (without quotes) where [LETTER] is one of A,B,C,D. Think step by step before answering.\n\nwhich of these define dairy item\n\nA) c\nB) D\nC) b\nD) a"
        },
        {
          "image": "[BASE64_IMAGE: png, ~226.2KB]"
        }
      ]
    }
  ],
  "choices": [
    "c",
    "D",
    "b",
    "a"
  ],
  "target": "B",
  "id": 0,
  "group_id": 0
}

提示模板

提示模板:

Answer the following multiple choice question. The last line of your response should be of the following format: 'ANSWER: [LETTER]' (without quotes) where [LETTER] is one of {letters}. Think step by step before answering.

{question}

{choices}

使用方法

使用 CLI

evalscope eval \
    --model YOUR_MODEL \
    --api-url OPENAI_API_COMPAT_URL \
    --api-key EMPTY_TOKEN \
    --datasets ai2d \
    --limit 10  # 正式评估时请删除此行

使用 Python

from evalscope import run_task
from evalscope.config import TaskConfig

task_cfg = TaskConfig(
    model='YOUR_MODEL',
    api_url='OPENAI_API_COMPAT_URL',
    api_key='EMPTY_TOKEN',
    datasets=['ai2d'],
    limit=10,  # 正式评估时请删除此行
)

run_task(task_cfg=task_cfg)