主管:中华人民共和国应急管理部
主办:应急管理部天津消防研究所
ISSN 1009-0029  CN 12-1311/TU

Fire Science and Technology ›› 2023, Vol. 42 ›› Issue (11): 1529-1534.

Previous Articles     Next Articles

Fire accident case named entity recognition based on BERT-CRF model

Guan Siqi1,2,3, Dong Tingting1,2,3, Wan Zijing1,2,3, He Yuansheng1,2,3   

  1. (1. Tianjin Fire Science and Technology Research Institute of MEM, Tianjin 300381, China; 2. Laboratory of Fire Protection Technology for Industry and Public Building, Ministry of Emergency Management, Tianjin 300381, China; 3. Tianjin Key Laboratory of Fire Safety Technology, Tianjin 300381, China)
  • Online:2023-11-15 Published:2023-11-15

Abstract: Aiming at the key information extraction from fire accident investigation files, we propose a BERT-CRF based named entity recognition method for obtaining information such as accident location, cause and effect, safety measure, etc. We firstly construct a fire accident text corpus by annotating 161 accident reports and using a data augmentation method on the labeled data. Then according to BERT pre?training model, the bidirectional feature extraction method is performed on the sentence sequences in the corpus. Extracted information of the accident text context is further predicted as the key entities by CRF model and the entity label transfer rules. Experiments show that the precision, recall and F1 values of the BERT-CRF model in the fire accident case named entity recognition task are 76.36%, 86.19%, and 80.97%, respectively, which are better than BERT and BERT-BiLSTM-CRF models, and the training time is 61 seconds shorter than that of model BERT-BiLSTM-CRF. Our final model can provide accurate entity construction services for downstream systems such as fire investigation knowledge base and file compilation.

Key words: named entity recognition, BERT-CRF, fire accident, fire information, fire investigation file, text corpus, fire accident text