arXiv:2512.17934cs.LGcs.AI2025-12被引 2

用机器学习预测美国县区肺癌死亡率,发现吸烟和族裔占比是关键因素。

Comparative Evaluation of Explainable Machine Learning Versus Linear Regression for Predicting County-Level Lung Cancer Mortality Rate in the United States

  • 采用随机森林等模型比线性回归更准确预测肺癌死亡率。
  • 随机森林模型R²达41.9%,误差为12.8,优于其他模型。
  • 揭示东部地区存在高死亡率聚集区,适合公共卫生决策参考。

肺癌是美国主要癌症致死原因。准确预测县区肺癌死亡率对制定靶向干预措施、缓解健康不平等至关重要。本研究比较了随机森林(RF)、梯度提升回归(GBR)与线性回归(LR)三种模型在全美县区层面的预测表现。模型性能通过决定系数(R²)和均方根误差(RMSE)评估,使用SHAP值分析变量重要性及其影响方向。地理空间分析采用Getis-Ord (Gi*)热点检测。结果显示,随机森林模型表现最优,R²为41.9%,RMSE为12.8。SHAP分析表明,吸烟率是最关键的预测因子,其次为中位房价和西班牙裔人口比例。空间分析发现美国中部东部县区存在显著的肺癌高死亡率聚集区。该结果凸显了吸烟流行率、住房价值及西班牙裔人口占比的关键作用,为制定筛查策略和改善高风险区域健康公平提供可操作依据。

原文摘要 · Abstract (English)

Lung cancer (LC) is a leading cause of cancer-related mortality in the United States. Accurate prediction of LC mortality rates is crucial for guiding targeted interventions and addressing health disparities. Although traditional regression-based models have been commonly used, explainable machine learning models may offer enhanced predictive accuracy and deeper insights into the factors influencing LC mortality. This study applied three models: random forest (RF), gradient boosting regression (GBR), and linear regression (LR) to predict county-level LC mortality rates across the United States. Model performance was evaluated using R-squared and root mean squared error (RMSE). Shapley Additive Explanations (SHAP) values were used to determine variable importance and their directional impact. Geographic disparities in LC mortality were analyzed through Getis-Ord (Gi*) hotspot analysis. The RF model outperformed both GBR and LR, achieving an R2 value of 41.9% and an RMSE of 12.8. SHAP analysis identified smoking rate as the most important predictor, followed by median home value and the percentage of the Hispanic ethnic population. Spatial analysis revealed significant clusters of elevated LC mortality in the mid-eastern counties of the United States. The RF model demonstrated superior predictive performance for LC mortality rates, emphasizing the critical roles of smoking prevalence, housing values, and the percentage of Hispanic ethnic population. These findings offer valuable actionable insights for designing targeted interventions, promoting screening, and addressing health disparities in regions most affected by LC in the United States.

肺癌预测可解释AI空间分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。