AI2D signifie Diagrammes AI2. L'ensemble de données comprend plus de 5 000 diagrammes sur des sujets scientifiques de l'école primaire avec plus de 15 000 annotations riches. Il comprend également l'analyse syntaxique de la vérité fondamentale des diagrammes et plus de 15 000 questions à choix multiples correspondantes.

L'ensemble de données AI2D crée quatre types d'éléments d'icônes : « blobs », texte écrit, flèches et pointes de flèches. Dans la carte de segmentation de mise en page et la carte d'analyse de l'ensemble de données AI2D, le rouge représente un blob, le bleu représente un texte écrit, le vert représente une flèche et l'orange représente une pointe de flèche.