MedSaab-US: A Backpropagation-Free Multi-Scale Wavelet-Saab Framework for Thyroid Nodule Segmentation in Ultrasound Images
Summary
MedSaab-US is a novel backpropagation-free framework designed for thyroid nodule segmentation in ultrasound images, addressing the high parameter counts and GPU dependency of deep learning methods. Grounded in the Green Learning paradigm, MedSaab-US extracts multi-scale spatial-frequency features by integrating multi-level Discrete Wavelet Transform (DWT) with multi-scale channel-wise Saab transforms, using patch sizes of 5 x 5, 11 x 11, and 21 x 21 pixels. It employs Label-Assisted Greedy (LAG) feature selection to identify discriminative features, which are then classified pixel-wise by an XGBoost model. Evaluated on the TN3K dataset (2,879 training, 614 test images), MedSaab-US achieved a mean Dice coefficient of 0.4784 +/- 0.2190. The framework boasts a model footprint under 500K parameters and performs CPU-only inference in approximately 0.3 seconds per image, offering an efficient non-DL baseline.
Key takeaway
For Machine Learning Engineers deploying medical image segmentation in resource-constrained environments, MedSaab-US presents a compelling alternative to deep learning. Its backpropagation-free design, low parameter count (under 500K), and CPU-only inference (0.3 seconds per image) make it highly efficient. You should consider this Green Learning-based framework to achieve robust thyroid nodule segmentation without the heavy computational demands of traditional DL models, especially when GPU resources are limited.
Key insights
MedSaab-US provides an efficient, backpropagation-free alternative for thyroid nodule segmentation, overcoming deep learning's resource demands.
Principles
- Green Learning enables resource-efficient models.
- Analytical parameter derivation reduces training complexity.
- Iterative greedy construction bypasses backpropagation.
Method
MedSaab-US combines multi-level DWT with multi-scale Saab transforms for feature extraction. Label-Assisted Greedy (LAG) selects discriminative features, which an XGBoost classifier then uses for pixel-wise prediction.
In practice
- Deploy in resource-constrained medical settings.
- Use as a non-DL baseline for segmentation.
- Enable rapid CPU-only inference for diagnostics.
Topics
- Thyroid Nodule Segmentation
- Ultrasound Imaging
- Green Learning
- Wavelet Transform
- Saab Transform
- XGBoost
Best for: Computer Vision Engineer, AI Scientist, Machine Learning Engineer, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.