大模型可高效分类生物医学文献,零样本表现接近传统算法。
Large Language Models for Biomedical Article Classification
- 用提示工程与输出概率提取实现文本分类
- 零样本平均PR AUC达0.4,少样本近0.5
- 适合无标注数据场景下的医学文献分类
本研究系统深入地考察了大语言模型在生物医学文章分类中的应用价值。实验涵盖多个开源小中型模型及部分闭源模型,评估范围更全面:包括不同提示方式、输出处理策略(用于生成类别和类别概率预测)、少量示例数量与选择方法。最成功配置的性能与传统分类算法对比显示,零样本提示下15个挑战性数据集的平均PR AUC超过0.4,少样本提示下接近0.5,已接近朴素贝叶斯(0.5)、随机森林(默认0.5或调参后0.55)及微调Transformer模型(0.5)的表现。结果验证了大模型在非简单领域作为文本分类器的有效性,并提供了具有前景的实际配置建议,尤其推荐使用输出标记概率进行类别概率预测。
原文摘要 · Abstract (English)
This work presents a systematic and in-depth investigation of the utility of large language models as text classifiers for biomedical article classification. The study uses several small and mid-size open source models, as well as selected closed source ones, and is more comprehensive than most prior work with respect to the scope of evaluated configurations: different types of prompts, output processing methods for generating both class and class probability predictions, as well as few-shot example counts and selection methods. The performance of the most successful configurations is compared to that of conventional classification algorithms. The obtained average PR AUC over 15 challenging datasets above 0.4 for zero-shot prompting and nearly 0.5 for few-shot prompting comes close to that of the naïve Bayes classifier (0.5), the random forest algorithm (0.5 with default settings or 0.55 with hyperparameter tuning) and fine-tuned transformer models (0.5). These results confirm the utility of large language models as text classifiers for non-trivial domains and provide practical recommendations of the most promising setups, including in particular using output token probabilities for class probability prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。