Suicide Phenotyping from Clinical Notes in Safety-Net Psychiatric Hospital Using Multi-Label Classification with Pre-Trained Language Models
Accurate identification and categorization of suicidal events can yield better suicide precautions, reducing operational burden, and improving care quality in high-acuity psychiatric settings. Pre-trained language models offer promise for identifying suicidality from unstructured clinical narratives. We evaluated the performance of four BERT-based models using two fine-tuning strategies (multiple single-label and single multi-label) for detecting coexisting suicidal events from 500 annotated psychiatric evaluation notes. The notes were labeled for suicidal ideation (SI), suicide attempts (SA), exposure to suicide (ES), and non-suicidal self-injury (NSSI). RoBERTa outperformed other models using binary relevance (acc=0.86, F1=0.78). MentalBERT (F1=0.74) also exceeded BioClinicalBERT (F1=0.72). RoBERTa fine-tuned with a single multi-label classifier further improved performance (acc=0.88, F1=0.81), highlighting that models pre-trained on domain-relevant data and the single multi-label classification strategy enhance efficiency and performance.
Authors
- Salih Selek (ORCID: https://orcid.org/0000-0001-5197-5682)
- Ming Huang (ORCID: https://orcid.org/0000-0001-7367-3626)
- Zehan Li (ORCID: https://orcid.org/0000-0002-2955-6809)
- Scott D. Lane
- Lokesh Shahani (ORCID: https://orcid.org/0000-0002-6731-6275)
- Jair C. Soares (ORCID: https://orcid.org/0000-0002-5466-5628)
- Hua Xu (ORCID: https://orcid.org/0000-0002-5274-4672)
- Yan Hu (ORCID: https://orcid.org/0009-0008-2413-5918)
- Hongfang Liu (ORCID: https://orcid.org/0000-0003-2570-3741)
- Rodrigo Machado-Vieira
Institutions
- Yale University (US)
Publication Details
- Journal
- PubMed
- Published
- 2024-09-27
- DOI
- https://doi.org/10.48550/arxiv.2409.18878
- Citations
- 3
- Primary Topic
- Suicide and Self-Harm Studies
- Type
- article
Funders
- National Institutes of Health
- U.S. National Library of Medicine