AI2D steht für AI2-Diagramme. Der Datensatz umfasst mehr als 5.000 Diagramme zu naturwissenschaftlichen Themen der Grundschule mit über 15.000 ausführlichen Anmerkungen. Es umfasst außerdem die Ground-Truth-Syntaxanalyse der Diagramme und mehr als 15.000 entsprechende Multiple-Choice-Fragen.

Der AI2D-Datensatz erstellt vier Arten von Symbolelementen: „Blobs“, geschriebenen Text, Pfeile und Pfeilspitzen. In der Layout-Segmentierungskarte und der Parsing-Karte des AI2D-Datensatzes steht Rot für Blob, Blau für geschriebenen Text, Grün für Pfeil und Orange für Pfeilspitze.