An edge detection-based deep learning approach for tear meniscus height measurement

Kesheng Wang,Kunhui Xu,Xiaoyu Chen,Chunlei He,Jianfeng Zhang,Dexing Kong,Qi Dai,Shoujun Huang

2024-03-23

Abstract:Automatic measurements of tear meniscus height (TMH) have been achieved by using deep learning techniques; however, annotation is significantly influenced by subjective factors and is both time-consuming and labor-intensive. In this paper, we introduce an automatic TMH measurement technique based on edge detection-assisted annotation within a deep learning framework. This method generates mask labels less affected by subjective factors with enhanced efficiency compared to previous annotation approaches. For improved segmentation of the pupil and tear meniscus areas, the convolutional neural network Inceptionv3 was first implemented as an image quality assessment model, effectively identifying higher-quality images with an accuracy of 98.224%. Subsequently, by using the generated labels, various algorithms, including Unet, ResUnet, Deeplabv3+FcnResnet101, Deeplabv3+FcnResnet50, FcnResnet50, and FcnResnet101 were trained, with Unet demonstrating the best performance. Finally, Unet was used for automatic pupil and tear meniscus segmentation to locate the center of the pupil and calculate TMH,respectively. An evaluation of the mask quality predicted by Unet indicated a Mean Intersection over Union of 0.9362, a recall of 0.9261, a precision of 0.9423, and an F1-Score of 0.9326. Additionally, the TMH predicted by the model was assessed, with the fitting curve represented as y= 0.982x-0.862, an overall correlation coefficient of r^2=0.961 , and an accuracy of 94.80% (237/250). In summary, the algorithm can automatically screen images based on their quality,segment the pupil and tear meniscus areas, and automatically measure TMH. Measurement results using the AI algorithm demonstrate a high level of consistency with manual measurements, offering significant support to clinical doctors in diagnosing dry eye disease.

Image and Video Processing,Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

This paper primarily discusses an automatic measurement method for Tear Meniscus Height (TMH) in Dry Eye Disease (DED). Existing measurement methods rely on deep learning techniques, but the annotation process is highly subjective and time-consuming. This research proposes a deep learning framework based on edge detection-assisted annotation to improve efficiency and reduce the subjectivity of manual annotation.

Firstly, the paper introduces an image quality evaluation model based on the Inceptionv3 convolutional neural network to identify high-quality ocular surface images. Then, fewer subjective factors influenced labels are generated through edge detection methods for training different segmentation algorithms such as Unet, with Unet performing the best. Additionally, this method enables simultaneous segmentation of the pupil and tear meniscus regions, as well as the calculation of TMH.

The experimental results show that the mask quality predicted by Unet (Mean Intersection over Union, MIoU) reaches 0.9362, with a recall rate of 0.9261, precision of 0.9423, F1 score of 0.9326, and a correlation coefficient of 0.982 for the fitting curve of TMH measurement. The overall correlation coefficient is 0.961, with an accuracy of 94.80%. Compared to manual measurement by experts, the algorithm's measurement results are highly consistent, providing strong support for clinical diagnosis of dry eye disease.

In conclusion, the proposed method in this study can automatically select images, segment the pupil and tear meniscus regions, and accurately measure TMH, reducing the burden of manual annotation and improving the reliability and efficiency of measurement.