评估BERT类模型在问答任务中的可靠性,发现RoBERTa最稳定。
Assessing Reliability of BERT-Based Models on Question Answering Tasks

- 用蒙特卡洛丢弃和改写输入测试模型稳定性
- RoBERTa比ALBERT和DistilBERT更一致,且丢弃不影响推理
- 适合关注模型可信度的开发者与研究者
大型语言模型的可靠性与准确性同样关键,直接影响其在实际应用中的可信度与鲁棒性。本研究聚焦基于Transformer的问答模型(BERT及其变体:RoBERTa、ALBERT、DistilBERT),评估其在两种条件下的响应稳定性:(1) 通过蒙特卡洛丢弃(MCD)引入内部模型变化;(2) 通过句子改写施加输入扰动。基于SQuAD和QuAC数据集,分析不同丢弃率对预测一致性的影响,以及词汇变化对答案稳定性的影响。结果表明,RoBERTa表现最优,而ALBERT与DistilBERT存在显著不一致性。统计分析验证了在推理中启用MCD不会破坏推理动态,支持其作为可靠性度量的有效性。研究强调,在实际应用中需同时评估准确性和稳定性。
原文摘要 · Abstract (English)
Reliability estimation of large language models is in many cases as crucial as their accuracy, as reliable models are more trustworthy, robust, and suitable for practical applications. Recent advancements in natural language processing (NLP), particularly those based on transformer architectures, have significantly accelerated progress across various NLP tasks. This study focuses on the reliability of transformer-based question answering (QA) models, specifically BERT models and its variants (RoBERTa, ALBERT, DistilBERT). These encoder-only pretrained transformers have demonstrated remarkable accuracy in QA tasks that can be treated as classification tasks. However, their reliability remains underexplored. This study evaluates the reliability of four BERT-based models by assessing response stability under two conditions: (1) internal model variations induced via Monte Carlo Dropout (MCD) and (2) input perturbations through paraphrasing. Using the SQuAD and QuAC datasets, we investigate how dropout rates affect prediction consistency and whether lexical changes impact answer stability. Our findings reveal that RoBERTa maintains higher reliability, whereas AlBERT and DistilBERT exhibit significant inconsistencies. Statistical analyses confirm that enabling MCD during prediction does not disrupt inference dynamics, validating its effectiveness as a reliability metric. These findings underscore the importance of evaluating both accuracy and stability in QA models to ensure stability in real-world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。