The features relevant to the task include: 
* question: the question to be answered from the image
* image_id: the path to the image the question refers to
* label: the annotations
We can remove the rest of the features as they won't be necessary: 
 

dataset = dataset.remove_columns(['question_type', 'question_id', 'answer_type'])