Enhancing medical image classification with generative AI using latent denoising diffusion probabilistic model and wiener filtering approach

Abstract

In the evolving landscape of medical diagnostics, a paradigm shift is being catalysed by the advent of generative artificial intelligence. Medical X-ray, CT and MRI images are essential diagnostic tools used by healthcare professionals to assess various musculoskeletal conditions. However, obtaining a sufficient number of medical images for training deep learning models can be challenging due to limited access to labelled data. Hence, the authors propose a novel Latent diffusion process for synthesizing medical images that closely resemble real patient images, aiming to address the challenge of limited access to labelled data in medical diagnostics. Leveraging deep learning and generative modelling techniques, this method synthesizes high-fidelity images that closely mimic real patient scans. By introducing noise and subsequently training the model to denoise, the approach captures intricate patterns inherent in authentic medical images. Among the four datasets utilized, the Diabetes Retinopathy dataset demonstrates superior performance, achieving the highest Mean Structural Similarity Index (MSSIM) score of 0.57 (compared to the dataset baseline of 0.62) and an accuracy of 93.75% when passed through the proposed pipeline. The Cataract dataset, registered a MSSIM score of 0.51 (versus the dataset baseline of 0.53) and an accuracy score of 97.52%, while the Knee OA dataset follows closely with MSSIM and accuracy scores of 0.65 (in contrast to the dataset baseline of 0.63) and 68.66% respectively. The results obtained are then compared with the results generated by the other state of the art models.