MIKA_BERT_FMEA_NER / README.md
sequoiaandrade's picture
updated examples
5069c85
metadata
license: other
language:
  - en
pipeline_tag: token-classification
tags:
  - token-classification
  - NER
library_name: transformers
widget:
  - text: >-
      While flying a fire, the UAS experienced an issue of unknown sorts and
      crashed to the ground. From the people watching the aircraft near the
      fire, they seem to think it was some sort of motor failure due to no more
      noise coming from the aircraft and it falling straight to the ground.
    example_title: Example 1
  - text: >-
      During a pre-flight engine run-up, a battery hatch cover disengaged from
      the fuselage and hit one of the vertical takeoff and landing {VTOL}
      propellers. The motor failsafe activated and the motors shut down.
    example_title: Example 2
  - text: >-
      UAS was climbing to 11,000 ft. msl on a reconnaissance mission when it
      experienced a rapid and uncommanded descent. The pilot took no action but
      monitored instruments until the aircraft regained a stable profile. 
    example_title: Example 3

Manager for Intelligent Knowledge Access (MIKA)

Custom Named-Entity Recognition (NER) for Failure Modes and Effects Analysis (FMEA)

base-bert-uncased model first further pre-trained then fine-tuned for custom NER to extract failure-relevant entities from incident and accident reports. The model was trained on manually annotated NASA LLIS reports and evaluated on SAFECOM reports.

NER model training was for 4 epochs with:BertForTokenClassification.from_pretrained , learning_rate=2e-5, weight_decay=0.01,

The model was trained to identify the following long-tailed entities:

  • CAU: failure cause
  • MOD: failure mode
  • EFF: failure effect
  • CON: control process
  • REC: recommendations

Performace:

Entity Precision Recall F-1 Support
CAU 0.31 0.19 0.23 1634
CON 0.49 0.34 0.40 3859
EFF 0.45 0.20 0.28 1959
MOD 0.19 0.52 0.28 594
REC 0.30 0.59 0.40 954
Average 0.41 0.32 0.33 9000

More infomation on training data, evaluation, and intended use can be found in the original publication

Citation: S. R. Andrade and H. S. Walsh, "What Went Wrong: A Survey of Wildfire UAS Mishaps through Named Entity Recognition," 2022 IEEE/AIAA 41st Digital Avionics Systems Conference (DASC), Portsmouth, VA, USA, 2022, pp. 1-10, doi: 10.1109/DASC55683.2022.9925798. https://ieeexplore.ieee.org/abstract/document/9925798


Notices:

Copyright © 2023 United States Government as represented by the Administrator of the National Aeronautics and Space Administration. All Rights Reserved.

Disclaimers

No Warranty: THE SUBJECT SOFTWARE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY OF ANY KIND, EITHER EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTY THAT THE SUBJECT SOFTWARE WILL CONFORM TO SPECIFICATIONS, ANY IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR FREEDOM FROM INFRINGEMENT, ANY WARRANTY THAT THE SUBJECT SOFTWARE WILL BE ERROR FREE, OR ANY WARRANTY THAT DOCUMENTATION, IF PROVIDED, WILL CONFORM TO THE SUBJECT SOFTWARE. THIS AGREEMENT DOES NOT, IN ANY MANNER, CONSTITUTE AN ENDORSEMENT BY GOVERNMENT AGENCY OR ANY PRIOR RECIPIENT OF ANY RESULTS, RESULTING DESIGNS, HARDWARE, SOFTWARE PRODUCTS OR ANY OTHER APPLICATIONS RESULTING FROM USE OF THE SUBJECT SOFTWARE. FURTHER, GOVERNMENT AGENCY DISCLAIMS ALL WARRANTIES AND LIABILITIES REGARDING THIRD-PARTY SOFTWARE, IF PRESENT IN THE ORIGINAL SOFTWARE, AND DISTRIBUTES IT "AS IS."

Waiver and Indemnity: RECIPIENT AGREES TO WAIVE ANY AND ALL CLAIMS AGAINST THE UNITED STATES GOVERNMENT, ITS CONTRACTORS AND SUBCONTRACTORS, AS WELL AS ANY PRIOR RECIPIENT. IF RECIPIENT'S USE OF THE SUBJECT SOFTWARE RESULTS IN ANY LIABILITIES, DEMANDS, DAMAGES, EXPENSES OR LOSSES ARISING FROM SUCH USE, INCLUDING ANY DAMAGES FROM PRODUCTS BASED ON, OR RESULTING FROM, RECIPIENT'S USE OF THE SUBJECT SOFTWARE, RECIPIENT SHALL INDEMNIFY AND HOLD HARMLESS THE UNITED STATES GOVERNMENT, ITS CONTRACTORS AND SUBCONTRACTORS, AS WELL AS ANY PRIOR RECIPIENT, TO THE EXTENT PERMITTED BY LAW. RECIPIENT'S SOLE REMEDY FOR ANY SUCH MATTER SHALL BE THE IMMEDIATE, UNILATERAL TERMINATION OF THIS AGREEMENT.