Back

Assessment of machine learning algorithms to predict medical specialty choice

Alvarez, D. V.; Abbiati, M.; Bornet, A.; Savoldelli, G.; Bajwa, N.; Teodoro, D.

2025-03-12 health informatics
10.1101/2025.03.06.25323485 medRxiv
Show abstract

Equitable distribution of physicians across specialties is a significant public health challenge. While previous studies primarily relied on classic statistics models to estimate factors affecting medical students career choices, this study explores the use of machine learning techniques to predict decisions early in their studies. We evaluated various supervised models, including support vector machines, artificial neural networks, extreme gradient boosting (XGBoost), and CatBoost using data from 399 medical students from medical faculties in Switzerland and France. Ensemble methods outperformed simpler models, with CatBoost achieving a macro AUROC of 76%. Post-hoc interpretability methods revealed key factors influencing predictions, such as motivation to become a surgeon and psychological traits like extraversion. These findings show that machine learning could be used for predicting medical career paths and inform better workforce planning.

Matching journals

The top 5 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.