Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Detection of Suicidal Risk on Social Media: A Hybrid Model

Created by
  • Haebom

저자

Zaihan Yang, Ryan Leonard, Hien Tran, Rory Driscoll, Chadbourne Davis

개요

본 논문은 Reddit 게시물을 이용하여 자살 위험도를 네 가지 수준으로 자동 분류하는 강력한 머신러닝 모델을 개발합니다. 최첨단 심층 학습 변환기 모델인 RoBERTa의 심층 문맥 임베딩과 TF-IDF의 통계적 용어 가중치를 PCA로 압축하여 결합한 RoBERTa-TF-IDF-PCA 하이브리드 모델을 제안합니다. 데이터 불균형 및 과적합 문제를 해결하기 위해 다양한 데이터 재샘플링 기법과 데이터 증강 전략을 활용하여 모델의 일반화 성능을 향상시켰으며, RoBERTa 단독, BERT 모델 및 기타 기존 머신러닝 분류기와 성능을 비교 분석했습니다. 실험 결과, 하이브리드 모델이 향상된 성능을 달성하여 최고 가중치 $F_{1}$ 점수 0.7512를 기록했습니다.

시사점, 한계점

시사점:
Reddit 게시물을 이용한 자살 위험도 자동 분류 모델 개발을 통해 조기 위험 감지 가능성 제시.
RoBERTa-TF-IDF-PCA 하이브리드 모델이 기존 모델 대비 향상된 성능을 보임.
데이터 불균형 및 과적합 문제 해결을 위한 다양한 기법 적용 및 효과 검증.
한계점:
모델의 일반화 성능 향상을 위한 추가적인 연구 필요. (데이터 증강 및 재샘플링 기법의 한계점)
실제 자살 위험도 평가와의 정확도 차이에 대한 추가 검증 필요.
$F_1$ score 0.7512는 완벽한 성능이 아니므로, 오류로 인한 위험 관리 방안 마련 필요.
Reddit 데이터의 편향성 및 대표성 문제 고려 필요.
👍