arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.02201cs.LGcs.CL
收藏

评估大型语言模型在SDG映射中的性能(技术报告)

Evaluating the Performance of Large Language Models for SDG Mapping (Technical Report)

Swinburne University of Technology(斯威本科技大学)

Hui Yin, Amir Aryani, Nakul Nambiar

AI总结:

本文以GPT-4o为基线,系统评估Mixtral、LLaMA系列、Gemma、Qwen2等模型在多标签SDG映射任务中的表现,发现LLaMA 2与Gemma改进空间较大,其余模型性能相近。

摘要:

大型语言模型(LLM)的使用正在迅速扩展,开源版本也不断出现,为用户提供了更安全、更具适应性的选择。这些模型使用户无需向第三方提供数据即可保护数据隐私,并且可以针对特定任务进行定制。在本研究中,我们以GPT-4o的输出为基线,比较了多种语言模型在可持续发展目标(Sustainable Development Goal,SDG)映射任务上的性能。所选用于比较的开源模型包括Mixtral、LLaMA 2、LLaMA 3、Gemma和Qwen2。此外,研究还纳入了GPT-4o的更专用版本GPT-4o-mini,以扩展比较范围。鉴于SDG映射任务具有多标签性质,我们采用F1分数、精确率和召回率等指标,并进行微平均,以评估模型性能的不同方面。这些指标由混淆矩阵得出,以确保全面评估。我们通过绘制不同阈值下基于F1分数、精确率和召回率的曲线,对每个模型的性能给出清晰的观察和分析。实验结果表明,LLaMA 2和Gemma仍有显著的改进空间。其他四个模型在性能上没有表现出特别大的差异。全部七个模型的输出均可在Zenodo获取:https://doi.org/10.5281/zenodo.12789375。

英文摘要:

The use of large language models (LLMs) is expanding rapidly, and open-source versions are becoming available, offering users safer and more adaptable options. These models enable users to protect data privacy by eliminating the need to provide data to third parties and can be customized for specific tasks. In this study, we compare the performance of various language models on the Sustainable Development Goal (SDG) mapping task, using the output of GPT-4o as the baseline. The selected open-source models for comparison include Mixtral, LLaMA 2, LLaMA 3, Gemma, and Qwen2. Additionally, GPT-4o-mini, a more specialized version of GPT-4o, was included to extend the comparison. Given the multi-label nature of the SDG mapping task, we employed metrics such as F1 score, precision, and recall with micro-averaging to evaluate different aspects of the models' performance. These metrics are derived from the confusion matrix to ensure a comprehensive evaluation. We provide a clear observation and analysis of each model's performance by plotting curves based on F1 score, precision, and recall at different thresholds. According to the results of this experiment, LLaMA 2 and Gemma still have significant room for improvement. The other four models do not exhibit particularly large differences in performance. The outputs from all seven models are available on Zenodo: https://doi.org/10.5281/zenodo.12789375.

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

↑