CoChat
Sign Up
PreprintCited by 15Topic ModelingDomain Adaptation and Few-Shot LearningNatural Language Processing Techniques

The DOI matches its DataCite record.

BLADE: Enhancing Black-box Large Language Models with Small Domain-Specific Models

Open access
Li, Haitao, Ai, Qingyao, Chen, Jia, Dong, Qian, Wu, Zhijing, Liu, Yiqun, Chen, Chong, Tian, Qi·2024·arXiv
DOI 10.48550/arxiv.2403.18365arXiv 2403.18365
Open this paper in CoChat — free accountOpen PDF

Abstract

Large Language Models (LLMs) like ChatGPT and GPT-4 are versatile and capable of addressing a diverse range of tasks. However, general LLMs, which are developed on open-domain data, may lack the domain-specific knowledge essential for tasks in vertical domains, such as legal, medical, etc. To address this issue, previous approaches either conduct continuous pre-training with domain-specific data or employ retrieval augmentation to support general LLMs. Unfortunately, these strategies are either cost-intensive or unreliable in practical applications. To this end, we present a novel framework named BLADE, which enhances Black-box LArge language models with small Domain-spEcific models. BLADE consists of a black-box LLM and a small domain-specific LM. The small LM preserves domain-specific knowledge and offers specialized insights, while the general LLM contributes robust language comprehension and reasoning capabilities. Specifically, our method involves three steps: 1) pre-training the small LM with domain-specific data, 2) fine-tuning this model using knowledge instruction data, and 3) joint Bayesian optimization of the general LLM and the small LM. Extensive experiments conducted on public legal and medical benchmarks reveal that BLADE significantly outperforms existing approaches. This shows the potential of BLADE as an effective and cost-efficient solution in adapting general LLMs for vertical domains.

Overview

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

DataCite · 10.48550/arxiv.2403.18365

Cites

Statistics

Verification

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

The self-check needs the paper’s full text.

Citations

Cites

No references could be resolved yet.

Research works printed but not linked: 40

  1. Aharoni, R.; and Goldberg, Y. 2020. Unsupervised do- main clusters in pretrained language models. arXiv preprint arXiv:2004.02105. · arXiv
  2. Arefeen, M. A.; Debnath, B.; and Chakradhar, S. 2023. Lean- Context: Cost-Efficient Domain-Specific Question Answer- ing Using LLMs. arXiv preprint arXiv:2309.00841. · arXiv
  3. Bai, J.; Bai, S.; Chu, Y.; Cui, Z.; Dang, K.; Deng, X.; Fan, Y.; Ge, W.; Han, Y.; Huang, F.; et al. 2023. Qwen technical report. arXiv preprint arXiv:2309.16609. · arXiv
  4. Baichuan. 2023. Baichuan 2: Open Large-scale Language Models. arXiv preprint arXiv:2309.10305. · arXiv
  5. Borgeaud, S.; Mensch, A.; Hoffmann, J.; Cai, T.; Rutherford, E.; Millican, K.; Van Den Driessche, G. B.; Lespiau, J.-B.; Damoc, B.; Clark, A.; et al. 2022. Improving language mod- els by retrieving from trillions of tokens. In International conference on machine learning, 2206-2240.
  6. PMLR.
  7. Chalkidis, I. 2023. ChatGPT may Pass the Bar Exam soon, but has a Long Way to Go for the LexGLUE benchmark. arXiv:2304.12202. · arXiv
  8. Chen, L.; Chen, J.; Goldstein, T.; Huang, H.; and Zhou, T. 2023. InstructZero: Efficient Instruction Optimization for Black-Box Large Language Models. arXiv preprint arXiv:2306.03082. · arXiv
  9. Cheng, D.; Huang, S.; and Wei, F. 2023. Adapting Large Language Models via Reading Comprehension. arXiv:2309.09530. · arXiv
  10. Cui, J.; Li, Z.; Yan, Y.; Chen, B.; and Yuan, L. 2023. ChatLaw: Open-Source Legal Large Language Model with Integrated External Knowledge Bases. arXiv:2306.16092. · arXiv
  11. Dai, Z.; Zhao, V. Y.; Ma, J.; Luan, Y.; Ni, J.; Lu, J.; Bakalov, A.; Guu, K.; Hall, K. B.; and Chang, M.-W. 2022. Promp- tagator: Few-shot dense retrieval from 8 examples. arXiv preprint arXiv:2209.11755. · arXiv
  12. Du, Z.; Qian, Y.; Liu, X.; Ding, M.; Qiu, J.; Yang, Z.; and Tang, J. 2022. GLM: General Language Model Pretraining with Autoregressive Blank Infilling. In Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), 320-335.
  13. Frazier, P. I. 2018. A tutorial on Bayesian optimization. arXiv preprint arXiv:1807.02811. · arXiv
  14. Hansen, N. 2016. The CMA evolution strategy: A tutorial. arXiv preprint arXiv:1604.00772. · arXiv
  15. Hu, E. J.; Shen, Y.; Wallis, P.; Allen-Zhu, Z.; Li, Y.; Wang, S.; Wang, L.; and Chen, W. 2021. Lora: Low-rank adaptation of large language models. arXiv preprint arXiv:2106.09685. · arXiv
  16. Huang, Q.; Tao, M.; An, Z.; Zhang, C.; Jiang, C.; Chen, Z.; Wu, Z.; and Feng, Y. 2023. Lawyer LLaMA Technical Report. ArXiv, abs/2305.15062. · arXiv
  17. Joshi, M.; Lee, K.; Luan, Y.; and Toutanova, K. 2020. Con- textualized representations using textual encyclopedic knowl- edge. arXiv preprint arXiv:2004.12006. · arXiv
  18. Kleinberg, J. M. 1997. Two algorithms for nearest-neighbor search in high dimensions. In Proceedings of the twenty-ninth annual ACM symposium on Theory of computing, 599-608.
  19. Lewis, P.; Perez, E.; Piktus, A.; Petroni, F.; Karpukhin, V.; Goyal, N.; Küttler, H.; Lewis, M.; Yih, W.-t.; Rocktäschel, T.; et al. 2020. Retrieval-augmented generation for knowledge- intensive nlp tasks. Advances in Neural Information Process- ing Systems, 33: 9459-9474.
  20. Lewis, P.; Stenetorp, P.; and Riedel, S. 2020. Question and Answer Test-Train Overlap in Open-Domain Question An- swering Datasets. arXiv:2008.02637. · arXiv
  21. Lewis, P.; Wu, Y.; Liu, L.; Minervini, P.; Küttler, H.; Pik- tus, A.; Stenetorp, P.; and Riedel, S. 2021. Paq: 65 million probably-asked questions and what you can do with them. Transactions of the Association for Computational Linguis- tics, 9: 1098-1115.
  22. Li, H.; Chen, Y.; Ai, Q.; Wu, Y.; Zhang, R.; and Liu, Y. 2024b. LexEval: A Comprehensive Chinese Legal Benchmark for Evaluating Large Language Models. arXiv:2409.20288. · arXiv
  23. Li, H.; Dong, Q.; Chen, J.; Su, H.; Zhou, Y.; Ai, Q.; Ye, Z.; and Liu, Y. 2024c. LLMs-as-Judges: A Comprehensive Sur- vey on LLM-based Evaluation Methods. arXiv:2412.05579. · arXiv
  24. Li, J.; Zhong, S.; and Chen, K. 2021. MLEC-QA: A Chinese multi-choice biomedical question answering dataset. In Pro- ceedings of the 2021 Conference on Empirical Methods in Natural Language Processing, 8862-8874.
  25. Li, Y.; Wu, Y.; Li, J.; and Liu, S. 2023a. Prompting large language models for zero-shot domain adaptation in speech recognition. arXiv preprint arXiv:2306.16007. · arXiv
  26. Li, Z.; Zhang, X.; Zhang, Y.; Long, D.; Xie, P.; and Zhang, M. 2023b. Towards General Text Embeddings with Multi-stage Contrastive Learning. arXiv:2308.03281. · arXiv
  27. Liu, J.; Liu, A.; Lu, X.; Welleck, S.; West, P.; Bras, R. L.; Choi, Y.; and Hajishirzi, H. 2021. Generated knowledge prompting for commonsense reasoning. arXiv preprint arXiv:2110.08387. · arXiv
  28. Luo, L.; Ning, J.; Zhao, Y.; Wang, Z.; Ding, Z.; Chen, P.; Fu, W.; Han, Q.; Xu, G.; Qiu, Y.; Pan, D.; Li, J.; Li, H.; Feng, W.; Tu, S.; Liu, Y.; Yang, Z.; Wang, J.; Sun, Y.; and Lin, H. 2023. Taiyi: A Bilingual Fine-Tuned Large Language Model for Diverse Biomedical Tasks. arXiv preprint arXiv:2311.11608. OpenAI. 2023. GPT-4 Technical Report. arXiv:2303.08774. · arXiv
  29. Rubin, D. B. 1980. Randomization analysis of experimental data: The Fisher randomization test comment. Journal of the American statistical association, 75(371): 591-593.
  30. Sachidananda, V.; Kessler, J. S.; and Lai, Y.-A. 2021. Ef- ficient domain adaptation of language models via adaptive tokenization. arXiv preprint arXiv:2109.07460. · arXiv
  31. Shi, W.; Min, S.; Yasunaga, M.; Seo, M.; James, R.; Lewis, M.; Zettlemoyer, L.; and Yih, W.-t. 2023. Replug: Retrieval- augmented black-box language models. arXiv preprint arXiv:2301.12652. · arXiv
  32. Sun, T.; Shao, Y.; Qian, H.; Huang, X.; and Qiu, X. 2022a. Black-box tuning for language-model-as-a-service. In Inter- national Conference on Machine Learning, 20841-20855.
  33. PMLR.
  34. Sun, Z.; Wang, X.; Tay, Y.; Yang, Y.; and Zhou, D. 2022b. Recitation-augmented language models. arXiv preprint arXiv:2210.01296. · arXiv
  35. Wang, W.; Srikumar, V.; Hajishirzi, H.; and Smith, N. A. 2022. Elaboration-generating commonsense question an- swering at scale. arXiv preprint arXiv:2209.01232. · arXiv
  36. Xiao, S.; Liu, Z.; Zhang, P.; and Muennighof, N. 2023. C- Pack: Packaged Resources To Advance General Chinese Em- bedding. arXiv preprint arXiv:2309.07597. · arXiv
  37. Yang, S.; Zhao, H.; Zhu, S.; Zhou, G.; Xu, H.; Jia, Y.; and Zan, H. 2023. Zhongjing: Enhancing the Chinese Medical Ca- pabilities of Large Language Model through Expert Feedback and Real-world Multi-turn Dialogue. arXiv:2308.03549. · arXiv
  38. Yu, W.; Iter, D.; Wang, S.; Xu, Y.; Ju, M.; Sanyal, S.; Zhu, C.; Zeng, M.; and Jiang, M. 2022. Generate rather than retrieve: Large language models are strong context generators. arXiv preprint arXiv:2209.10063. · arXiv
  39. Zeng, A.; Liu, X.; Du, Z.; Wang, Z.; Lai, H.; Ding, M.; Yang, Z.; Xu, Y.; Zheng, W.; Xia, X.; et al. 2022. Glm- 130b: An open bilingual pre-trained model. arXiv preprint arXiv:2210.02414. · arXiv
  40. Zhong, H.; Xiao, C.; Tu, C.; Zhang, T.; Liu, Z.; and Sun, M. 2020. JEC-QA: a legal-domain question answering dataset. In Proceedings of the AAAI Conference on Artificial Intelli- gence, volume 34, 9701-9708.

Cited by

Sorted by citations

Metadata

Authors
Li, H., Ai, Q., Chen, J., Dong, Q., Wu, Z., Liu, Y., Chen, C., & Tian, Q.
Affiliations
  • Tsinghua University (CN)
  • Xiaohongshu (Little Red Book) (China) (CN)
  • Beijing Institute of Technology (CN)
ORCID
Venue
arXiv
Publisher
arXiv
Published
2024
Volume
39
Issue
23
Pages
24422-24430
ISSN
2159-5399 (ISSN-L)
DOI
10.48550/arxiv.2403.18365
arXiv
2403.18365
OpenAlex
W4409348182
Cited by
15
References
0
Citations by year
2026: 6 · 2025: 7 · 2024: 2
FWCI
12.43
Type
conference paper
Language
en
Open access
Licence
Locations
  • Proceedings of the AAAI Conference on Artificial Intelligence (publishedVersion, other-oa)
  • arXiv (Cornell University) (submittedVersion, other-oa)
  • AAAI Publications (The Association for the Advancement of Artificial Intelligence (AAAI)) (submittedVersion)
  • arXiv (Cornell University)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
Keywords
large language models, domain-specific models, question answering, domain adaptation, Bayesian optimization, small language models

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review