A method of using a machine learning model to output a task-specific prediction may include receiving a digitized cytology image of a cytology sample and applying a machine learning model to isolate cells of the digitized cytology image. The machine learning model may include identifying a plurality of sub-portions of the digitized cytology image, identifying, for each sub-portion of the plurality of sub-portions, either background or cell, and determining cell sub-images of the digitized cytology image. Each cell sub-image may comprise a cell of the digitized cytology image, based on the identifying either background or cell. The method may further comprise determining a plurality of features based on the cell sub-images, each of the cell sub-images being associated with at least one of the plurality of features, determining an aggregated feature based on the plurality of features, and training a machine learning model to predict a target task based on the aggregated feature.