Fine-Tuning Large Language Models for Environmental Health and Safety Applications
This study aims to simplify Environmental Health and Safety (EHS) by leveraging the power of Large Language Models (LLMs). In this research, we focus on fine-tuning three LLMs — Llama, Mistral, and Falcon — using PEFT techniques such as QLoRA and SFT, to address domain-specific needs such as safety compliance, incident reporting, and knowledge dissemination. Our research methodology involves fine-tuning each LLM model on a custom dataset compiled from various regulatory agencies, supplemented by targeted web scraping and manual collection of questionnaires to capture and enrich the models with the latest regulations and guidelines. This study aims to compare the effectiveness of these fine-tuned models to identify the most effective model and fine-tuning techniques for specific EHS applications. We found that Mistral outperformed Falcon and Llama on ROUGE, BERTscore, and a manual evaluation.
Authors
- Genya Ishigaki (ORCID: https://orcid.org/0000-0003-3655-7532)
- William Andreopoulos (ORCID: https://orcid.org/0000-0001-9097-1123)
- Mohammad Adil Ansari
Institutions
- San Jose State University (US)
Publication Details
- Published
- 2024-09-30
- DOI
- https://doi.org/10.1109/aixset62544.2024.00012
- Citations
- 1
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 0.35