无需反向传播的多尺度小波分割框架,实现超声甲状腺结节快速精准定位。
MedSaab-US: A Backpropagation-Free Multi-Scale Wavelet-Saab Framework for Thyroid Nodule Segmentation in Ultrasound Images

- 结合小波变换与Saab变换提取多尺度特征,避免深度学习训练
- 在TN3K数据集上达Dice 0.478,单图推理仅需0.3秒
- 模型小于50万参数,适合低算力设备部署,特别适合等回声结节
深度学习方法虽在超声甲状腺结节分割中表现优异,但依赖反向传播、参数量大且难以解释。本文提出MedSaab-US,一种基于Green Learning范式的无反向传播分割框架。该方法通过5×5、11×11和21×21像素块的多级离散小波变换(DWT)与多尺度通道式Saab变换,提取多尺度空间-频率特征;采用标签辅助贪心(LAG)特征选择保留最具判别性的特征,输入XGBoost分类器进行像素级预测。Saab变换参数由数据统计解析确定,XGBoost通过迭代贪心树构建,无需反向传播。在包含2,879张训练图和614张测试图的TN3K数据集上,模型平均Dice系数为0.4784±0.2190,精确率0.5768,召回率0.5604,模型规模低于50万参数,支持纯CPU推理,单图耗时约0.3秒。研究将此作为甲状腺超声分割的非深度学习基准,并分析等回声结节带来的挑战。消融实验量化了各模块贡献,包括特征选择与训练集规模的影响。
原文摘要 · Abstract (English)
Deep learning (DL) methods dominate thyroid nodule segmentation in ultrasound (US) images, achieving high Dice scores but at the cost of millions of parameters, GPU-dependent training via backpropagation, and limited mathematical tractability. These limitations impede deployment in resource-constrained environments. In this paper, we propose MedSaab-US, a backpropagation-free segmentation framework grounded in the Green Learning paradigm. MedSaab-US extracts multi-scale spatial-frequency features by combining multi-level Discrete Wavelet Transform (DWT) with multi-scale channel-wise Saab (Subspace Approximation with Adjusted Bias) transforms at patch sizes of 5 x 5, 11 x 11, and 21 x 21 pixels. Label-Assisted Greedy (LAG) feature selection retains the most discriminative features, which are fed to an XGBoost classifier for pixel-wise prediction. The Saab transform parameters are determined analytically from data statistics, while XGBoost employs iterative greedy tree construction without requiring backpropagation. Evaluated on the TN3K dataset (2,879 training and 614 test images), MedSaab-US achieves a mean Dice coefficient of 0.4784 +/- 0.2190, precision of 0.5768, and recall of 0.5604, with a model footprint under 500K parameters and CPU-only inference in approximately 0.3 seconds per image. We present this result as an exploratory non-DL baseline for thyroid ultrasound segmentation and analyze the specific challenges posed by isoechoic nodules. An ablation study further quantifies the contribution of each pipeline component, including separate evaluations of LAG feature selection and training-set size.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。