Abstract: A method for training a machine learning model. The method comprises receiving a training dataset that includes a plurality of images. The method also includes identifying, by a machine learning model, at least one portion of at least one image of the plurality of images in the training dataset associated with a first object type. The method further includes identifying other images having at least one portion that includes the first object type. The method also includes grouping the identified other images into a first image group. The method also includes generating for display a first user interface, that at least includes a rank matrix, wherein a first row of the rank matrix represents the images of the first image object. The user may provide feedback for the visualization using the first interface. The method may also include training the machine learning model based on the user feedback.