Logo image
Large Language Models for Failure Mode Classification: An Investigation
Preprint   Open access

Large Language Models for Failure Mode Classification: An Investigation

Michael Stewart, Melinda Hodkiewicz and Sirui Li
ArXiv.org
Cold Spring Harbor Laboratory
2023
pdf
Preprint279.64 kBDownloadView
CC BY V4.0 Open Access

Abstract

Technical Language Processing Large language model Natural Language Processing Failure Mode Maintenance
In this paper we present the first investigation into the effectiveness of Large Language Models (LLMs) for Failure Mode Classification (FMC). FMC, the task of automatically labelling an observation with a corresponding failure mode code, is a critical task in the maintenance domain as it reduces the need for reliability engineers to spend their time manually analysing work orders. We detail our approach to prompt engineering to enable an LLM to predict the failure mode of a given observation using a restricted code list. We demonstrate that the performance of a GPT-3.5 model (F1=0.80) fine-tuned on annotated data is a significant improvement over a currently available text classification model (F1=0.60) trained on the same annotated data set. The fine-tuned model also outperforms the out-of-the box GPT-3.5 (F1=0.46). This investigation reinforces the need for high quality fine-tuning data sets for domain-specific tasks using LLMs.

Details

Metrics

5 File views/ downloads
60 Record Views
Logo image