AI2D는 AI2 다이어그램을 의미합니다. 이 데이터 세트에는 초등학교 과학 주제에 대한 5,000개 이상의 다이어그램과 15,000개 이상의 풍부한 주석이 포함되어 있습니다. 여기에는 다이어그램의 기본 구문 분석과 해당되는 15,000개 이상의 객관식 문제도 포함되어 있습니다.

AI2D 데이터 세트는 '블롭', 문자, 화살표, 화살촉 등 4가지 유형의 아이콘 요소를 생성합니다. AI2D 데이터 세트의 레이아웃 분할 맵과 파싱 맵에서 빨간색은 얼룩을, 파란색은 쓰여진 텍스트를, 녹색은 화살표를, 주황색은 화살촉을 나타냅니다.