AI2D Dataset merupakan kumpulan data yang digunakan untuk melatih dan menguji model machine learning dalam mengenali diagram. Dataset ini berisi gambar diagram dengan anotasi yang mendetail. AI2D Dataset sangat berguna dalam pengembangan sistem yang mampu memahami dan memproses informasi visual dari diagram.