arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2305.15154cs.CV

用于OCT生物标志物分类的临床标注对比学习

Clinically Labeled Contrastive Learning for OCT Biomarker Classification

Georgia Institute of Technology(佐治亚理工学院);Retina Consultants of Texas(德克萨斯视网膜顾问公司)

Kiran Kokilepersaud, Stephanie Trejo Corona, Mohit Prabhushankar, Ghassan AlRegib, Charles Wykoff

AI总结:

本文提出一种利用临床标签作为伪标签的医学图像对比学习策略,通过有监督对比损失训练骨干网络并用少量生物标志物数据微调,在OCT生物标志物分类中实现最高5%的AUROC提升。

摘要:

本文提出了一种基于可从临床数据中提取标签的医学图像对比学习正负样本集选择新策略。在医学领域,存在多种数据标签,在诊断和治疗过程的不同阶段发挥不同作用,临床标签和生物标志物标签即为两例。通常,临床标签更容易大量获取,因为它们在常规临床护理中定期收集;而生物标志物标签则需要专家分析和解读才能获得。在眼科学领域,先前的工作表明,临床数值与光学相干断层扫描(OCT)中显现的生物标志物结构存在相关性。我们利用这一关系,将临床数据作为无生物标志物标签数据的伪标签,以选择正负实例,从而通过有监督对比损失训练骨干网络。通过这种方式,骨干网络学习到与可用临床数据分布相一致的表示空间。随后,我们使用少量带生物标志物标签的数据,通过交叉熵损失对以此方式训练的网络进行微调,以便直接从OCT扫描中分类这些关键疾病指标。我们还对此概念进行了扩展,提出一种使用临床对比损失线性组合的方法。我们在具有不同粒度生物标志物的新颖设置中,将我们的方法与最先进的自监督方法进行了基准比较。结果表明,在总生物标志物检测AUROC方面,性能提升最高达5%。

英文摘要:

This paper presents a novel positive and negative set selection strategy for contrastive learning of medical images based on labels that can be extracted from clinical data. In the medical field, there exists a variety of labels for data that serve different purposes at different stages of a diagnostic and treatment process. Clinical labels and biomarker labels are two examples. In general, clinical labels are easier to obtain in larger quantities because they are regularly collected during routine clinical care, while biomarker labels require expert analysis and interpretation to obtain. Within the field of ophthalmology, previous work has shown that clinical values exhibit correlations with biomarker structures that manifest within optical coherence tomography (OCT) scans. We exploit this relationship by using the clinical data as pseudo-labels for our data without biomarker labels in order to choose positive and negative instances for training a backbone network with a supervised contrastive loss. In this way, a backbone network learns a representation space that aligns with the clinical data distribution available. Afterwards, we fine-tune the network trained in this manner with the smaller amount of biomarker labeled data with a cross-entropy loss in order to classify these key indicators of disease directly from OCT scans. We also expand on this concept by proposing a method that uses a linear combination of clinical contrastive losses. We benchmark our methods against state of the art self-supervised methods in a novel setting with biomarkers of varying granularity. We show performance improvements by as much as 5\% in total biomarker detection AUROC.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑