用非负矩阵分解与扩散去噪提升脑肿瘤分类的准确性与抗干扰能力
Diffusion-Based Feature Denoising and Using NNMF for Robust Brain Tumor Classification
- 通过NNMF提取可解释的医学图像特征,结合轻量CNN分类
- 在对抗攻击下仍保持高准确率,干净数据与鲁棒性均优于基准
- 适合对模型可解释性与安全性要求高的医疗影像应用
基于磁共振成像(MRI)的脑肿瘤分类在计算机辅助诊断中具有关键作用。近年来深度学习模型虽取得高分类精度,但对对抗扰动敏感,影响其在医疗场景中的可靠性。本文提出一种融合非负矩阵分解(NNMF)、轻量卷积神经网络(CNN)和基于扩散的特征净化机制的鲁棒分类框架。首先将MRI图像预处理为非负数据矩阵,从中提取紧凑且可解释的NNMF特征表示;利用AUC、Cohen's d和p值等统计指标筛选最具区分性的特征成分;随后在选定特征组上训练轻量CNN分类器。为进一步提升对抗鲁棒性,引入前向加噪与可学习去噪网络构成的特征空间净化模块。系统性能通过干净准确率及在AutoAttack生成的强大对抗攻击下的鲁棒准确率评估。实验结果表明,该框架在保持竞争力分类性能的同时,显著增强对对抗扰动的抵抗力。研究证明,结合可解释的NNMF表示、轻量深度模型与扩散防御技术,能有效应对医疗图像分类中的对抗威胁。
原文摘要 · Abstract (English)
Brain tumor classification from magnetic resonance imaging, which is also known as MRI, plays a sensitive role in computer-assisted diagnosis systems. In recent years, deep learning models have achieved high classification accuracy. However, their sensitivity to adversarial perturbations has become an important reliability concern in medical applications. This study suggests a robust brain tumor classification framework that combines Non-Negative Matrix Factorization (NNMF or NMF), lightweight convolutional neural networks (CNNs), and diffusion-based feature purification. Initially, MRI images are preprocessed and converted into a non-negative data matrix, from which compact and interpretable NNMF feature representations are extracted. Statistical metrics, including AUC, Cohen's d, and p-values, are used to rank and choose the most discriminative components. Then, a lightweight CNN classifier is trained directly on the selected feature groups. To improve adversarial robustness, a diffusion-based feature-space purification module is introduced. A forward noise method followed by a learned denoiser network is used before classification. System performance is estimated using both clean accuracy and robust accuracy under powerful adversarial attacks created by AutoAttack. The experimental results show that the proposed framework achieves competitive classification performance while significantly enhancing robustness against adversarial perturbations.The findings presuppose that combining interpretable NNMF-based representations with a lightweight deep approach and diffusion-based defense technique supplies an effective and reliable solution for medical image classification under adversarial conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。