Diffusion-Based Feature Denoising and Using NNMF for Robust Brain Tumor Classification
- URL: http://arxiv.org/abs/2603.13182v1
- Date: Fri, 13 Mar 2026 17:15:46 GMT
- Title: Diffusion-Based Feature Denoising and Using NNMF for Robust Brain Tumor Classification
- Abstract summary: This study suggests a robust brain tumor classification framework that combines Non-Negative Matrix Factorization (NNMF), lightweight convolutional neural networks (CNNs), and diffusion-based feature purification.<n>The experimental results show that the proposed framework achieves competitive classification performance while significantly enhancing robustness against adversarial perturbations.
- Score: 0.0
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: 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.
Related papers
- SIINR: Structurally Informed Implicit Neural Representations for super-resolution with uncertainty quantification of clinical quality diffusion MRI datasets [3.1420911290508897]
We introduce SIINR, a framework for super-resoltion of clinical dMRI datasets.<n>We validate framework on a diverse set of open-access dMRI datasets.
arXiv Detail & Related papers (2026-07-22T09:16:23Z) - Diffusion-Based Feature Denoising with NNMF for Robust handwritten digit multi-class classification [0.0]
This work presents a robust multi-class classification framework for handwritten digits that combines diffusion-driven feature denoising with a hybrid feature representation.<n>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.
arXiv Detail & Related papers (2026-03-31T15:57:18Z) - Optimized Weighted Voting System for Brain Tumor Classification Using MRI Images [0.0]
This paper presents a weighted ensemble learning approach that combines deep learning and traditional machine learning models to improve classification performance.<n>A weighted voting mechanism assigns higher influence to models with better individual accuracy, ensuring robust decision-making.<n> Experimental evaluations on the Figshare and Kaggle MRI datasets demonstrate that the proposed method achieves state-of-the-art accuracy.
arXiv Detail & Related papers (2026-03-30T12:24:54Z) - Adapting HFMCA to Graph Data: Self-Supervised Learning for Generalizable fMRI Representations [57.054499278843856]
Functional magnetic resonance imaging (fMRI) analysis faces significant challenges due to limited dataset sizes and domain variability between studies.<n>Traditional self-supervised learning methods inspired by computer vision often rely on positive and negative sample pairs.<n>We propose adapting a recently developed Hierarchical Functional Maximal Correlation Algorithm (HFMCA) to graph-structured fMRI data.
arXiv Detail & Related papers (2025-10-05T12:35:01Z) - A Confounding Factors-Inhibition Adversarial Learning Framework for Multi-site fMRI Mental Disorder Identification [22.749962294853187]
We propose a novel multi-site adversarial learning network (MSalNET) for fMRI-based mental disorder detection.<n>The proposed method was evaluated on two multi-site fMRI datasets, i.e., Autism Brain Imaging Data Exchange (ABIDE) and ADHD-200.
arXiv Detail & Related papers (2025-04-12T10:58:19Z) - Revisiting Medical Image Retrieval via Knowledge Consolidation [46.6989555659494]
We propose a novel method to consolidate knowledge of hierarchical features and functions.<n>We introduce Depth-aware Representation Fusion (DaRF) and Structure-aware Contrastive Hashing (SCH)<n>Our method achieves a 5.6-38.9% improvement in mean Average Precision on the anatomical radiology dataset.
arXiv Detail & Related papers (2025-03-12T13:16:42Z) - Spatial-Temporal DAG Convolutional Networks for End-to-End Joint
Effective Connectivity Learning and Resting-State fMRI Classification [42.82118108887965]
Building comprehensive brain connectomes has proved to be fundamental importance in resting-state fMRI (rs-fMRI) analysis.
We model the brain network as a directed acyclic graph (DAG) to discover direct causal connections between brain regions.
We propose Spatial-Temporal DAG Convolutional Network (ST-DAGCN) to jointly infer effective connectivity and classify rs-fMRI time series.
arXiv Detail & Related papers (2023-12-16T04:31:51Z) - Brain Imaging-to-Graph Generation using Adversarial Hierarchical Diffusion Models for MCI Causality Analysis [44.45598796591008]
Brain imaging-to-graph generation (BIGG) framework is proposed to map functional magnetic resonance imaging (fMRI) into effective connectivity for mild cognitive impairment analysis.
The hierarchical transformers in the generator are designed to estimate the noise at multiple scales.
Evaluations of the ADNI dataset demonstrate the feasibility and efficacy of the proposed model.
arXiv Detail & Related papers (2023-05-18T06:54:56Z) - A Novel Solution of an Elastic Net Regularization for Dementia Knowledge
Discovery using Deep Learning [10.839925851203786]
This paper aims to increase the accuracy and reduce the processing time of classification through Deep Learning Architecture.
The proposed system consists of Convolutional Neural Network (CNN) to enhance the accuracy of classification and prediction.
In addition to that, the proposed method has improved the classification accuracy by 5% on average and reduced the processing time by 30 40 seconds on average.
arXiv Detail & Related papers (2021-08-21T11:53:25Z) - Computational Intelligence Approach to Improve the Classification
Accuracy of Brain Neoplasm in MRI Data [8.980876474818153]
This report presents two improvements for brain neoplasm detection in MRI data.
An advanced preprocessing technique is proposed to improve the area of interest in MRI data.
A hybrid technique using CNN for feature extraction followed by Support Vector Machine (SVM) for classification is also proposed.
arXiv Detail & Related papers (2021-01-24T06:45:26Z) - Improved Slice-wise Tumour Detection in Brain MRIs by Computing
Dissimilarities between Latent Representations [68.8204255655161]
Anomaly detection for Magnetic Resonance Images (MRIs) can be solved with unsupervised methods.
We have proposed a slice-wise semi-supervised method for tumour detection based on the computation of a dissimilarity function in the latent space of a Variational AutoEncoder.
We show that by training the models on higher resolution images and by improving the quality of the reconstructions, we obtain results which are comparable with different baselines.
arXiv Detail & Related papers (2020-07-24T14:02:09Z) - Rectified Meta-Learning from Noisy Labels for Robust Image-based Plant
Disease Diagnosis [64.82680813427054]
Plant diseases serve as one of main threats to food security and crop production.
One popular approach is to transform this problem as a leaf image classification task, which can be addressed by the powerful convolutional neural networks (CNNs)
We propose a novel framework that incorporates rectified meta-learning module into common CNN paradigm to train a noise-robust deep network without using extra supervision information.
arXiv Detail & Related papers (2020-03-17T09:51:30Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.