A machine learning-based risk scoring system for infertility considering different age groups

摘要

The application of artificial intelligence (AI) methods in medical field is increasing year by year; however, few studies have applied AI methods in the re-productive field. In view of the complexity of infertility diagnosis and treatment, a machine learning-based risk scoring system for infertility was constructed in this paper to help clinicians better grasp the patient's condition. First, eight key features of infertility are screened out by feature selection. Second, the entropy-based feature discretization method was used to divide the feature abnormal intervals, and the random forest was used to determine the weight of each feature. Finally, the pregnancy outcome can be predicted according to the overall risk score of patients, which is helpful for doctors to choose targeted treatment more efficiently. It is worth noting that, to further improve the accuracy of the diagnosis, we also divided the patients into age groups and constructed the cor-responding risk scoring system for patients of different age groups. The stability test results show the good performance of the system. The risk scoring system for infertility built in this paper is a meaningful exploration of the application of AI in the field of reproduction.