Application of Large Language Model and In-Context Learning for Aviation Safety Prediction
The analysis of aviation safety reports using natural language processing (NLP) techniques is well-explored, but leveraging large language models (LLMs) in this domain remains under investigation. Recent studies have shown that LLMs, equipped with tens of billions of parameters and pre-trained on massive and diverse datasets, have the ability for in-context learning (ICL). This capability allows them to learn from prompts consisting of specific instructions and a few labeled examples without requiring model fine-tuning. This paper explores the use of LLMs to classify aviation safety reports through a case study focusing on domain-specific data, utilizing ICL and prompt engineering. Using BM25 to select eight examples for the prompt resulted in the highest performance, with an accuracy of 80.24% and an F1-score of 84.15%, demonstrating the LLM's generalization ability on domain-specific data through an engineered prompt.
Authors
- Donghui Shi (ORCID: https://orcid.org/0000-0001-5301-368X)
- Jozef Zurada (ORCID: https://orcid.org/0000-0003-1633-3152)
- Jian Guan (ORCID: https://orcid.org/0000-0002-0945-1081)
- Yuchen Yang (ORCID: https://orcid.org/0000-0001-9004-8638)
Institutions
- Anhui Jianzhu University (CN)
- University of Louisville (US)
Publication Details
- Published
- 2024-09-13
- DOI
- https://doi.org/10.1109/icacte62428.2024.10871713
- Citations
- 2
- Primary Topic
- Risk and Safety Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.92