arXiv:2606.04453cs.CVcs.LG2026-06

用深度网络梯度损失筛选肺癌分期特征,准确率达90.22%

Radiomic Feature Selection Using Gradient Loss of Deep Neural Network for Lung Cancer Stage Detection

  • 基于深度网络损失梯度计算特征重要性,递归剔除无关特征
  • 仅用15个核心特征即实现90.22%分类准确率
  • 适合小样本高维医学数据,结果可解释性强

放射组学能从医学影像中提取定量生物标志物,已成为辅助癌症诊断的重要工具。然而,放射组学数据集通常维度高、样本少,特征选择成为构建可靠预测模型的关键步骤。本研究提出一种梯度损失递归特征消除(GL-RFE)框架,利用深度神经网络的梯度敏感性分析,识别对肺癌分期最具影响力的放射组学特征。共从胸部CT扫描中使用3D Slicer平台的PyRadiomics插件提取了106个放射组学特征。该方法通过计算网络损失对输入特征的梯度,递归剔除贡献最小的特征。最终选用前15个特征训练深度神经网络分类器,用于区分早期与晚期肺癌。该框架在测试集上达到90.22%准确率、90.10%精确率、90.24%召回率和90.16%F1分数。可视化分析(包括相关热图和分布图)进一步证实特征冗余减少且类别可分性提升。相比传统特征选择方法,GL-RFE能有效捕捉非线性特征交互,增强模型泛化能力。所提方案为放射组学驱动的癌症分期检测提供了可复现、可解释的方法,尤其适用于小样本高维生物医学数据,具有在基因组学和多模态临床分析等领域的潜在应用价值。

原文摘要 · Abstract (English)

Radiomics enables extraction of quantitative imaging biomarkers from medical images and has become an important tool for computer-aided cancer diagnosis. However, radiomics datasets are typically high-dimensional with limited samples, making feature selection a critical step for building reliable predictive models. This study proposes a Gradient-Loss Recursive Feature Elimination (GL-RFE) framework that integrates gradient sensitivity analysis from a deep neural network to identify the most influential radiomic features for lung cancer stage detection. A total of 106 radiomic features were extracted from chest Computed Tomography (CT) scans using the PyRadiomics extension of the 3D Slicer platform. The proposed method evaluates feature importance by computing gradients of the network loss with respect to input features and recursively eliminates features with minimal contribution. The resulting top-15 radiomic features are used to train a deep neural network classifier for distinguishing early-stage and advanced-stage lung cancer. The proposed framework achieves strong classification performance, with accuracy of 90.22%, precision of 90.10%, recall of 90.24%, and F1-score of 90.16% on the test dataset. Visualization analyses, including correlation heat maps and distribution plots, further confirm reduced feature redundancy and improved class separability. Compared to conventional feature selection techniques, GL-RFE effectively captures nonlinear feature interactions and enhances model generalization. The presented protocol provides a reproducible and interpretable methodology for radiomics-based cancer stage detection and is particularly suitable for high-dimensional, small-sample biomedical datasets, with potential applications in other domains such as genomics and multimodal clinical analysis.

放射组学特征选择肺癌分期深度学习

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