UNICORN: Ultrasound Nakagami Imaging via Score Matching and Adaptation for Assessing Hepatic Steatosis
Organizations: Department of AI Convergence, GIST · Kim Jae Chul Graduate School of AI, KAIST · Samsung Medison
Abstract
Ultrasound imaging is an essential first-line tool for assessing hepatic steatosis. While conventional B-mode ultrasound imaging has limitations in providing detailed tissue characterization, ultrasound Nakagami imaging holds promise for visualizing and quantifying tissue scattering in backscattered signals, with potential applications in fat fraction analysis. However, existing methods for Nakagami imaging struggle with optimal window size selection and suffer from estimator instability, leading to degraded image resolution. To address these challenges, we propose a novel method called UNICORN (Ultrasound Nakagami Imaging via Score Matching and Adaptation), which offers an accurate, closed-form estimator for Nakagami parameter estimation based on the score function of the ultrasound envelope signal. Unlike methods that visualize only specific regions of interest (ROI) and estimate parameters within fixed window sizes, our approach provides comprehensive parameter mapping by providing a pixel-by-pixel estimator, resulting in high-resolution imaging. We demonstrated that our proposed estimator effectively assesses hepatic steatosis and provides visual distinction in the backscattered statistics associated with this condition. Through extensive experiments using real envelope data from patient, we validated that UNICORN enables clinical detection of hepatic steatosis and exhibits robustness and generalizability.
Figures & tables
| MRI-PDFF | ||||
| Dataset | (Normal) | - (Mild) | (Severe) | Total |
| SMC | 224 | 170 | 55 | 449 |
| Dataset | MNIST | BUSI | ||||
| Metric | WS | PSNR (dB) | RMSE | WS | PSNR (dB) | RMSE |
| Measurement | - | 9.17 | 0.695 | - | 10.14 | 0.651 |
| Momentum | 9 | 21.01 | 0.179 | 9 | 18.58 | 0.230 |
| Momentum | 11 | 21.28 | 0.173 | 13 | 21.22 | 0.174 |
| Momentum | 13 | 20.78 | 0.184 | 25 | 23.31 | 0.132 |
| MLE | 11 | 20.45 | 0.191 | 25 | 22.65 | 0.148 |
| Method | SMC | |
| PCC(r) | SS | |
| Momentum | 0.53 | |
| MLE | 0.65 | |
| WMC | 0.54 | |
| UNICORN(Ours) | 0.77 | |
| AUC | Sensitivity | Specificity | PPV | NPV | F1 | |
| (A) Normal vs Mild | 0.87 | |||||
| (a) Optimal | 0.73 | 0.87 | 0.81 | 0.80 | 0.77 | |
| (b) 90% Specificity | 0.68 | 0.90 | 0.85 | 0.78 | 0.76 | |
| (c) 90% Sensitivity | 0.90 | 0.57 | 0.62 | 0.88 | 0.74 | |
| (B) Mild vs Severe | 0.80 | |||||
| (a) Optimal | 0.83 | 0.62 | 0.42 | 0.92 | 0.55 |
| Method | AUC | Sensitivity | Specificity | PPV | NPV | F1 | |
| (A) Normal vs Mild | Momentum | 0.83 | 0.85 | 0.68 | 0.68 | 0.85 | 0.76 |
| MLE | 0.88 | 0.83 | 0.78 | 0.75 | 0.85 | 0.79 | |
| WMC | 0.82 | 0.88 | 0.65 | 0.68 | 0.87 | 0.76 | |
| UNICORN | 0.87 | 0.73 | 0.87 | 0.81 | 0.80 | 0.77 | |
| (B) Mild vs Severe | Momentum | 0.68 | 0.70 | 0.60 | 0.36 | 0.86 | 0.47 |
| MLE | 0.74 | 0.68 | 0.67 | 0.41 | 0.87 | 0.51 |
| Method | Momentum | MLE | WMC | UNICORN |
| Train Memory (GB) | - | - | - | 13.68 |
| Duration (sec/patient) | 0.084 0.003 | 33.79 2.451 | 0.131 0.005 | 0.004 0.001 |
Ultrasound Imaging
Ultrasound imaging uses sound waves to visualize anatomy and function, with AI improving image reconstruction, diagnosis, measurement, and scan guidance.
See all 150 papers