Steerable Pyramid Transform Enables Robust Left Ventricle Quantification
Xiangyang Zhu, Kede Ma, Wufeng Xue
TL;DR
This work addresses the vulnerability of deep learning–based left ventricle quantification to input perturbations in CMR. It introduces a fixed front-end steerable pyramid transform to provide orientation- and scale-aware representations aligned with LV anatomy, discarding the highpass residual to promote robust learning. The method combines SPT with a CNN-LSTM pipeline to predict eleven LV indices, with a regularized loss to encourage robust, diverse features. On Cardiac-DIG data, the approach achieves competitive accuracy while substantially improving robustness to spatial transformations, distortions, and adversarial attacks; ablation studies confirm the benefit of multi-scale/orientation sharing.
Abstract
Predicting cardiac indices has long been a focal point in the medical imaging community. While various deep learning models have demonstrated success in quantifying cardiac indices, they remain susceptible to mild input perturbations, e.g., spatial transformations, image distortions, and adversarial attacks. This vulnerability undermines confidence in using learning-based automated systems for diagnosing cardiovascular diseases. In this work, we describe a simple yet effective method to learn robust models for left ventricle (LV) quantification, encompassing cavity and myocardium areas, directional dimensions, and regional wall thicknesses. Our success hinges on employing the biologically inspired steerable pyramid transform (SPT) for fixed front-end processing, which offers three main benefits. First, the basis functions of SPT align with the anatomical structure of LV and the geometric features of the measured indices. Second, SPT facilitates weight sharing across different orientations as a form of parameter regularization and naturally captures the scale variations of LV. Third, the residual highpass subband can be conveniently discarded, promoting robust feature learning. Extensive experiments on the Cardiac-Dig benchmark show that our SPT-augmented model not only achieves reasonable prediction accuracy compared to state-of-the-art methods, but also exhibits significantly improved robustness against input perturbations.
