Application of Vision-Series Transformer in Screening for Coronary Heart Diseases Using Coronary CT Angiography.

Kunlun Wang,Hanyang Meng,Xingjun Wang
DOI: https://doi.org/10.1145/3603781.3603858
2023-01-01
Abstract:With the development of computer vision, more and more researches support the utilization of machine learning (ML) in the diagnosis of coronary heart disease (CHD) using CT angiography. Artificial intelligence shows strength in assisting researchers and clinicians in CT scan information reading. CTA data provides a series of scans of images from patients and thus could be reconstructed as three dimensions (3-D) images, also known as volume rendering images. We use the projection of those 3-D images as a dataset and developed a deep neural network with Transformer to diagnose coronary heart diseases automatically. Compared with several traditional and classic deep learning methods, i.e. simple CNN, VGGNet, and ResNet which show accuracy of 52.5%, 57.82%, and 76.3%, respectively, vision transformer(ViT) improves it to 81.5%. Besides, we take great advantage of the vision transformer and series of CTA. Based on that, we developed a new deep learning model Vision-Series Transformer(ViST). It has better performance in accuracy(83.78%). 5-fold cross-validation was used to determine whether it performs better in other statistical indicators. It shows the same high sensitivity and specificity as ViT, better than that of other models. The results show that both ViT and ViST could be used in screening for CHD, while ViST has better statistical indicators.
What problem does this paper attempt to address?