2021-08-14 - TechRxiv .pdf (3.51 MB)
An Encoder-decoder Deep Learning Model Combining Mixed Attention Mechanism and Asymmetric Convolution for Automation of Retinal Vessels Segmentation
preprint
posted on 2021-09-01, 04:02 authored by Jiajia Cao, Qin Zhou, Yi ChenYi Chen, Lin Yin, Fei ZhangThe segmentation of the retinal vascular tree is the fundamental step for diagnosing ophthalmological diseases and cardiovascular diseases. Most existing vessel segmentation methods based on deep learning give the learned features equal importance. Ignored the highly imbalanced ratio between background and vessels (the majority of vessel pixels belong to the background), the learned features would be dominantly guided by background, and relatively little influence comes from vessels, often leading to low model sensitivity and prediction accuracy. The reduction of model size is also a challenge. We propose a mixed attention mechanism and asymmetric convolution encoder-decoder structure(MAAC) for segmentation in Retinal Vessels to solve these problems. In MAAC, the mixed attention is designed to emphasize the valid features and suppress the invalid features. It not only identifies information that helps retinal vessels recognition but also locates the position of the vessel. All square convolutions are replaced by asymmetric convolutions because it is more robust to rotational distortions and small convolutions are more suitable for extracting vessel features (based on the thin characteristics of vessels). The employment of asymmetric convolution reduces model parameters and improve the recognition of thin vessel. The experiments on public datasets DRIVE, STARE, and CHASE\_DB1 demonstrated that the proposed MAAC could more accurately segment vessels with a global AUC of 98.17$\%$, 98.67$\%$, and 98.53$\%$, respectively. The mixed attention proposed in this study can be applied to other deep learning models for performance improvement without changing the network architectures.
Funding
Ministry of Science and Technology of China (Grant No. 2018AAA0101300)
Guangdong Regular Institutions of characteristic innovation project (No.2017KTSCX176)
DGUT innovation center of robotics and intelligent equipment of China (No.KCYCXPT2017006)
KEY Laboratory of Robotics and Intelligent Equipment of Guangdong Regular Institutions of Higher Education (No.2017KSYS009)
National Natural Science Foundation of China (No. 62073091)
Key fields (new generation information technology) special projects of universities in Guangdong Province (No.2020ZDZX3042)
Open Fund of Hunan Provincial Key Laboratory of Mechanical Equipment Health Maintenance(No.21903)
Royal Society International Exchanges Scheme (IEC-NSFC-201029)
History
Email Address of Submitting Author
leo.chen@newcastle.ac.ukORCID of Submitting Author
0000-0001-7960-8374Submitting Author's Institution
Newcastle UniversitySubmitting Author's Country
- United Kingdom