Personalized venlafaxine dose prediction using artificial intelligence technology: a retrospective analysis based on real-world data

Venlafaxine dose regimens vary considerably between individuals, requiring personalized dosing. This study aimed to identify dose-related influencing factors of venlafaxine through real-world data analysis and to construct a personalized dose model using advanced artificial intelligence techniques. We conducted a retrospective study on patients with depression treated with venlafaxine. Significant variables were selected through a univariate analysis. Subsequently, the predictive performance of seven models (XGBoost, LightGBM, CatBoost, GBDT, ANN, TabNet, and DT) was compared. The algorithm that demonstrated optimal performance was chosen to establish the dose prediction model. Model validation used confusion matrices and ROC analysis. Additionally, a dose subgroup analysis was conducted. A total of 298 patients were included. TabNet was selected to establish the venlafaxine dose prediction model, which exhibited the highest performance with an accuracy of 0.80. The analysis identified seven crucial variables correlated with venlafaxine daily dose, including blood venlafaxine concentration, total protein, lymphocytes, age, globulin, cholinesterase, and blood platelet count. The area under the curve (AUC) for predicting venlafaxine doses of 75 mg, 150 mg, and 225 mg were 0.90, 0.85, and 0.90, respectively. We successfully developed a TabNet model to predict venlafaxine doses using real-world data. This model demonstrated substantial predictive accuracy, offering a personalized dosing regimen for venlafaxine. These findings provide valuable guidance for the clinical use of the drug.