arXiv:2412.01255eess.IVcs.CV2024-12被引 10

用合成胚胎图像增强真实数据,提升AI预测胚胎发育阶段的准确率

Merging synthetic and real embryo data for advanced AI predictions

  • 用两个生成模型合成2-8细胞、桑葚胚和囊胚阶段图像
  • 融合合成与真实数据使分类准确率达97%(纯真实数据为94.5%)
  • 合成数据效果接近真实数据,适合数据稀缺领域的研究者

精准评估胚胎形态对辅助生殖技术中选择高活力胚胎至关重要。人工智能有潜力提升该过程,但胚胎数据有限制约了深度学习模型的训练。为此,我们利用两个数据集——一个自建并公开的数据集和一个现有公开数据集——训练了两种生成模型,以生成包括2细胞、4细胞、8细胞、桑葚胚和囊胚在内的多个发育阶段的合成胚胎图像。这些合成图像与真实图像结合,用于训练胚胎细胞阶段分类模型。结果表明,融合合成数据后,模型分类准确率达到97%,高于仅使用真实数据时的94.5%。在另一家诊所的外部囊胚数据集上测试也保持此趋势。值得注意的是,即使仅用合成数据训练并在真实数据上测试,模型仍达到92%的准确率。此外,结合两个生成模型的合成数据比单一模型效果更优。四位胚胎学家通过图灵测试评估合成图像保真度,指出扩散模型表现优于生成对抗网络,欺骗率分别为66.6%和25.3%,且弗雷歇起始距离更低。

原文摘要 · Abstract (English)

Accurate embryo morphology assessment is essential in assisted reproductive technology for selecting the most viable embryo. Artificial intelligence has the potential to enhance this process. However, the limited availability of embryo data presents challenges for training deep learning models. To address this, we trained two generative models using two datasets-one we created and made publicly available, and one existing public dataset-to generate synthetic embryo images at various cell stages, including 2-cell, 4-cell, 8-cell, morula, and blastocyst. These were combined with real images to train classification models for embryo cell stage prediction. Our results demonstrate that incorporating synthetic images alongside real data improved classification performance, with the model achieving 97% accuracy compared to 94.5% when trained solely on real data. This trend remained consistent when tested on an external Blastocyst dataset from a different clinic. Notably, even when trained exclusively on synthetic data and tested on real data, the model achieved a high accuracy of 92%. Furthermore, combining synthetic data from both generative models yielded better classification results than using data from a single generative model. Four embryologists evaluated the fidelity of the synthetic images through a Turing test, during which they annotated inaccuracies and offered feedback. The analysis showed the diffusion model outperformed the generative adversarial network, deceiving embryologists 66.6% versus 25.3% and achieving lower Frechet inception distance scores.

AI辅助生殖生成模型数据增强胚胎发育

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。