arXiv:2409.10156cs.CVcs.LG2024-09

对比学习在古希腊纸草文字识别中效果不如传统方法。

Contrastive Learning for Character Detection in Ancient Greek Papyri

  • 用SimCLR进行无监督预训练,再微调至小数据集
  • 交叉熵基线模型性能优于对比学习与三元组模型
  • 图像裁剪策略可能破坏语义,影响模型效果

本论文研究SimCLR这一对比学习方法在古希腊字母识别中的有效性,重点考察不同数据增强策略的影响。我们使用Alpub数据集预训练SimCLR主干网络,并在较小的ICDAR数据集上进行微调,与采用交叉熵和三元组损失函数的传统基线模型进行对比。方法上比较三种模型:(1) 交叉熵基线模型,(2) 带分类层的三元组嵌入模型,(3) 带分类层的SimCLR预训练模型。初始阶段,在ResNet-18和ResNet-50上使用93种增强策略训练三类模型,通过t检验筛选出表现最优的前四种增强方式。随后在Alpub上预训练SimCLR,再于ICDAR上微调;三元组模型也基于前四类增强进行预训练后微调。实验表明,SimCLR未超越基线模型,交叉熵模型表现最佳。研究指出,尽管预训练数据量大,但SimCLR的裁剪策略可能导致输入图像语义偏移,降低训练效率。代码已开源。

原文摘要 · Abstract (English)

This thesis investigates the effectiveness of SimCLR, a contrastive learning technique, in Greek letter recognition, focusing on the impact of various augmentation techniques. We pretrain the SimCLR backbone using the Alpub dataset (pretraining dataset) and fine-tune it on a smaller ICDAR dataset (finetuning dataset) to compare SimCLR's performance against traditional baseline models, which use cross-entropy and triplet loss functions. Additionally, we explore the role of different data augmentation strategies, essential for the SimCLR training process. Methodologically, we examine three primary approaches: (1) a baseline model using cross-entropy loss, (2) a triplet embedding model with a classification layer, and (3) a SimCLR pretrained model with a classification layer. Initially, we train the baseline, triplet, and SimCLR models using 93 augmentations on ResNet-18 and ResNet-50 networks with the ICDAR dataset. From these, the top four augmentations are selected using a statistical t-test. Pretraining of SimCLR is conducted on the Alpub dataset, followed by fine-tuning on the ICDAR dataset. The triplet loss model undergoes a similar process, being pretrained on the top four augmentations before fine-tuning on ICDAR. Our experiments show that SimCLR does not outperform the baselines in letter recognition tasks. The baseline model with cross-entropy loss demonstrates better performance than both SimCLR and the triplet loss model. This study provides a detailed evaluation of contrastive learning for letter recognition, highlighting SimCLR's limitations while emphasizing the strengths of traditional supervised learning models in this task. We believe SimCLR's cropping strategies may cause a semantic shift in the input image, reducing training effectiveness despite the large pretraining dataset. Our code is available at https://github.com/DIVA-DIA/MT_augmentation_and_contrastive_learning/.

对比学习文字识别古希腊文献图像增强

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。