用预训练BERT构建金融问答系统,提升非事实型答案选择准确率。
FinBERT-QA: Financial Question Answering with pre-trained BERT Language Models
- 采用检索+重排序框架,先用BM25找候选答案,再用BERT模型重排筛选
- 在FiQA数据集任务2上,MRR提升16%,NDCG提升17%,Precision@1提升21%
- 适合金融领域自动化决策支持,尤其对缺乏标注数据的场景有强适应性
为应对金融行业对大规模非结构化与结构化数据自动分析的需求,问答(QA)系统可助力金融顾问决策,带来显著竞争优势。为此,我们提出一种基于Transformer预训练BERT语言模型的新型金融问答系统,以解决金融领域数据稀缺与语言特异性问题。系统聚焦于金融非事实型答案选择,即检索段落级文本并选出最相关答案。为提高效率,将答案选择任务建模为重排序问题:首先使用简单的信息检索方法BM25作为答案检索器,返回候选答案列表;再通过变体BERT模型构建答案重排序器,对候选答案进行重排序并选出最优答案。我们研究了多种学习、进一步预训练及微调策略。实验表明,采用迁移与适配式微调及点对点学习方法构建的FinBERT-QA模型表现最佳,在FiQA数据集任务2上,相比现有最优结果,平均排序位置(MRR)提升16%,归一化折扣累积增益(NDCG)提升17%,精确率@1(Precision@1)提升21%。
原文摘要 · Abstract (English)
Motivated by the emerging demand in the financial industry for the automatic analysis of unstructured and structured data at scale, Question Answering (QA) systems can provide lucrative and competitive advantages to companies by facilitating the decision making of financial advisers. Consequently, we propose a novel financial QA system using the transformer-based pre-trained BERT language model to address the limitations of data scarcity and language specificity in the financial domain. Our system focuses on financial non-factoid answer selection, which retrieves a set of passage-level texts and selects the most relevant as the answer. To increase efficiency, we formulate the answer selection task as a re-ranking problem, in which our system consists of an Answer Retriever using BM25, a simple information retrieval approach, to first return a list of candidate answers, and an Answer Re-ranker built with variants of pre-trained BERT language models to re-rank and select the most relevant answers. We investigate various learning, further pre-training, and fine-tuning approaches for BERT. Our experiments suggest that FinBERT-QA, a model built from applying the Transfer and Adapt further fine-tuning and pointwise learning approach, is the most effective, improving the state-of-the-art results of task 2 of the FiQA dataset by 16% on MRR, 17% on NDCG, and 21% on Precision@1.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。