A Spatially Constrained Density-Weighted Clustering Method for Brain Tumor Segmentation in MRI Images

Lakshmi Pravallika R; Pradeep Kumar Reddy R

doi:10.54392/irjmt25325

Authors

Lakshmi Pravallika R Department of Computer Science Engineering, Y.S.R. Engineering College of YVU, Andhra Pradesh, India. Author
Pradeep Kumar Reddy R Department of Computer Science Engineering, Y.S.R. Engineering College of YVU, Andhra Pradesh, India Author

DOI:

https://doi.org/10.54392/irjmt25325

Keywords:

Accuracy, Brain MRI, Tumor segmentation, K-Means Clustering, Iterative Refinement, Density-Based Weighting, Spatial constraints, Centroid Adjustment, MSE, DSC, JI

Abstract

The article proposes a modified K-means clustering algorithm to improve the accuracy of identifying tumors in brain MRI images, a crucial challenge in diagnosing and treating neurological disorders. The algorithm incorporates iterative refinement with tolerance-based convergence, density-based weighting for centroid adjustment, and spatial constraints to enhance segmentation accuracy. The proposed method introduces new methods to improve the segmentation of brain MRI images. A process of repeating small improvements on cluster centroids is used, helping to achieve low intra-cluster variation and ensuring that the algorithm reaches a good calculation point without repeating unnecessary parts. An important aspect is the added density-based weighting which updates centroid positions based on the local distribution of data points. As a result, the algorithm works better with groups that are uneven in size or shape. In addition, spatial issues are covered so that clustering is influenced by where the pixels are and how bright they are, so as to hold anatomical structures. All these methods make it possible to exactly segment the different and complicated parts of the brain in MRI scans. Results from experiments on benchmark brain scans show that the proposed method gets better results compared to both traditional and advanced segmentation methods. The model shows an average increase of 6.41% in DSC, 10.52% in JI and 6.41% in F1 Score, accompanied by a 38.43% decline in MSE. These outcomes suggest that the algorithm is a promising tool for clinical use in brain MRI analysis.

References

A. Tamimi, I. Tamimi, M. Barbarawi, A.K. Qudah, F. Tamimi, (2025) Brain Concussion: Recent Advances, Outcomes, and Challenges. IntechOpen, https://doi.org/10.5772/intechopen.1005615

Y. Ma, Federated Learning for Brain Tumor Diagnosis: Methods, Challenges and Future Prospects. ITM Web of Conferences. 70, (2025) 9. https://doi.org/10.1051/itmconf/20257003028

F.J. Dorfner J.B. Patel, J. Kalpathy-Cramer, E.R. Gerstner, C.P. Bridge, A review of deep learning for brain tumor analysis in MRI. NPJ Precision Oncology, 9(1), (2025) 2. https://doi.org/10.1038/s41698-024-00789-2

M.K. Abd-Ellah, A.I. Awad, A.A. Khalaf, H.F. Hamed, A review on brain tumor diagnosis from MRI images: Practical implications, key achievements, and lessons learned. Magnetic resonance imaging, 61, (2019) 300–318. https://doi.org/10.1016/j.mri.2019.05.028

B. Ramesh, S.Kiran, Image Segmentation in the Framework of Deep Learning a Comprehensive Survey. International Journal of Computer Sciences and Engineering, 9(4), (2021) 35–40. https://doi.org/10.26438/ijcse/v9i4.3540

H. Chetty, M. Shah, S. Kabaria, S. Verma, (2017) A Survey on brain tumor extraction approach from mri images using image processing, In 2nd International Conference for Convergence in Technology (I2CT), IEEE, India, 534–538. https://doi.org/10.1109/I2CT.2017.8226187

M.K. Islam, M.S. Ali, M.S. Miah, M.M. Rahman, M.S. Alam, M.A. Hossain, Brain tumor detection in MR image using superpixels, principal component analysis and template based K-means clustering algorithm, Machine Learning with applications, 5(15), (2021) 100044. https://doi.org/10.1016/j.mlwa.2021.100044

A. Ashabi, S.B. Sahibuddin, M. Salkhordeh Haghighi, (2020) The Systematic Review of K-Means Clustering Algorithm. The 9th International Conference on Networks, Communication and Computing, Association for computing machinery, 13–18. https://doi.org/10.1145/3447654.3447657

D.M. Kumar, D. Satyanarayana, M.N.G. Prasad, An improved Gabor wavelet transform and rough K-means clustering algorithm for MRI brain tumor image segmentation. Multimedia Tools and Applications, 80(5), (2021) 6939–6957. https://doi.org/10.1007/s11042-020-09635-6

M. Joshi, B.K. Singh, (2024) Deep Learning Techniques for Brain Lesion Classification Using Various MRI (from 2010 to 2022): Review and Challenges. Med informatics. https://orcid.org/0009-0006-4761-7800

L. Nanni, B. Bonato, A. Bertoldo, (2025) Advancing Precision: A Comprehensive Review of MRI Segmentation Datasets from BraTS Challenges (2012–2024). https://doi.org/10.20944/preprints202501.0255.v1

S. Murali, H. Ding, F. Adedeji, C. Qin, J. Obungoloch, I. Asllani, U. Anazodo, N.A. Ntusi, R. Mammen, T. Niendorf, S. Adeleke, Bringing MRI to low‐and middle‐income countries: directions, challenges and potential solutions. NMR in Biomedicine, 37(7), (2024) e4992. https://doi.org/10.1002/nbm.4992

B. Ramesh Babu, S. Kiran, Medical Image Segmentation Based On U-Net: A Review of Theory and Applications.

M. Lather, P. Singh, Investigating Brain Tumor Segmentation and Detection Techniques. Procedia Computer Science, 167, (2020) 121–130. https://doi.org/10.1016/j.procs.2020.03.189

R. Ranjbarzadeh, A. Caputo, E.B. Tirkolaee, S. Jafarzadeh Ghoushchi, M. Bendechache, Brain tumor segmentation of MRI images: A comprehensive review on the application of artificial intelligence tools. Computers in biology and medicine, 152, (2023) 106405. https://doi.org/10.1016/j.compbiomed.2022.106405

B.R. Babu, S. Kiran, Advancing Seismic Image Segmentation: UNET++ with GLCM Integration. Journal of Electrical Systems, 20(3), 2024.

B. Ramesh Babu, S. Kiran, Enhancing Seismic Image Segmentation Using Deep Learning Methods, International Journal of Intelligent Systems and Applications in Engineering, 12(3), (2024) 4331-4340.

W. Wu, D. Li, J. Du, X. Gao, W. Gu, F. Zhao, X. Feng, H. Yan, An Intelligent Diagnosis Method of Brain MRI Tumor Segmentation Using Deep Convolutional Neural Network and SVM Algorithm. Computational and mathematical methods in medicine, 2020 (1), 2020 6789306. https://doi.org/10.1155/2020/6789306

B. Lee, N. Yamanakkanavar, J.Y. Choi, Automatic segmentation of brain MRI using a novel patch-wise U-net deep architecture. PLoS One, 15(8), (2020) e0236493. https://doi.org/10.1371/journal.pone.0236493

P. Chatterjee, K.D. Sharma, A. Chakrabarti, A stochastic approach for automated brain MRI segmentation. IET Image Processing, 15(3), (2021) 735–745. https://doi.org/10.1049/ipr2.12058

L. Hua, Y. Gu, X. Gu, J. Xue, T. Ni, A Novel Brain MRI Image Segmentation Method Using an Improved Multi-View Fuzzy c-Means Clustering Algorithm. Frontiers in Neuroscience. 15, (2021) 662674. https://doi.org/10.3389/fnins.2021.662674

S.V. Aruna Kumar, E. Yaghoubi, H. Proença, A Fuzzy Consensus Clustering Algorithm for MRI Brain Tissue Segmentation, Applied Sciences, 12(15), (2022) 7385. https://doi.org/10.3390/app12157385

E.U. Haq, H. Jianjun, X. Huarong, K. Li, L. Weng, A Hybrid Approach Based on Deep CNN and Machine Learning Classifiers for the Tumor Segmentation and Classification in Brain MRI. Computational and mathematical methods in medicine, 2022(1), (2022) 6446680. https://doi.org/10.1155/2022/6446680

A.Z. Atiyah, K.H. Ali, Brain MRI Images Segmentation Based on U-Net Architecture. Iraqi Journal for Electrical and Electronic Engineering, 18(1), (2022) 21–27. https://doi.org/10.37917/ijeee.18.1.3

P.A. Thakur, P. Pudke, R. Das, L. Suman, P. Bansod, Brain Tumor Segmentation Using K-means Clustering Algorithm. International Journal for Research in Applied Science & Engineering Technology, 10(2), (2022)1–8. https://doi.org/10.22214/ijraset.2022.40112

S.R. Sharma, S. Alshathri, B. Singh, M. Kaur, R.R. Mostafa, W. El-Shafai, Hybrid Multilevel Thresholding Image Segmentation Approach for Brain MRI. Diagnostics, 13(5), (2023) 925. https://doi.org/10.3390/diagnostics13050925

K.M. Mohamed Sudheer, Image Segmentation-A Novel Doping of Feature Extraction and Pixel Level Abstraction of MRI Medical Images, International Journal of Science and Research, 12(6), ( 2023) 1212–1218.

A.K. Mandle, S.P. Sahu, G. Gupta, Brain Tumor Segmentation and Classification in MRI using clustering and Kernel-Based SVM. Biomedical and Pharmacology Journal, 15(2), (2022) 699–716 https://dx.doi.org/10.13005/bpj/2409

S. Saeed, F.A. Abbasi, K. Dahri, Z.A. Shaikh, M.A. Nizamani, M.A. Memon, Improved Hybrid K-Nearest Neighbours Techniques in Segmentation of Low-Grade Tumor and Cerebrospinal Fluid. The Asian Bulletin of Big Data Management, 4(2), (2024) 58-86. https://doi.org/10.62019/abbdm.v4i02.151

C. Reshma, T. Shirisha, Tumor Detection in Brain Using MRI Images. International Journal of Scientific Research in Engineering and Management, 07(08), (2023) 1-3. https://doi.org/10.55041/ijsrem25039

J.A. Hartigan, M.A. Wong, Algorithm AS 136: A K-Means Clustering Algorithm. Journal of the royal statistical society, 28(1), (1979) 100-108. https://doi.org/10.2307/2346830

S.M. Aqil Burney, H. Tariq, K-Means Cluster Analysis for Image Segmentation. International Journal of Computer Applications, 96(4), (2014) 1-8.

J. Sander, M. Ester, H.-P. Kriegel, X. Xu, Density-Based Clustering in Spatial Databases: The Algorithm GDBSCAN and Its Applications. Data mining and knowledge discovery, 2(2), (1998) 169–194. https://doi.org/10.1023/A:1009745219419

M. Luo, Y.F. Ma, H.J. Zhang, (2003) a spatial constrained K-means approach to image segmentation. In Fourth International Conference on Information, Communications and Signal Processing, 2003 and the Fourth Pacific Rim Conference on Multimedia. Proceedings of the 2003 Joint, IEEE, Singapore, 738–742. https://doi.org/10.1109/ICICS.2003.1292554

J. Cheng, (2017) Brain Tumor Dataset. figshare. https://figshare.com/articles/dataset/brain_tumor_dataset/1512427

D. Martin, C. Fowlkes, D. Tal, J. Malik, (2001) A database of human segmented natural images and its application to evaluating segmentation algorithms and measuring ecological statistics, in Proceedings Eighth IEEE International Conference on Computer Vision. ICCV 2001, IEEE Vancouver, Canada. https://doi.org/10.1109/ICCV.2001.937655

F.J. Estrada, A.D. Jepson, Quantitative Evaluation of a Novel Image Segmentation Algorithm. In 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR’05), IEEE, San Diego, CA, USA, 2, (2005) 1132–1139. https://doi.org/10.1109/CVPR.2005.284

Y. Jiang, B. Cukic, Y. Ma, Techniques for evaluating fault prediction models. Empirical Software Engineering, 13(5), (2008) 561–595. https://doi.org/10.1007/s10664-008-9079-3

L.R. Dice, Measures of the Amount of Ecologic Association between Species. Ecology, 26(3), (1945) 297–302. https://doi.org/10.2307/1932409

P. Jaccard, The Distribution of the Flora in the Alpine Zone.1. New Phytologist, 11(2), (1912) 37–50. https://doi.org/10.1111/j.1469-8137.1912.tb05611.x

M. Hossin, M.N. Sulaiman, a Review on Evaluation Metrics for Data Classification Evaluations. International journal of data mining & knowledge management process, 5(2), (2015) 01–11. https://doi.org/10.5121/ijdkp.2015.5201

D. Harvey, S. Leybourne, P. Newbold, Testing the equality of prediction mean squared errors, International Journal of forecasting, 13(2), (1997) 281–291. https://doi.org/10.1016/S0169-2070(96)00719-4

S.B. Dhumal, M. Tamboli, (2022) Segmentation of Brain Tumor Using Different Clustering Algorithms. International Journal of Creative Research Thoughts (IJCRT), 10(5), 823- 828.

M. Sangeetha, P. Keerthika, K. Devendran, S. Sridhar, S.S. Raagav, T. Vigneshwar, Brain tumor segmentation and prediction on MRI images using deep learning network. International journal of health sciences, 6(S2), (2022) 13486–13503. https://doi.org/10.53730/ijhs.v6nS2.8542

P.G. Brindha, M. Kavinraj, P. Manivasakam, P. Prasanth, Brain tumor detection from MRI images using deep learning techniques. IOP conference series: materials science and engineering, 1055(1) (2021) 012115. https://doi.org/10.1088/1757-899X/1055/1/012115

M.I. Khaleel, M.A. Mohammed, M. Qays, Dominant Gray Level-based Genetic K-means Clustering Algorithm for MRI Image Segmentation. Engineering, Technology & Applied Science Research 14(3), (2024) 14355–14360. https://doi.org/10.48084/etasr.7125