An uncertainty-aware approach to postoperative knee satisfaction prediction after total knee arthroplasty: nested cross-fitted ensemble learning with conformal prediction intervals

Prediction of continuous patient-reported outcomes from limited tabular clinical data is challenging because model rankings may vary across learning algorithms and predictive uncertainty is not captured by point estimates alone. This study evaluated postoperative knee satisfaction using individual regression learners, nested cross-fitted ensemble methods, and distribution-free prediction intervals. An analytic dataset of 98 observations was evaluated using Ridge, Elastic Net, Random Forest, XGBoost, and CatBoost. Four ensemble strategies were considered: uniform averaging, inverse-MAE weighting, Ridge stacking, and a convex Super Learner. Predictive performance was assessed using five-fold outer cross-validation. Within each outer-training set, a separate five-fold inner cross-validation procedure was used to estimate all data-dependent ensemble weights and meta-learner coefficients. Split-conformal prediction was embedded within the outer evaluation to generate finite-sample corrected 95% prediction intervals. SHAP and a perturbation-based local surrogate analysis were used for post-hoc interpretation of full-data base-learner refits. Random Forest achieved the lowest overall MAE (2.150) and MAPE (8.603%), whereas the cross-fitted convex Super Learner yielded the lowest RMSE (3.257) and the highest \(\:{R}^{2}\) (0.792). Its numerical advantage over Ridge was small, with Ridge obtaining an RMSE of 3.280 and an \(\:{R}^{2}\) of 0.789. Among the ensemble methods, inverse-MAE weighting produced the lowest MAE (2.168) and MAPE (8.766%), while uniform averaging showed similar performance. Empirical conformal coverage ranged from 0.949 to 0.980. Ridge generated the narrowest mean prediction intervals overall (17.683), whereas the convex Super Learner generated the narrowest intervals among the ensemble methods (19.812). The SHAP and local surrogate analyses primarily attributed predictions to KSS-related subscale variables. Nested cross-fitting prevented the second-stage ensemble parameters from being estimated and evaluated on the same observations. No single learning method consistently dominated all performance measures, and the observed advantages were small and metric-dependent. The framework provides an internally evaluated approach for combining point prediction with marginal uncertainty quantification, but external validation with larger cohorts and clinically independent predictors is required.

Authors

Institutions

Publication Details

Journal
BMC Medical Informatics and Decision Making
Published
2026-08-19
DOI
https://doi.org/10.1186/s12911-026-03771-z
Primary Topic
Total Knee Arthroplasty Outcomes
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

An uncertainty-aware approach to postoperative knee satisfaction prediction after total knee arthroplasty: nested cross-fitted ensemble learning with conformal prediction intervals

İbrahim Budak, Recep Taşkın
BMC Medical Informatics and Decision Making
Total Knee Arthroplasty Outcomes
article

An uncertainty-aware approach to postoperative knee satisfaction prediction after total knee arthroplasty: nested cross-fitted ensemble learning with conformal prediction intervals

İbrahim Budak, Recep Taşkın
article en

Abstract

Prediction of continuous patient-reported outcomes from limited tabular clinical data is challenging because model rankings may vary across learning algorithms and predictive uncertainty is not captured by point estimates alone. This study evaluated postoperative knee satisfaction using individual regression learners, nested cross-fitted ensemble methods, and distribution-free prediction intervals. An analytic dataset of 98 observations was evaluated using Ridge, Elastic Net, Random Forest, XGBoost, and CatBoost. Four ensemble strategies were considered: uniform averaging, inverse-MAE weighting, Ridge stacking, and a convex Super Learner. Predictive performance was assessed using five-fold outer cross-validation. Within each outer-training set, a separate five-fold inner cross-validation procedure was used to estimate all data-dependent ensemble weights and meta-learner coefficients. Split-conformal prediction was embedded within the outer evaluation to generate finite-sample corrected 95% prediction intervals. SHAP and a perturbation-based local surrogate analysis were used for post-hoc interpretation of full-data base-learner refits. Random Forest achieved the lowest overall MAE (2.150) and MAPE (8.603%), whereas the cross-fitted convex Super Learner yielded the lowest RMSE (3.257) and the highest \(\:{R}^{2}\) (0.792). Its numerical advantage over Ridge was small, with Ridge obtaining an RMSE of 3.280 and an \(\:{R}^{2}\) of 0.789. Among the ensemble methods, inverse-MAE weighting produced the lowest MAE (2.168) and MAPE (8.766%), while uniform averaging showed similar performance. Empirical conformal coverage ranged from 0.949 to 0.980. Ridge generated the narrowest mean prediction intervals overall (17.683), whereas the convex Super Learner generated the narrowest intervals among the ensemble methods (19.812). The SHAP and local surrogate analyses primarily attributed predictions to KSS-related subscale variables. Nested cross-fitting prevented the second-stage ensemble parameters from being estimated and evaluated on the same observations. No single learning method consistently dominated all performance measures, and the observed advantages were small and metric-dependent. The framework provides an internally evaluated approach for combining point prediction with marginal uncertainty quantification, but external validation with larger cohorts and clinically independent predictors is required.

BMC Medical Informatics and Decision Making
Kastamonu University (TR)
Openalex Percentile: Top 8%
Total Knee Arthroplasty Outcomes
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.