MACHINE LEARNING MODELS FOR CELL LOCALIZATION AND CLASSIFICATION LEARNED USING REPEL CODING

The present disclosure relates to computer-implement techniques for cell localization and classification. Particularly, aspects of the present disclosure are directed to accessing an image for a biological sample, where the image depicts cells comprising a staining pattern of a biomarker; inputting the image into a machine learning model; encoding, by the machine learning model, the image into a feature representation comprising extracted discriminative features; combining, by the machine learning model, feature and spatial information of the cells and the staining pattern of the biomarker through a sequence of up-convolutions and concatenations with the extracted discriminative features from the feature representation; and generating, by the machine learning model, two or more segmentation masks for the biomarker in the image based on the combined feature and spatial information of the cells and the staining pattern of the biomarker.