SOTAVerified

BloombergGPT: A Large Language Model for Finance

2023-03-30Code Available0· sign in to hype

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

Code Available — Be the first to reproduce this paper.

Reproduce

Code

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

Reproductions