SOTAVerified

BloombergGPT: A Large Language Model for Finance

2023-03-30Unverified0· sign in to hype

Shijie Wu, Ozan Irsoy, Steven Lu, Vadim Dabravolski, Mark Dredze, Sebastian Gehrmann, Prabhanjan Kambadur, David Rosenberg, Gideon Mann

Unverified — Be the first to reproduce this paper.

Reproduce

Abstract

The use of NLP in the realm of financial technology is broad and complex, with applications ranging from sentiment analysis and named entity recognition to question answering. Large Language Models (LLMs) have been shown to be effective on a variety of tasks; however, no LLM specialized for the financial domain has been reported in literature. In this work, we present BloombergGPT, a 50 billion parameter language model that is trained on a wide range of financial data. We construct a 363 billion token dataset based on Bloomberg's extensive data sources, perhaps the largest domain-specific dataset yet, augmented with 345 billion tokens from general purpose datasets. We validate BloombergGPT on standard LLM benchmarks, open financial benchmarks, and a suite of internal benchmarks that most accurately reflect our intended usage. Our mixed dataset training leads to a model that outperforms existing models on financial tasks by significant margins without sacrificing performance on general LLM benchmarks. Additionally, we explain our modeling choices, training process, and evaluation methodology. We release Training Chronicles (Appendix C) detailing our experience in training BloombergGPT.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
arc_challengeGPT-NeoX 20B (1-shot)Accuracy45.39—Unverified
arc_challengeOPT 66B (one-shot)Accuracy44.54—Unverified
arc_challengeBloomberg GPT 50B (1-shot)Accuracy48.63—Unverified
arc_challengeBLOOM 176B (1-shot)Accuracy50.85—Unverified
arc_easyBloomberg GPT 50B (1-shot)Accuracy73.99—Unverified
arc_easyBLOOM 176B (1-shot)Accuracy75.93—Unverified
arc_easyOPT 66B (1-shot)Accuracy71.25—Unverified
arc_easyGPT-NeoX 20B (1-shot)Accuracy70.79—Unverified
BIG-bench (Causal Judgment)OPT 66B (few-shot, k=3)Accuracy51.87—Unverified
BIG-bench (Causal Judgment)GPT-NeoX 20B (few-shot, k=3)Accuracy52.41—Unverified
BIG-bench (Causal Judgment)BloombergGPT 50B (few-shot, k=3)Accuracy49.73—Unverified
BIG-bench (Causal Judgment)BLOOM 176B (few-shot, k=3)Accuracy51.87—Unverified
BIG-bench (Causal Judgment)PaLM 540B (few-shot, k=3)Accuracy61—Unverified
BIG-bench (Date Understanding)GPT-NeoX 20B (few-shot, k=3)Accuracy45.6—Unverified
BIG-bench (Date Understanding)BLOOM 176B (few-shot, k=3)Accuracy50—Unverified
BIG-bench (Date Understanding)OPT 66B (few-shot, k=3)Accuracy49.6—Unverified
BIG-bench (Date Understanding)Bloomberg GPT 50B (few-shot, k=3)Accuracy54.8—Unverified
BIG-bench (Date Understanding)PaLM 540B (few-shot,k=3)Accuracy53.6—Unverified
BIG-bench (Disambiguation QA)PaLM 540B (few-shot, k=3)Accuracy60.8—Unverified
BIG-bench (Disambiguation QA)GPT-NeoX 20B (few-shot, k=3)Accuracy40.8—Unverified
BIG-bench (Disambiguation QA)Bloomberg GPT 50B (few-shot, k=3)Accuracy34—Unverified
BIG-bench (Disambiguation QA)BLOOM 176B (few-shot, k=3)Accuracy40.4—Unverified
BIG-bench (Disambiguation QA)OPT 66B (few-shot, k=3)Accuracy40.4—Unverified
BIG-bench (Sports Understanding)GPT-NeoX (few-shot, k=3)Accuracy53.2—Unverified
BIG-bench (Sports Understanding)Bloomberg GPT (few-shot, k=3)Accuracy62.8—Unverified
BIG-bench (Sports Understanding)PaLM 540B (few-shot, k=3)Accuracy80.4—Unverified
BIG-bench (Sports Understanding)OPT 66B (few-shot, k=3)Accuracy54.4—Unverified
CommonsenseQAOPT 66B (1-shot)Accuracy66.4—Unverified
CommonsenseQABloomberg GPT 50B (1-shot)Accuracy65.5—Unverified
CommonsenseQABLOOM 176B (1-shot)Accuracy64.2—Unverified
CommonsenseQAGPT-NeoX 20B (1-shot)Accuracy60.4—Unverified
ReCoRDOPT 66B (1-shot)F182.5—Unverified
ReCoRDBLOOM 176B (1-shot)F178—Unverified
ReCoRDGPT-NeoX 20B (1-shot)F167.9—Unverified
ReCoRDBloomberg GPT 50B (1-shot)F182.8—Unverified
WinoGrandeGPT-NeoX (one-shot)Accuracy60.6—Unverified
WinoGrandeBloomberg GPT (one-shot)Accuracy64.1—Unverified
WinoGrandeOPT 66B (1-shot)Accuracy66.1—Unverified
WinoGrandeBLOOM 176B (1-shot)Accuracy67—Unverified

Reproductions