Robust medical image classification with curvature regularization on the PATHMNIST

Guohao Yang; Jiacheng Qi; Xubin Sun

doi:10.1117/12.3044867

9 October 2024 Robust medical image classification with curvature regularization on the PATHMNIST

Guohao Yang, Jiacheng Qi, Xubin Sun

Author Affiliations +

Proceedings Volume 13288, Fourth International Conference on Computer Graphics, Image, and Virtualization (ICCGIV 2024); 132880W (2024) https://doi.org/10.1117/12.3044867
Event: Fourth International Conference on Computer Graphics, Image, and Virtualization (ICCGIV 2024), 2024, Chengdu, China

Abstract

In medical image analysis, the robustness and accuracy of classification models are paramount, especially under adverse conditions such as data scarcity, class imbalance, and potential adversarial attacks. This paper introduces a novel deep learning architecture employing curvature regularization (CURE) to enhance the robustness of medical image classification on the PATHMNIST dataset. Our approach integrates advanced convolutional techniques including depthwise and dilated convolutions with an attention mechanism, focusing on precise and detailed feature extraction essential for medical diagnostics. We incorporate curvature regularization to stabilize the learning process by controlling the magnitude of the Hessian's eigenvalues, making the model less sensitive to input variations. The effectiveness of our architecture is demonstrated through rigorous testing, including adversarial scenarios using the Fast Gradient Sign Method (FGSM) and Projected Gradient Descent (PGD). Results show our model not only achieves superior accuracy compared to traditional architectures like ResNet-18 and ResNet-50 but also maintains higher resilience against adversarial attacks. This robustness is critical for practical deployment in medical image analysis.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Guohao Yang, Jiacheng Qi, and Xubin Sun "Robust medical image classification with curvature regularization on the PATHMNIST", Proc. SPIE 13288, Fourth International Conference on Computer Graphics, Image, and Virtualization (ICCGIV 2024), 132880W (9 October 2024); https://doi.org/10.1117/12.3044867

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE