家庭質問を正確に分類する小型LLMの微調整実験

要約

この記事では、家庭に関する質問を分類するために、ローカルLLM(Qwen 3:0.6B)を微調整するプロジェクトについて紹介しています。著者は、このチャットボットが家庭のメンテナンスや医者の予約など、様々な質問に答えることを目指しています。質問の分類精度を向上させるために、質問を事前処理し、特定のメタデータカテゴリにマッピングしています。これにより、ベクトル検索の対象を絞り込み、より正確な回答を得ることが可能となります。

プロジェクトでは、Qwen 3:4BとQwen 3:0.6Bの二つのモデルを使用しており、4Bバージョンは一般的な質問応答に、0.6Bバージョンは質問の分類に特化しています。この実験の目的は、非常に小型のLLMが家庭関連の質問データセットで信頼性のある分類を行えるかを検証することです。著者は、Unslothというオープンソースのフレームワークを使用して微調整を行い、850件のデータを70/15/15の比率で訓練、評価、テストデータに分割しています。

また、事前に設定したベースラインモデルを利用して、微調整前の性能を測定しています。ベースラインでは、元のQwen 0.6Bモデルを単独で使用し、提示された質問に対して正しいカテゴリが返されるかを評価しましたが、結果は芳しくありませんでした。これに対して、微調整を行うことで精度を向上させることが期待されています。


元記事: https://www.teachmecoolstuff.com/viewarticle/fine-tuning-a-local-llm-to-categorize-questions

公開日: Sun, 21 Jun 2026 22:55:23 +0000


この記事はAIアシスト編集により作成されています。

📰 元記事: 元記事を読む

コメントする