Diffusion-Based Feature Denoising with NNMF for Robust handwritten digit multi-class classification
This work addresses robustness to noise and adversarial attacks for handwritten digit classification, but it is incremental as it adapts methods from prior work on brain tumor classification.
The paper tackled robust multi-class classification of handwritten digits by combining diffusion-based feature denoising with a hybrid representation, resulting in a model that outperformed CNN baselines in both baseline and adversarial settings using AutoAttack.
This work presents a robust multi-class classification framework for handwritten digits that combines diffusion-driven feature denoising with a hybrid feature representation. Inspired by our previous work on brain tumor classification, the proposed approach operates in a feature space to improve the robustness to noise and adversarial attacks. First, the input images are converted into tight, interpretable exemplification using Nonnegative Matrix Factorization (NNMF). In parallel, special deep features are extracted using a computational neural network (CNN). These integral features are combined into a united hybrid representation. To improve robustness, a step diffusion operation is used in the feature space by gradually adding Gaussian noise. A feature denoiser network is trained to reverse this operation and rebuild clean representations from tilted inputs. The courteous features are then applied for multi-class classification. The suggested method is evaluated in both baseline and adversarial settings using AutoAttack. The experimental outcome present that the diffusion-based hybrid model is both effective and robust, the CNN baseline models outperforming while maintain powerful classification performance. These results explain the activity of feature-level diffusion defense for reliable multi-class handwritten digit classification.