GraphMriNet: a few-shot brain tumor MRI image classification model based on Prewitt operator and graph isomorphic network

Abstract

Brain tumors are regarded as one of the most lethal forms of cancer, primarily due to their heterogeneity and low survival rates. To tackle the challenge posed by brain tumor diagnostic models, which typically require extensive data for training and are often confined to a single dataset, we propose a diagnostic model based on the Prewitt operator and a graph isomorphic network. Firstly, during the graph construction stage, edge information is extracted from MRI (magnetic resonance imaging) images using the Prewitt filtering algorithm. Pixel points with a gray value intensity greater than 128 are designated as graph nodes, while the remaining pixel points are treated as edges of the graph. Secondly, the graph data is inputted into the GIN model for training, with model parameters optimized to enhance performance. Compared with existing work using small sample sizes, the GraphMriNet model has achieved classification accuracies of 100%, 100%, 100%, and 99.68% on the BMIBTD, CE-MRI, BTC-MRI, and FSB open datasets, respectively. The diagnostic accuracy has improved by 0.8% to 5.3% compared to existing research. In a few-shot scenario, GraphMriNet can accurately diagnose various types of brain tumors, providing crucial clinical guidance to assist doctors in making correct medical decisions. Additionally, the source code is available at this link: https://github.com/keepgoingzhx/GraphMriNet.