OpenAIのAPI関連メモ

tokoi 2023/02/19に更新

fine-tuningについて

promptを長くして色々指示を入れ込めば大抵のことはできるけれど、トークン消費激しい。ファインチューニングするとトークン消費を押さえつつ、ねらった回答を生成できる。（あと多分時間パフォーマンスも向上する）

prompt, completionのペアを含むデータセットをCLI（※1）でなげると、ファインチューニングJOBのIDがもらえる。
終わったらファインチューニングされた新しいモデルの名前を取得できる。
以降はそれを指定してcompletions-APIを呼び出せば良い。

※1 : pip install openaiすると、CLIのコマンドopenaiもインストールされる。

tokoi 2023/02/19

fine-tuningに関連する費用

トレーニング費用
- (トレーニングファイル内のトークン * トレーニングエポックの数) = 合計トレーニングトークン
- 上記 1kトレーニングトークンに対して、davinciなら0.03USDかかる。
fine-tuningモデルの使用費用
- 通常モデルと同じ計算方式で使用トークンが決まる
- 上記 1k使用トークンに対して、davinciなら 0.12USDかかる。

※素のdavinci使用料は、0.02USD / kToken

tokoi 2023/02/19

fine-tuning可能なモデル

残念ながら、text-davinci-003をfine-tuningすることはできない。

fine-tuningできるのは以下。

ベースモデルであるdavinci
ユーザが（ベースモデルを）fine-tuningしたモデル

tokoi 2023/02/19

分類とかどうやってやるの

ファインチューニングの時に、{'prompto': '{テキスト}', 'completion': '{正解ラベル}'} って感じのデータセットでトレーニングさせるだけ。

notebook例:

https://github.com/openai/openai-cookbook/blob/main/examples/Fine-tuned_classification.ipynb

tokoi 2023/02/19に更新

text-davinci-003 で会話させるには

promptに会話ログを含める。
promptの最後の行がAI:になるようにすれば、続きの文を考えてくれる、という寸法。

例:

prompt = """The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly.

Human: Hello, who are you?
AI: I am an AI created by OpenAI. How can I help you today?
Human: I'd like to cancel my subscription.
AI:
"""

あとは、AIが生成したもの、人間が書き送ったもの、を同様の形式で数珠つなぎにしたものを毎回promptとしてopenai.Completion.createを実行すればよい。

tokoi 2023/02/19

会話を続けていくと、途中でtokenが足りなくなるが、会話ログ部分の先頭を削ってローリングする感じで回していけば、健忘っぽくなるもののそこそこ自然に振る舞えるのでは無いか。（やったことない）

tokoi 2023/02/19

同じ事を（素の）davinciでやろうとしても、一人芝居が始まってしまう。
数百のデータでfine-tuningすればいけるのか？（とりあえず数十程度ではダメだった）

tokoi 2023/02/20

Embeddingsってなに

文字列の概念を数列化したもの。
さまざまな文書／文字列について、それぞれの概念のようなものを数値化して一緒に保管しておけば、「関連性」に基づいた文書／文字列検索システムを構築できる。

OpenAIは、text-embedding-ada-002モデルをEmbedingsAPIとして提供している。

Completionには最大トークン数という"限界"があるが、Embeddingsを上手く使うと適切（概念的に近そう）な外部データだけをその都度参照して補完させることができる。

Embedingsを使った外部データソースから、元promptに近い文書を検索
得られた文書だけをpromptの先頭などに追加する
補完する