| 講演抄録/キーワード |
| 講演名 |
2025-01-24 10:40
大規模言語モデル Llama 3 を用いた会話からのうつ病検出 ○池内謙祐・岸本泰士郎・中井文哉・ブアジジ モンデル・奥西泰地・鄭 楚恒・北沢桃子・堀込俊郎・大槻知明(慶大) SeMI2024-77 |
| 抄録 |
(和) |
うつ病は社会に多大な影響を与えている. うつ病は早期治療により被害を抑えられるため, 早期発見が求められている. Llama 3 など, 最新の大規模言語モデルを利用したモデルに関して, プロンプトエンジニアリングを行ったモデルは高精度でうつ病を検出できることが報告されている. 一方, ファインチューニング手法は, 推論時間は短いが, 学習するために必要なメモリが多いうえ, 分類モデルにおいて高いうつ病検出精度を達成したという報告はない. 本研究はファインチューニングを用いてプロンプトエンジニアリングを行ったモデルと同程度の精度でより高速にうつ病を検出することが目的である. まず, 大規模言語モデルのパラメータを量子化し, 学習に必要なメモリを削減した. 次に, 分類層を用意するのではなく, 元の大規模言語モデルの構造のまま利用し, 生成モデルとして学習させた. DAIC-WOZ データセットで検証した結果, うつ病検出において F1 score は 84% となり, プロンプトエンジニアリングを行ったモデルと同程度のうつ病検出精度を達成しつつ, より高速に推論することを可能にした. また, 本モデルをプライベートデータセット PROMPT を用いてうつ病検出精度を検証した結果, F1 score は 82% となった. これにより, モデルの高いうつ病検出が, 事前学習された可能性があるオープンデータだけに依存しているわけではないことを示した. |
| (英) |
Depression has a significant impact on society. Early detection is essential because early treatment can mitigate its damages. Models that use state-of-the-art large language models, such as Llama 3, and employ prompt engineering have been reported to detect depression with high accuracy. On the other hand, while fine-tuning techniques offer shorter inference times, they require substantial memory for training and have not been reported to achieve high detection accuracy in classification or regression models.
This study aims to detect depression more rapidly with accuracy comparable to models that use prompt engineering, using fine-tuning techniques. First, the parameters of the large language model were quantized to reduce the memory required for training. Instead of introducing classification or regression layers, the original structure of the large language model was preserved and trained as a generative model.
Validation using the DAIC-WOZ dataset resulted in an F1 score of 84%, achieving depression detection accuracy similar to models using prompt engineering, while enabling faster inference. Additionally, when evaluated using the private dataset PROMPT, the model achieved an F1 score of 82%, demonstrating that its high accuracy in detecting depression is not solely reliant on open datasets that might have been pre-trained. |
| キーワード |
(和) |
うつ病検出 / 大規模言語モデル / ファインチューニング / 量子化 / / / / |
| (英) |
Depression detection / Large Language Model / Fine-tuning / Quantization / / / / |
| 文献情報 |
信学技報, vol. 124, no. 353, SeMI2024-77, pp. 88-93, 2025年1月. |
| 資料番号 |
SeMI2024-77 |
| 発行日 |
2025-01-16 (SeMI) |
| ISSN |
Online edition: ISSN 2432-6380 |
著作権に ついて |
技術研究報告に掲載された論文の著作権は電子情報通信学会に帰属します.(許諾番号:10GA0019/12GB0052/13GB0056/17GB0034/18GB0034) |
| PDFダウンロード |
SeMI2024-77 |
|