Learning sperm cells part segmentation with class-specific data augmentation

Infertility affects around 15% of couples worldwide. Male fertility problems include poor sperm quality and low sperm count. The advanced fertility treatment methods like ICSI are nowadays supported by vision systems to assist embryologists in selecting good quality sperm. Computer-Assisted Semen Analysis (CASA) provides quantitative and qualitative sperm analysis concerning concentration, motility, morphology, vitality, and fragmentation. However, fertility assessment algorithms often neglect individual spermatozoon tail and its beating patterns because recognizing the tails in blurry microscopic images reliably is challenging. In this article, we propose that models trained with head and tail part classes can better localize parts and segment the whole spermatozoon objects. Usually, the training of segmentation sperm models is supported by image-level augmentation. We argue that models guided by class-specific data augmentation attend to less discriminative sperm parts. To demonstrate this, we decouple the augmentation into object-level and background augmentation for the sperm part segmentation problem. Our proposed method outperforms state-of-the-art methods on the SegSperm dataset. Moreover, our ablation studies confirm the effectiveness of the proposed part-based object representation and augmentation.