每天早上一封邮件,把昨天的 AI 梳理好订阅邮件

METAL LAB

Reliable Financial Named Entity Recognition under Domain Shift

arXiv:2608.195582026-08-21

AI在正式文件里学到的自信,一到推特上就变得不可信

研究者用美国SEC财务文件训练的命名实体识别AI,拿去测试财经新闻和普通社交媒体文本,发现不仅准确率下降,连AI自己给出的置信度也开始失灵。在训练领域内表现最好的整句概率置信度,一旦领域变化就变得不可靠,而只看实体片段的概率以及多次生成结果的一致性(自洽性)则更加稳健。但在极端的社交媒体场景下,没有任何置信度指标能可靠地找出一批可以放心自动处理的预测。

他们做了什么

  1. 用SEC财务文件训练BERT标注器和经LoRA微调的小型语言模型(Qwen2.5 0.5B/1.5B),再拿到财经新闻和普通话题推特数据上测试,重复三个随机种子并计算自助法置信区间
  2. 比较了五种推理时置信度信号:整句输出概率、词元概率、实体片段概率、实体类型概率,以及基于多次采样的自洽性
  3. 整句概率在领域内是最强的错误检测信号,但一旦领域偏移就明显退化;实体片段概率和自洽性更稳健,其中自洽性无需额外校准就更准确地反映真实正确率
  4. 只自动处理置信度最高的40%句子时,领域内句子错误率从34.3%降到2%以下,在财经新闻上依然有效,但在极端社交媒体数据上找不到足够大的可靠子集
  5. 据此提出分阶段部署思路:先检测输入是否发生严重领域偏移,再仅在可靠领域内使用置信度来决定是否自动放行结果
Fig. 1: Reliability diagrams for self-consistency confidence (0.5B, pooled seeds). Near-diagonal in-domain; on both shifted tiers the top-confidence bin inverts.
Fig. 1: Reliability diagrams for self-consistency confidence (0.5B, pooled seeds). Near-diagonal in-domain; on both shifted tiers the top-confidence bin inverts.
TABLE I: Dataset statistics after harmonization and filtering.
SplitDomainSent.Ent.PERORGLOC
TrainFIN (filings) [27]1014980648175157
ValidFIN (filings) [27]150177976812
TestFIN (filings) [27]2992952015638
TestFiNER-ORD (news) [28]3003227815193
TestTweetNER7 (tweets) [30]300619381135103
Fig. 2: Sentence-level risk–coverage (Qwen2.5-0.5B; line = 3-seed mean, band = seed range; lower is better). Abstention sharply reduces risk in-domain and on news, but cannot rescue the far-shift tier.
Fig. 2: Sentence-level risk–coverage (Qwen2.5-0.5B; line = 3-seed mean, band = seed range; lower is better). Abstention sharply reduces risk in-domain and on news, but cannot rescue the far-shift tier.
TABLE II: Main results (%; mean±std over 3 seeds; Qwen2.5-1.5B is a single-seed scale check).
ModelDomainPRF1Halluc.%ECE
BERT-baseFIN66.7±2.374.2±1.470.3±1.50.102±0.016
BERT-baseFiNER-ORD39.1±0.337.6±2.338.3±1.10.071±0.017
BERT-baseTweetNER735.2±5.218.4±1.724.1±2.50.074±0.007
Qwen-0.5BFIN40.5±0.434.7±2.137.4±1.33.6±2.40.362±0.009
Qwen-0.5BFiNER-ORD30.7±1.318.0±4.422.6±3.88.2±0.70.407±0.005
Qwen-0.5BTweetNER730.8±0.723.7±4.226.7±2.85.6±1.10.370±0.007
Qwen-1.5BFIN47.744.746.24.30.322
Qwen-1.5BFiNER-ORD56.348.452.13.60.227
Qwen-1.5BTweetNER744.950.947.76.80.358
TABLE III: Error-detection AUROC and ECE per confidence signal (Qwen2.5-0.5B; mean±std over 3 seeds), by test domain.
SignalFINFiNER-ORDTweetNER7
AUROCECEAUROCECEAUROCECE
Sequence prob.0.839±0.0140.512±0.0050.661±0.0150.593±0.0110.626±0.0190.586±0.006
Token prob.0.823±0.0120.531±0.0040.666±0.0190.615±0.0130.605±0.0180.611±0.006
Span prob.0.801±0.0040.362±0.0090.690±0.0300.407±0.0050.721±0.0110.370±0.007
Type prob.0.492±0.0190.506±0.0040.536±0.0220.533±0.0190.511±0.0100.573±0.002
Self-consistency0.690±0.0190.116±0.0290.671±0.0480.120±0.0150.682±0.0170.100±0.008
TABLE IV: Scale check: error-detection AUROC by signal and domain for Qwen2.5-0.5B (3-seed mean) vs. Qwen2.5-1.5B (single seed).
Signal0.5B (3 seeds)1.5B (1 seed)
FINFiNERTweetFINFiNERTweet
Sequence prob.0.8390.6610.6260.7640.7200.637
Span prob.0.8010.6900.7210.8190.7290.638
Self-consistency0.6900.6710.6820.7350.7440.663
TABLE V: Entity-level selective precision (%) at 100/80/60% coverage (Qwen2.5-0.5B; mean±std over 3 seeds).
DomainSignalP@100%P@80%P@60%
FINSeq. prob.40.5±0.449.0±0.858.4±0.6
FINSpan prob.40.5±0.449.2±0.356.4±0.6
FINSelf-consistency40.5±0.447.7±0.452.8±1.0
FiNERSeq. prob.30.7±1.334.9±3.040.1±1.3
FiNERSpan prob.30.7±1.335.3±1.740.8±2.6
FiNERSelf-consist.30.7±1.334.9±3.041.4±2.3
TweetSeq. prob.30.8±0.734.2±1.036.8±2.9
TweetSpan prob.30.8±0.736.6±1.142.7±1.1
TweetSelf-consist.30.8±0.735.5±1.740.5±1.5
TABLE VI: TweetNER7 filtering sensitivity, main (drop-sentence) vs. relaxed (ignore-entity) protocol. Encoder: 3-seed mean; Qwen: seed 42.
FilteredRelaxed
Sentences dropped (of 2807)21790
Evaluated (cap 300)300300
Gold PER/ORG/LOC619443
Ignored entities0488
Encoder F124.1±2.519.5±1.0
Encoder MSP AUROC0.615±0.0540.650±0.047
Qwen-0.5B F129.729.1
Qwen-0.5B span AUROC0.7250.715

为什么重要

在金融合规、风险监控等场景中,单纯的准确率数字无法告诉从业者何时可以放心让AI自动处理结果。这项研究表明,当输入数据类型发生变化时,AI自身汇报的置信度可能悄悄失效,并为选择合适的置信度指标提供了具体依据。

本文术语

  • 命名实体识别(NER) · 从文本中识别并标注人名、机构名、地名等专有名词的AI任务
  • 领域偏移(distribution shift) · 实际使用时的文本风格或主题与AI训练数据不同的现象
  • 选择性预测(selective prediction) · 让AI在没有把握时拒绝回答、转交人工复核的处理方式
  • 自洽性(self-consistency) · 通过让AI对同一输入多次生成结果、看结果是否一致来衡量置信度的方法
  • LoRA · 一种轻量级微调方法,只训练额外添加的小部分参数,而不重新训练整个大模型
  • AUROC / ECE · AUROC衡量置信度分数区分对错预测的能力,ECE衡量AI给出的置信度数值与实际正确率的吻合程度

论文原文摘要(英文)

Financial AI systems often train information extractors on one textual register and deploy them across filings, news, and user-generated content, while standard F1 scores do not indicate which predictions remain safe to automate when the input distribution changes. We study confidence estimation and selective prediction for financial named entity recognition (NER) on a three-tier stress test spanning SEC filings, financial news, and general-topic social media as an extreme out-of-domain condition. We evaluate a BERT tagger and LoRA-tuned Qwen2.5-0.5B/1.5B models using five inference-time confidence signals, three training seeds, and bootstrap intervals. Confidence rankings themselves change under distribution shift: whole-output probability is the strongest in-domain error detector but deteriorates out of domain, whereas entity-span probability and self-consistency are more robust; self-consistency is also better calibrated without post-hoc fitting. Abstention reduces sentence error from 34.3% to below 2% on the highest-confidence 40% of in-domain inputs and remains useful on financial news, but recovers no usefully large clean subset under the extreme social-media shift. These results motivate a staged deployment strategy that detects severe distribution shift upstream before applying prediction-level confidence gating.

作者 · Zihao Zheng, Baichuan Li, Junyi Yao, Jiayu Long

在 arXiv 阅读

最新论文

全部论文 →

METAL LAB 最新报道

图片来源: Zihao Zheng et al., arXiv:2608.19558, arxiv-nonexclusive