Parameter Efficient Transfer Learning for Suicide Attempt and Ideation Detection

Bhanu Pratap Singh Rawat, Hong Yu


Abstract
Pre-trained language models (LMs) have been deployed as the state-of-the-art natural language processing (NLP) approaches for multiple clinical applications. Model generalisability is important in clinical domain due to the low available resources. In this study, we evaluated transfer learning techniques for an important clinical application: detecting suicide attempt (SA) and suicide ideation (SI) in electronic health records (EHRs). Using the annotation guideline provided by the authors of ScAN, we annotated two EHR datasets from different hospitals. We then fine-tuned ScANER, a publicly available SA and SI detection model, to evaluate five different parameter efficient transfer learning techniques, such as adapter-based learning and soft-prompt tuning, on the two datasets. Without any fine-tuning, ScANER achieve macro F1-scores of 0.85 and 0.87 for SA and SI evidence detection across the two datasets. We observed that by fine-tuning less than ~2% of ScANER’s parameters, we were able to further improve the macro F1-score for SA-SI evidence detection by 3% and 5% for the two EHR datasets. Our results show that parameter-efficient transfer learning methods can help improve the performance of publicly available clinical models on new hospital datasets with few annotations.
Anthology ID:
2022.louhi-1.13
Volume:
Proceedings of the 13th International Workshop on Health Text Mining and Information Analysis (LOUHI)
Month:
December
Year:
2022
Address:
Abu Dhabi, United Arab Emirates (Hybrid)
Editors:
Alberto Lavelli, Eben Holderness, Antonio Jimeno Yepes, Anne-Lyse Minard, James Pustejovsky, Fabio Rinaldi
Venue:
Louhi
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
108–115
Language:
URL:
https://aclanthology.org/2022.louhi-1.13
DOI:
10.18653/v1/2022.louhi-1.13
Bibkey:
Cite (ACL):
Bhanu Pratap Singh Rawat and Hong Yu. 2022. Parameter Efficient Transfer Learning for Suicide Attempt and Ideation Detection. In Proceedings of the 13th International Workshop on Health Text Mining and Information Analysis (LOUHI), pages 108–115, Abu Dhabi, United Arab Emirates (Hybrid). Association for Computational Linguistics.
Cite (Informal):
Parameter Efficient Transfer Learning for Suicide Attempt and Ideation Detection (Singh Rawat & Yu, Louhi 2022)
Copy Citation:
PDF:
https://aclanthology.org/2022.louhi-1.13.pdf
Video:
 https://aclanthology.org/2022.louhi-1.13.mp4