Support: (+84) 28 7308 5885 - (Ext) 303 or 310

EmbryosFormer: Deformable Transformer and Collaborative Encoding-Decoding for Embryos Stage Development Classification

https://doi.org/10.48550/arXiv.2210.04615

Tien-Phat Nguyen1,4, Trong-Thang Pham*4, Tri Nguyen**7, Hieu Le*1, Dung Nguyen5, Hau Lam6, Phong Nguyen1, Jennifer Fowler8, Minh-Triet Tran2,3,4, Ngan Le9

Published: February 9, 2023

Author information

Abstract

The timing of cell divisions in early embryos during the
In-Vitro Fertilization (IVF) process is a key predictor of embryo viability. However, observing cell divisions in Time-Lapse Monitoring (TLM) is a time-consuming process and highly depends on experts. In this paper, we propose EmbryosFormer, a computational model to automatically detect and classify cell divisions from original time-lapse images. Our proposed network is designed as an encoder-decoder deformable transformer with collaborative heads. 

The transformer contracting path predicts per-image labels and is optimized by a classification head. The transformer expanding path models the temporal coherency between embryo images to ensure monotonic non-decreasing constraint and is optimized by a segmentation head. Both contracting and expanding paths are synergetically learned by a collaboration head. We have benchmarked our proposed EmbryosFormer on two datasets: a public dataset with mouse embryos with 8-cell stage and an in-house dataset with human embryos with  4-cell stage. Source code: https://github.com/UARK-AICV/Embryos.