Denoising diffusion model for increased performance of detecting structural heart disease
Streiffer, C. D.; Levin, M. G.; Witschey, W. R.; Anyanwu, E. C.
Show abstract
Recent advancements in generative artificial intelligence have shown promise in producing realistic images from complex data distributions. We developed a denoising diffusion probabilistic model trained on the CheXchoNet dataset, encoding the joint distribution of demographic data and echocardiogram measurements. We generated a synthetic dataset skewed towards younger patients with a higher prevalence of structural left ventricle disease. A diagnostic deep learning model trained on the synthetic dataset performed comparably to one trained on real data producing an AUROC=0.75(95%CI 0.72-0.77), with similar performance on an internal dataset. Combining real data with positive samples from the synthetic data improved diagnostic accuracy producing an AUROC=0.80(95%CI 0.78-0.82). Subgroup analysis showed the largest performance improvement across younger patients. These results suggest diffusion models can increase diagnostic accuracy and fine-tune models for specific populations.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Uncovering the effects of model initialization on deep model generalization: A study with adult and pediatric chest X-ray images 96%
- Designing a computer-assisted diagnosis system for cardiomegaly detection and radiology report generation 95%
- A recurrent neural network and parallel hidden Markov model algorithm to segment and detect heart murmurs in phonocardiograms 95%
Similar papers in this journal
Similar papers in this journal
- BenchXAI: Comprehensive Benchmarking of Post-hoc Explainable AI Methods on Multi-Modal Biomedical Data 94%
- MultiHeadGAN: A Deep Learning Method for Low Contrast Retinal Pigment Epithelium Cells Segmentation in Fluorescent Flatmount Microscopy Images 94%
- Super-resolution Left Ventricular Flow and Pressure Mapping by Navier-Stokes-Informed Neural Networks 94%
Similar papers in this journal
- STAMP: Simultaneous Training and Model Pruning for Low Data Regimes in Medical Image Segmentation 93%
- Strain estimation in aortic roots from 4D echocardiographic images using medial modeling and deformable registration 93%
- Clinical Validation of Saliency Maps for Understanding Deep Neural Networks in Ophthalmology 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.