arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04073cs.CLcs.AIcs.DB

AnyMatch——使用小型语言模型的高效零样本实体匹配

AnyMatch -- Efficient Zero-Shot Entity Matching with a Small Language Model

University of Amsterdam(阿姆斯特丹大学);Amsterdam UMC(阿姆斯特丹大学医学中心);BIFOLD(柏林工业大学数据与智能基础研究所);TU Berlin(柏林工业大学)

Zeyu Zhang, Paul Groth, Iacer Calixto, Sebastian Schelter

AI总结:

针对零样本实体匹配中LLM成本高、吞吐低的问题,本文提出经迁移学习微调并采用困难样本筛选等数据选择策略的小型模型AnyMatch,在九项基准上以极低推理成本取得接近MatchGPT的效果。

摘要:

实体匹配(EM)是判断两条记录是否指向同一现实世界实体的问题,在数据集成中至关重要,例如用于产品目录或地址数据库。许多EM方法的一个主要缺点是依赖标注示例。因此,我们关注具有挑战性的零样本实体匹配设置,即对于未见过的目标数据集,没有任何标注示例可用。近年来,大型语言模型(LLM)在零样本EM中展现出良好效果,但其低吞吐量和高昂部署成本限制了适用性与可扩展性。我们通过AnyMatch重新审视零样本EM问题,它是一个在迁移学习设置下微调的小型语言模型。我们提出若干新颖的数据选择技术来为模型生成微调数据,例如通过AutoML过滤器选择难以匹配的记录对、生成额外的属性级示例,以及控制数据中的标签不平衡。我们在九个基准数据集上与十三个基线进行比较,对模型的预测质量和部署成本开展广泛评估。我们发现,尽管参数量很小,AnyMatch仍提供具有竞争力的预测质量:其总体F1分数达到第二高,并优于若干其他使用数千亿参数模型的方法。此外,我们的方法展现出显著成本优势:AnyMatch的平均预测质量与采用专有万亿参数模型GPT-4的最先进方法MatchGPT相差不超过4.4%,但AnyMatch所需参数少四个数量级,并且推理成本(以每1000个token的美元计)降低3899倍。

英文摘要:

Entity matching (EM) is the problem of determining whether two records refer to same real-world entity, which is crucial in data integration, e.g., for product catalogs or address databases. A major drawback of many EM approaches is their dependence on labelled examples. We thus focus on the challenging setting of zero-shot entity matching where no labelled examples are available for an unseen target dataset. Recently, large language models (LLMs) have shown promising results for zero-shot EM, but their low throughput and high deployment cost limit their applicability and scalability. We revisit the zero-shot EM problem with AnyMatch, a small language model fine-tuned in a transfer learning setup. We propose several novel data selection techniques to generate fine-tuning data for our model, e.g., by selecting difficult pairs to match via an AutoML filter, by generating additional attribute-level examples, and by controlling label imbalance in the data. We conduct an extensive evaluation of the prediction quality and deployment cost of our model, in a comparison to thirteen baselines on nine benchmark datasets. We find that AnyMatch provides competitive prediction quality despite its small parameter size: it achieves the second-highest F1 score overall, and outperforms several other approaches that employ models with hundreds of billions of parameters. Furthermore, our approach exhibits major cost benefits: the average prediction quality of AnyMatch is within 4.4% of the state-of-the-art method MatchGPT with the proprietary trillion-parameter model GPT-4, yet AnyMatch requires four orders of magnitude less parameters and incurs a 3,899 times lower inference cost (in dollars per 1,000 tokens).

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

↑