针对眼病早期诊断,提出新损失函数提升小样本下眼部多区域分割精度。
Topology and Intersection-Union Constrained Loss Function for Multi-Region Anatomical Segmentation in Ocular Images
- 设计拓扑与交并集约束损失,增强模型对解剖结构的几何一致性理解。
- 在55人2197图数据上平均Dice达83.12%,小样本(10%)时提升8.32%。
- 适用于眼科影像分析,尤其适合训练数据少的医疗场景。
眼肌型重症肌无力(OMG)早期难以检测,但症状常先出现在眼肌,如眼睑下垂和复视。通过分割角膜、虹膜、瞳孔等区域并计算面积比,可辅助早期诊断。然而当前缺乏公开数据集与工具。为此,我们提出一种拓扑与交并集约束损失函数(TIU loss),在小样本下提升性能。在包含55名受试者、2197张图像的公开数据集上,该方法在三种深度学习网络中均优于两种常用损失函数,平均Dice分数为83.12% [82.47%, 83.81%](95%置信区间)。在仅用10%训练数据的低样本场景下,相比基线提升8.32%。临床验证中对47名受试者、501张图像测试,获得64.44% [63.22%, 65.62%]的Dice分数,但存在一定临床偏差。结果表明该方法准确有效,代码与模型已开源。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。