AI model predicts IVF live birth rates using novel attention network

Researchers have developed a novel hybrid attention network combined with a logistic regression model to predict live birth rates in IVF/ICSI cycles. This model, trained on a curated dataset of 616 cycles, achieved the highest test AUC-ROC of 0.710, outperforming classical machine learning models and foundation models like TabPFN. The study highlights the effectiveness of attention mechanisms in feature weighting for small clinical datasets and suggests that differential weighting of the inner cell mass and trophectoderm morphology is a key predictor. AI

IMPACT This research demonstrates a novel application of attention mechanisms in machine learning for small, tabular clinical datasets, potentially improving predictive accuracy in healthcare.

RANK_REASON The cluster contains an academic paper detailing a new machine learning model for a specific application. [lever_c_demoted from research: ic=1 ai=1.0]

Read on arXiv cs.LG →

• Abdul Raheem Shahzad
• AUC-ROC
• Hybrid Attention Network for Language-Based Person Search
• inner cell mass
• IVF/ICSI Protocols in Poor Responders With Growth Hormone
• k-nearest neighbors algorithm
• logistic regression model
• TabPFN

• paper
• model release
• other

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →