Abstract: A method may include obtaining a dataset including one or more data points. The method may include separating the dataset into one or more partitions based on a target number of subjects and a dimensionality of the data points included in the dataset. The method may include obtaining one or more weight vectors, each respective weight vector corresponding to a respective subject. The method may include selecting a first partition of the plurality of partitions to remove from the dataset based on respective relationships between a first weighted centroid of the dataset and first partition weights corresponding to each of the partitions. The method may include obtaining a first subset of the dataset by removing the data points associated with the selected first partition from the dataset. The method may include training a machine learning model based on the first subset of the dataset.