Topology and Intersection-Union Constrained Loss Function for Multi-Region Anatomical Segmentation in Ocular Images

Ruiyu Xia,Jianqiang Li,Xi Xu,Guanghui Fu
2024-11-01
Abstract:Ocular Myasthenia Gravis (OMG) is a rare and challenging disease to detect in its early stages, but symptoms often first appear in the eye muscles, such as drooping eyelids and double vision. Ocular images can be used for early diagnosis by segmenting different regions, such as the sclera, iris, and pupil, which allows for the calculation of area ratios to support accurate medical assessments. However, no publicly available dataset and tools currently exist for this purpose. To address this, we propose a new topology and intersection-union constrained loss function (TIU loss) that improves performance using small training datasets. We conducted experiments on a public dataset consisting of 55 subjects and 2,197 images. Our proposed method outperformed two widely used loss functions across three deep learning networks, achieving a mean Dice score of 83.12% [82.47%, 83.81%] with a 95% bootstrap confidence interval. In a low-percentage training scenario (10% of the training data), our approach showed an 8.32% improvement in Dice score compared to the baseline. Additionally, we evaluated the method in a clinical setting with 47 subjects and 501 images, achieving a Dice score of 64.44% [63.22%, 65.62%]. We did observe some bias when applying the model in clinical settings. These results demonstrate that the proposed method is accurate, and our code along with the trained model is publicly available.
Computer Vision and Pattern Recognition,Image and Video Processing
What problem does this paper attempt to address?
The problems that this paper attempts to solve are as follows: When performing multi - region anatomical segmentation in eye images, the existing methods perform poorly in small - data - set training and clinical applications. Specifically: 1. **Lack of public data sets and tools**: Currently, there are no publicly available data sets and tools that can be used for the early diagnosis of Ocular Myasthenia Gravis (OMG). 2. **Segmentation accuracy problems**: When segmenting different eye regions (such as the sclera, iris and pupil), the existing methods fail to fully consider the topological relationships and intersection - union relationships among these regions, resulting in inconsistent segmentation results. 3. **Challenges in small - data - set training**: Deep - learning models usually require a large amount of data for training. However, in practical applications, especially in the medical field, the cost of obtaining a large amount of labeled data is very high. Therefore, methods that can perform well on small data sets are required. To solve these problems, the author proposes a new loss function - **Topology and Intersection - Union Constrained Loss Function (TIU loss)**, aiming to improve the segmentation performance by introducing topological constraints and intersection - union constraints, especially in the case of small - data - set training. Specific objectives include: - **Improve segmentation accuracy**: By introducing topological and intersection - union constraints, ensure that the segmentation results conform to the anatomical logical relationships. - **Enhance the effect of small - data - set training**: Under limited training data, it is still possible to obtain relatively good segmentation performance. - **Be applicable to clinical data**: Verify the effectiveness of this method in actual clinical data, despite the problem of domain shift. Through experiments, the author demonstrates the superior performance of this method on public data sets and clinical data sets, and points out that future work will focus on further optimizing the effect of clinical applications.