Improved Firefly Algorithm-Optimized ResNet18 for Partial Discharge Pattern Recognition Within Small-Sample Scenarios

The growing complexity of electrical infrastructure has elevated partial discharge (PD) detection to a crucial methodology for ensuring power system safety. Current PD pattern recognition approaches encounter persistent challenges in low-data scenarios, particularly regarding classification accuracy and model generalizability. This study develops a Firefly Algorithm with a Black Hole Mechanism-ResNet18 (FBH-ResNet18) framework that synergistically integrates the Firefly Algorithm with the Black Hole Mechanism (FBH algorithm) optimization with residual neural networks for PD signal classification using phase-resolved partial discharge (PRPD) mappings. A dedicated experimental platform first acquires PD signals through UHF sensors, which are subsequently converted into two-dimensional PRPD representations. The FBH algorithm systematically optimizes four key hyperparameters within the ResNet18 architecture during network training. The Black Hole Mechanism and improved population dynamics enhance optimization efficiency, resulting in more accurate hyperparameter tuning and improved model performance. Comparative evaluations demonstrate the enhanced performance of this parameter-optimized model against alternative configurations. Experimental results indicate that the improved ResNet18 achieves fast convergence and strong generalization on small-sample datasets, significantly enhancing recognition accuracy. During the first 80 generations of training, the classification accuracy reaches 89.11%, and in the final iteration, the model’s recognition accuracy increases to 92.55%, outperforming other models with accuracies generally below 90%. Additionally, the model shows excellent performance on the test set, with a loss function value of 0.250785, significantly lower than that of other models, indicating superior performance on small sample datasets. This research provides an effective solution for power cable fault diagnosis, offering high practical value.