MedSaab-US: A Backpropagation-Free Multi-Scale Wavelet-Saab Framework for Thyroid Nodule Segmentation in Ultrasound Images

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Medical Imaging AI · Depth: Expert, quick

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

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

Topics

Best for: Computer Vision Engineer, AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.