The disclosed technology provides solutions for finding samples from image data that are similar to failure cases, by constructing N-dimensional vectors of the failure cases. The vectors of failure cases are compared to other image data, with the objective of identifying groups of images that can be labeled. The labeled images are then used to retrain a model. Systems and machine-readable media are also provided.