ArXiv Picks

💬

軽量心理カウンセリングAI「PsyLite」：安全性とユーモアで心のケアを身近に

原文

PsyLite Technical ReportPagePDF

著者

Fangjun Ding, Renyu Zhang, Xinyu Feng, Chengye Xie, Zheng Zhang, Yanting Zhang

公開日

概要

本研究では、メンタルヘルス分野で重要性を増すAI心理カウンセリングに着目し、軽量で安全な対話が可能な大規模言語モデル「PsyLite」を開発しました。既存モデルの課題である対話の安全性、シナリオ対応力、リソース制約下での展開を解決するため、InternLM2.5-7B-chatを基盤とし、深層推論能力、心理カウンセリング能力、安全な対話能力を向上。ユーモアを交えた対話でユーザー体験を高め、リソースの限られた環境でも利用可能な心理カウンセリング・ソリューションを提供します。

デジタル技術の発展に伴い、AIを活用した心理カウンセリングは、メンタルヘルス分野においてますます重要な役割を担うようになっています。しかし、既存のモデルには、対話の安全性、詳細なシナリオへの対応、軽量な実装という点で課題が残されています。本研究では、これらの課題を解決するために、軽量な心理カウンセリング大規模言語モデルエージェント「PsyLite」を提案します。PsyLiteは、基盤モデルであるInternLM2.5-7B-chatをベースに開発されました。

背景

従来の心理カウンセリングは、自然言語処理（NLP）技術を利用していましたが、ハードウェアのリソース制限により、大規模な普及とタイムリーなサービス提供が困難でした。AIの発展は、いつでもどこでも心理的なサポートと調整を受けられる新たな機会をもたらしました。しかし、大規模な心理カウンセリングモデルを設計し、会話の安全性を確保し、ユーザー体験を向上させるためには、課題が残っています。

課題

既存の心理カウンセリングモデルには、以下の3つの課題があります。

  • 対話の安全性：有害な情報を出力する可能性
  • シナリオ対応力：多様な状況や感情に対応できない
  • 軽量な実装：リソースの限られた環境では利用できない

解決策

これらの課題を解決するために、本研究では、以下の技術を組み合わせたPsyLiteを開発しました。

  • 深層推論能力の強化：ハイブリッド蒸留データによるファインチューニングとORPO（Odds Ratio Preference Optimization）による選好最適化
  • 対話安全性の向上：対話安全性に関する選好データセットを使用し、ORPO強化学習戦略を採用
  • ユーモアの導入：条件付きRAG（Retrieval-Augmented Generation）を設計し、心理カウンセリング中に適切なタイミングで漫才の要素を導入
  • 軽量化：量子化技術（GGUF q4_k_m）を使用し、低ハードウェアでの実装を実現（5GBのメモリで動作可能）

技術の中身

PsyLiteは、以下の2段階のトレーニング戦略を採用しています。

  1. ハイブリッド蒸留データによるファインチューニング: 一般的なデータと心理カウンセリングのデータを組み合わせて学習させることで、専門的な知識だけでなく、深く考える能力を養います。
  2. ORPOによる選好最適化: 対話の安全性を高めるために、有害な情報を出力しないように学習させます。

また、PsyLiteは、異なる対話シナリオに応じて異なる戦略を採用しています。通常の会話に加えて、ユーザーが深刻な状況にある場合には、ユーモアを交えた対話で気分転換を促し、ユーザー体験を向上させます。

実験と結果

PsyLiteの性能は、以下の3つの評価基準で検証されました。

  • CEval（中国語汎用評価）
  • CPsyCounE（心理カウンセリング専門評価）
  • SafeDialBench（対話安全性評価）

その結果、PsyLiteは、特に心理カウンセリングの専門性（CPsyCounEスコアが47.6%向上）と対話安全性（SafeDialBenchスコアが2.4%向上）において、既存のモデルを上回る性能を発揮しました。

応用・社会的インパクト

PsyLiteは、リソースの限られた環境でも利用可能な心理カウンセリング・ソリューションを提供することで、メンタルヘルスケアの普及に貢献することが期待されます。また、ユーモアを交えた対話でユーザー体験を高めることで、より親しみやすく、効果的なカウンセリングを実現します。

まとめ

本研究では、軽量で安全な対話が可能な大規模言語モデル「PsyLite」を開発しました。PsyLiteは、深層推論能力、心理カウンセリング能力、安全な対話能力を向上させ、ユーモアを交えた対話でユーザー体験を高めます。また、リソースの限られた環境でも利用可能な心理カウンセリング・ソリューションを提供することで、メンタルヘルスケアの普及に貢献することが期待されます。

補足

●大規模言語モデル（LLM）とは？

大量のテキストデータで学習された、自然言語処理を行うAIモデルの一種です。質問応答、文章生成、翻訳など、さまざまなタスクを実行できます。近年のAI技術の発展により、LLMは急速に進化しており、その応用範囲はますます広がっています。

図1 PsyLiteの全体アーキテクチャ PsyLiteのアーキテクチャ

PsyLiteは、InternLM2.5-7B-chatをベースモデルとして、2段階のトレーニング戦略と条件付きRAGを組み合わせることで、軽量で安全な対話が可能な心理カウンセリング・ソリューションを実現しています。

今後の展望

PsyLiteは、メンタルヘルスケアの分野に大きな変革をもたらす可能性を秘めています。今後は、以下のような場面で注目されることが期待されます。

  • オンラインカウンセリング: いつでもどこでも気軽に利用できる心理カウンセリング・サービス
  • 企業: 従業員のメンタルヘルスケアをサポートするツール
  • 教育機関: 学生の悩み相談やメンタルヘルス教育に活用

PsyLiteの技術は、今後ますます進化し、より多くの人々のメンタルヘルスをサポートする存在となるでしょう。