用频谱总变差特征构建弱学习器集成,提升骨转移预测准确率
Ensemble of Weak Spectral Total Variation Learners: a PET-CT Case Study
- 基于频谱总变差特征设计低相关性弱学习器并集成
- 在457例配准数据上AUC达0.87,优于深度学习和影像组学
- 细尺度频谱特征对高代谢病灶预测最具指示意义
通过机器学习解决计算机视觉问题时,常面临训练数据不足。为此,本文提出基于频谱总变差(STV)特征的弱学习器集成方法(Gilboa 2014)。STV特征与总变差次梯度的非线性本征函数相关,可有效表征多尺度纹理。已有研究(Burger et al. 2016)表明,一维情况下生成正交特征,二维情况下特征经验上低相关。集成学习理论支持使用低相关弱学习器,因此本文设计基于STV特征的学习器集成。为验证有效性,研究聚焦一项困难的医学影像任务:预测疑似骨转移患者中正电子发射断层扫描(PET)高摄取的临床价值。数据集包含457例扫描,共1524对注册的CT与PET切片。方法与深度学习及影像组学特征对比,结果显示STV学习器表现最佳(AUC=0.87),优于神经网络(AUC=0.75)和影像组学(AUC=0.79)。研究发现,CT图像中精细尺度的STV特征尤其能指示PET高摄取的存在。
原文摘要 · Abstract (English)
Solving computer vision problems through machine learning, one often encounters lack of sufficient training data. To mitigate this we propose the use of ensembles of weak learners based on spectral total-variation (STV) features (Gilboa 2014). The features are related to nonlinear eigenfunctions of the total-variation subgradient and can characterize well textures at various scales. It was shown (Burger et-al 2016) that, in the one-dimensional case, orthogonal features are generated, whereas in two-dimensions the features are empirically lowly correlated. Ensemble learning theory advocates the use of lowly correlated weak learners. We thus propose here to design ensembles using learners based on STV features. To show the effectiveness of this paradigm we examine a hard real-world medical imaging problem: the predictive value of computed tomography (CT) data for high uptake in positron emission tomography (PET) for patients suspected of skeletal metastases. The database consists of 457 scans with 1524 unique pairs of registered CT and PET slices. Our approach is compared to deep-learning methods and to Radiomics features, showing STV learners perform best (AUC=0.87), compared to neural nets (AUC=0.75) and Radiomics (AUC=0.79). We observe that fine STV scales in CT images are especially indicative for the presence of high uptake in PET.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。