A method including receiving, by an image classification engine, a number K of clusters to be created by a clustering algorithm. The method further including receiving, by the image classification engine, a set of elements based on an image, executing, by the image classification engine, the clustering algorithm on the set of elements to create K clusters, each cluster having a respective subset of the set of elements, for each cluster, computing, by the image classification engine, a centroid of the cluster, for each cluster, creating, by the image classification engine, a generator of the cluster based on the respective subset of the set of elements corresponding to each cluster, and for each element of each cluster, computing, by the image classification engine, a cost function corresponding to the element based on the centroid, the respective subset of the set of elements, and the generator corresponding to the cluster.