English

Black-Box Tuning for Language-Model-as-a-Service

Computation and Language 2022-06-28 v4 Artificial Intelligence

Abstract

Extremely large pre-trained language models (PTMs) such as GPT-3 are usually released as a service. It allows users to design task-specific prompts to query the PTMs through some black-box APIs. In such a scenario, which we call Language-Model-as-a-Service (LMaaS), the gradients of PTMs are usually unavailable. Can we optimize the task prompts by only accessing the model inference APIs? This paper proposes the black-box tuning framework to optimize the continuous prompt prepended to the input text via derivative-free optimization. Instead of optimizing in the original high-dimensional prompt space, which is intractable for traditional derivative-free optimization, we perform optimization in a randomly generated subspace due to the low intrinsic dimensionality of large PTMs. The experimental results show that the black-box tuning with RoBERTa on a few labeled samples not only significantly outperforms manual prompt and GPT-3's in-context learning, but also surpasses the gradient-based counterparts, i.e., prompt tuning and full model tuning.

Keywords

Cite

@article{arxiv.2201.03514,
  title  = {Black-Box Tuning for Language-Model-as-a-Service},
  author = {Tianxiang Sun and Yunfan Shao and Hong Qian and Xuanjing Huang and Xipeng Qiu},
  journal= {arXiv preprint arXiv:2201.03514},
  year   = {2022}
}

Comments

Accepted by ICML 2022. Camera-ready version

R2 v1 2026-06-24T08:45:20.266Z