Postoperative corneal topography generation based on attention mechanism and Pix2Pix network
10.3980/j.issn.1672-5123.2023.6.24
- VernacularTitle:基于注意力机制和Pix2Pix网络的术后角膜地形图生成
- Author:
Guang-Hua ZHANG
1
,
2
,
3
;
Nan CHENG
1
,
2
,
3
;
Zhe ZHANG
1
,
2
,
3
;
Xiao-Na LI
1
,
2
,
3
;
Jing PAN
1
,
2
,
3
;
En-Hui LI
1
,
2
,
3
;
Wei-Yi CHEN
1
,
2
,
3
Author Information
1. Institute of Big Data Smart Diagnosis and Treatment Industry
2. Graphics and Imaging Laboratory, University of Girona, Girona 17004, Catalonia, Spain
3. Department of Intelligence and Automation, Taiyuan Normal University, Taiyuan 030619, Shanxi Province, China
- Publication Type:Journal Article
- Keywords:
Pix2Pix network;
generative adversarial network;
attention mechanism;
corneal topography;
deep learning
- From:
International Eye Science
2023;23(6):1001-1006
- CountryChina
- Language:Chinese
-
Abstract:
AIM:To explore the use of attention mechanism and Pix2Pix generative adversarial network to predict the postoperative corneal topography of age-related cataract patients undergone femtosecond laser arcuate keratotomy.METHODS:In this retrospective case series study, the 210 preoperative and postoperative corneal topographies from 87 age-related cataract patients(105 eyes)undergoing femtosecond laser arcuate keratotomy at Shanxi Eye Hospital between March 2018 and March 2020 were selected and divided into a training set(180)and a test set(30)for model training and testing. The peak signal-to-noise ratio(PSNR), structural similarity(SSIM)and Alpins astigmatism vector analysis were used to compare the accuracy of postoperative corneal topography prediction under different attention mechanisms.RESULTS:The model based on attention mechanism and Pix2Pix network can predict postoperative corneal topography, among which the model based on Self-Attention mechanism has the best prediction effect, with PSNR and SSIM reaching 16.048 and 0.7661, respectively. There were no statistically significant differences in the difference vector, difference vector axis position, surgically induced astigmatism, and correction index between real and generated corneal topography on the 3mm and 5mm rings(all P>0.05).CONCLUSION:Based on the Self-Attention mechanism and Pix2Pix network, the postoperative corneal topography can be well predicted, which can provide reference for the surgical planning and postoperative effects of ophthalmic clinicians.
