Automated cell-level annotations for machine learning