Iterative pseudo balancing for stem cell microscopy image classification

Sci Rep. 2024 Feb 24;14(1):4489. doi: 10.1038/s41598-024-54993-y.

Abstract

Many critical issues arise when training deep neural networks using limited biological datasets. These include overfitting, exploding/vanishing gradients and other inefficiencies which are exacerbated by class imbalances and can affect the overall accuracy of a model. There is a need to develop semi-supervised models that can reduce the need for large, balanced, manually annotated datasets so that researchers can easily employ neural networks for experimental analysis. In this work, Iterative Pseudo Balancing (IPB) is introduced to classify stem cell microscopy images while performing on the fly dataset balancing using a student-teacher meta-pseudo-label framework. In addition, multi-scale patches of multi-label images are incorporated into the network training to provide previously inaccessible image features with both local and global information for effective and efficient learning. The combination of these inputs is shown to increase the classification accuracy of the proposed deep neural network by 3[Formula: see text] over baseline, which is determined to be statistically significant. This work represents a novel use of pseudo-labeling for data limited settings, which are common in biological image datasets, and highlights the importance of the exhaustive use of available image features for improving performance of semi-supervised networks. The proposed methods can be used to reduce the need for expensive manual dataset annotation and in turn accelerate the pace of scientific research involving non-invasive cellular imaging.

Keywords: Deep learning; Pseudo-labels; Stem cell microscopy.

MeSH terms

  • Humans
  • Image Processing, Computer-Assisted
  • Learning*
  • Microscopy*
  • Neural Networks, Computer
  • Product Labeling
  • Stem Cells
  • Supervised Machine Learning