Research Article
Center Loss Guided Prototypical Networks for Unbalance Few-Shot Industrial Fault Diagnosis
Table 5
Quantitative comparison of models in different groups.
| | Methods | Plan 1 | Plan 2 | Plan 3 | Plan 4 | | ACC (%) | F1 (%) | ACC (%) | F1 (%) | ACC (%) | F1 (%) | ACC (%) | F1 (%) |
| | MAML [10] | 33.3 | 46.2 | 43.4 | 41.6 | 44.6 | 50.3 | 38.9 | 41.4 | | MAML + center loss | 53.9 | 66.6 | 56.3 | 61.7 | 52.9 | 68.2 | 49.2 | 56.3 | | Prototypical networks [38] | 46.1 | 51.7 | 43.2 | 41.5 | 47.3 | 47.7 | 42.7 | 52.6 | | Prototypical networks + focal loss | 80.3 | 81.6 | 71.5 | 70.2 | 70.2 | 71.6 | 81.7 | 82.8 | | Prototypical networks + center loss | 91.7 | 97.2 | 84.6 | 83.9 | 78.1 | 78.1 | 79.9 | 83.3 |
|
|