Importance of dataset design in developing robust U-Net models for label-free cell morphology evaluation

Advances in regenerative medicine highlighted the need for label-free cell image analysis to replace conventional microscopic observation for non-invasive cell quality evaluation. Image-based evaluation provides an efficient, quantitative, and automated approach to cell analysis, but segmentation remains a critical and challenging step. In this study, we investigated how training dataset design influenced the robustness of U-Net models for cell segmentation, focusing on challenges posed by limited data availability in cell culture. Using 2592 image pairs from four cell types representing key morphological categories, we constructed 42 investigation patterns to evaluate the effects of dataset size, dataset content, and morphological diversity on model performance. Our results showed that robust segmentation models could be developed with approximately 10 raw images captured using a 4× objective lens, a much smaller dataset than typically assumed. The dataset content was found to be crucial: training dataset images that captured commonly observed cell patterns yielded more robust models compared to those capturing rare or irregular cell patterns, which often impaired model performance with large deviations. Additionally, including both spindle and round cell morphologies in the training datasets improved model robustness when tested across all four cell types, while datasets restricted to a single morphology type could not achieve robust models. These findings highlight the importance of curating datasets that capture representative yet diverse cell morphologies. By addressing critical questions about dataset design, this study provides actionable guidance for the effective use of deep learning-based cell segmentation models in manufacturing and research applications.