Back

Radiomics-Based Lung Nodule Classification with Stacking Ensembles

Lam, I.; Koh, S. M.; Thanh, M. N.

2025-04-29 oncology
10.1101/2025.04.28.25326620 medRxiv
Show abstract

Radiomics, an emerging field in medical imaging, leverages advanced mathematical analysis to extract quantitative metrics from medical images, aiding in the early detection, diagnosis, and treatment of lung cancer. This study focuses on improving the risk prediction of small lung nodules using machine learning models. We employed a stacking ensemble approach, integrating Principal Component Analysis (PCA) for dimensionality reduction and selected features from the Small Nodule Radiomics-Predictive Vector (SN-RPV). Base models employed in the stacking ensemble were Support Vector Machine (SVM), Random Forest, k-Nearest Neighbors (KNN), and Naive Bayes classifiers. Despite the theoretical advantages of stacking ensembles, our models demonstrated poorer performance on the test set compared to the simpler SN-RPV model by Hunter et al. This outcome highlights the challenges of overfitting and underscores the importance of model simplicity and interpretability in clinical applications. Future research should explore alternative regularization techniques to improve the generalization of complex ensemble methods.

Matching journals

The top 4 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.