AI2D は AI2 ダイアグラムの略です。このデータセットには、小学校の自然科学のトピックに関する 5,000 を超える図と、15,000 を超える豊富な注釈が含まれています。また、グラフの正確な構文分析と、対応する 15,000 を超える多肢選択式の質問も含まれています。

AI2D データセットは、「ブロブ」、書かれたテキスト、矢印、矢じりの 4 種類のアイコン要素を確立します。 AI2D データセットのレイアウト分割図と分析図では、赤がブロブ、青が書かれたテキスト、緑が矢印、オレンジが矢じりを表します。