1. Introduction
The reliable operation of high-voltage (HV) electrical equipment, particularly power transformers, is essential for the continuity and stability of electrical grids and transmission networks [
1]. The performance and service life of transformers depend strongly on the condition of their insulation systems, which deteriorate progressively during operation [
2,
3]. Insulation degradation is influenced by several interacting factors, including ageing, operating conditions, chemical processes, and internal mechanical and electrical stresses [
4]. As this degradation progresses, the probability of dielectric defects and insulation failure increases, making accurate condition monitoring a critical requirement for transformer asset management [
5,
6]. One of the major manifestations of insulation deterioration in transformers is partial discharge (PD) [
7]. PD is a localized electrical discharge that occurs within a small portion of an insulation system without completely bridging the electrodes [
8]. Although initially limited in extent, repeated PD activity gradually weakens the dielectric structure and may eventually lead to severe insulation failure if it is not detected and mitigated in time [
9,
10]. In practice, PD may occur in several forms, including surface discharge, corona discharge, internal or void discharge, and treeing, depending on the nature of the insulation defect and the dielectric medium involved [
11]. These discharge mechanisms may develop in air, liquid, or solid insulation systems, and each produces different electrical signatures and different degrees of insulation damage.
The various forms of PD generate distinct features and release different levels of energy, which in turn reflect the severity of the underlying insulation defect. The detection and interpretation of these features can therefore provide useful information for identifying fault conditions and supporting maintenance decisions in transformer operation [
12].
PD activity can be detected using several measurement approaches, depending on the insulation system, operating environment, and required sensitivity. The conventional electrical detection method standardized in IEC 60270 [
13] remains one of the most widely adopted approaches for transformer PD monitoring because of its high sensitivity and established calibration procedures. However, alternative techniques such as ultra-high-frequency (UHF) sensing, acoustic-emission (AE) monitoring, optical methods, and radio-frequency (RF) measurements have also received considerable research attention for detecting and characterizing discharge phenomena in high-voltage systems and related electrical applications [
14,
15,
16]. In recent years, RF-based discharge monitoring has demonstrated strong capability for identifying spark-discharge behaviour and operating states in electrical discharge machining (EDM) systems. Research has shown that RF signatures can effectively distinguish different discharge conditions and machining states, thereby improving process stability and efficiency [
17,
18,
19]. These findings are relevant to PD diagnostics because both EDM spark discharges and transformer PD events involve transient electrical discharge phenomena with characteristic electromagnetic signatures. Consequently, recent advances in RF-based discharge analysis and intelligent classification provide additional motivation for the development of machine-learning-assisted PD monitoring frameworks.
However, the relationship between PD type, measured signal characteristics, and discharge severity is often complex. In many cases, the separation between severity classes is not visually obvious, particularly when intermediate operating conditions are considered. As a result, conventional threshold-based or visually interpreted methods may be insufficient for robust PD diagnosis. To address this limitation, machine learning techniques have increasingly been applied to PD analysis and classification [
20]. These methods are well suited to modelling the nonlinear relationships that often exist between measured variables and insulation condition states. Among them, Artificial Neural Networks (ANNs) have attracted particular attention because of their strong pattern-recognition capability and their ability to classify complex electrical fault signatures with high accuracy [
19,
20]. Nevertheless, ANN-based classifiers are sensitive to the representation of the input space. When trained directly on limited raw measurements, important discriminatory information may remain underrepresented. Conversely, when many engineered features are introduced, feature redundancy and inter-feature correlation may increase model complexity and lead to overfitting, particularly when the dataset is limited or imbalanced [
20]. Feature engineering offers one way to improve class separability by constructing physically meaningful variables from the measured data. In PD applications, such variables can reflect energy-related behaviour, voltage stress, relative discharge intensity, and scale-normalized characteristics. However, an expanded feature space also introduces strong statistical dependence between variables derived from the same underlying measurements. Under these conditions, dimensionality-reduction methods become necessary to retain the informative structure of the data while reducing redundancy. Principal Component Analysis (PCA) is particularly suitable for this purpose because it transforms correlated variables into a smaller number of orthogonal components that capture the dominant variance structure of the dataset. In this study, a PD severity classification framework is proposed for transformer condition assessment using physics-informed feature engineering, PCA, and an ANN classifier. PD measurements were collected from a physical transformer using the IEC 60270 electrical measurement method, and the resulting dataset was grouped into four severity classes: normal PD, low PD, medium PD, and high PD. Starting from the two primary measured variables, namely apparent discharge magnitude and applied voltage, additional derived features were constructed to better represent the physical behaviour of the discharge process. PCA was then applied to obtain a compact set of informative features, which were subsequently used as inputs to a multilayer perceptron classifier. The proposed framework was further evaluated against alternative classification approaches in order to assess its diagnostic effectiveness.
The main contribution of this study is the development of a compact and physically grounded framework for transformer PD severity classification that integrates feature engineering, dimensionality reduction, and neural-network-based prediction. Unlike approaches that rely only on the raw measured variables, the proposed method constructs a richer feature space from IEC 60270-aligned PD data and then compresses that space before classification. This allows the model to retain the discriminatory information contained in the engineered features while avoiding the penalty of excessive dimensionality. The novelty of the study lies in three aspects. First, it formulates a physics-informed feature space for PD severity analysis using derived descriptors based on apparent charge magnitude and applied voltage. Second, it applies PCA specifically as a feature-compression stage for transformer PD severity classification, thereby reducing feature redundancy before ANN training. Third, it demonstrates that a reduced number of principal components can preserve the dominant diagnostic information required for accurate four-class severity classification. In this way, the study contributes not only a classification model, but also to a practical methodological pathway for combining physically meaningful PD descriptors with low-dimensional machine-learning representations.
The remainder of this manuscript is organized as follows.
Section 2 describes the dataset, class definitions, and the feature-engineering procedure.
Section 3 presents the proposed methodology, including correlation analysis, PCA, and the ANN architecture.
Section 4 discusses the classification results, including dimensionality-reduction performance, confusion-matrix analysis, per-class evaluation, and benchmarking against alternative classifiers.
Section 5 concludes the paper by summarizing the main findings, providing recommendations, and outlining directions for future research.
4. Results and Discussion
4.1. Effect of Dimensionality Reduction Ratio
Table 3 presents the training and testing accuracy obtained for all retained principal-component dimensions, with
varying from 1 to 10. The best overall test performance was achieved at
, corresponding to a dimensionality-reduction ratio of
, where the model reached a test accuracy of 98.3%. This indicates that two principal components were sufficient to preserve the dominant discriminatory information required for PD severity classification. Although several higher-dimensional configurations also achieved the same test accuracy, the
model is preferred because it provides the most compact representation of the engineered feature space while maintaining optimal predictive performance. By contrast, the
configuration produced a lower test accuracy of 93.2%, indicating that a single principal component did not retain enough information for robust class separation. These results show that a small increase from one to two retained components yields a substantial improvement in classification performance, whereas further increases in dimensionality provide little or no additional benefit. The results in
Table 3 are illustrated in
Figure 10, which plots training and testing accuracy as a function of the dimensionality-reduction ratio. The figure shows that training accuracy reaches 100.0% from
onward, while testing accuracy stabilizes at 98.3% for several configurations.
This pattern suggests that the essential class structure of the dataset is already captured in the first two principal components, and that retaining additional components does not materially improve generalization performance. Accordingly, was selected as the optimal PCA configuration for the subsequent analyses.
4.2. Confusion Matrix and Per-Condition Analysis
Figure 11 presents the confusion matrix for the optimal PCA-ANN model obtained at
. The matrix shows that the classifier achieved perfect recognition of the High PD and Medium PD classes on the test set, with all 46 High PD samples and both Medium PD samples classified correctly. This result is particularly important because these classes represent the more severe insulation conditions and are therefore of greatest operational concern.
For the Low PD class, the model correctly classified 9 of the 9 test samples, corresponding to a recall of 100. In the Normal Operation class, both test samples were correctly identified, yielding a recall of 100.0%. Overall, the confusion matrix indicates that the proposed model is highly effective in distinguishing the critical medium- and high-severity PD states, while the only residual ambiguity occurs at the boundary between Normal Operation and Low PD Activity.
4.3. Measures to Prevent Overfitting and Stability Validation
To guarantee robust model generalization and eliminate the risk of high-variance noise memorization, several distinct anti-overfitting mechanisms were integrated into the design. The primary structural safeguard against overfitting is the deployment of PCA feature compression immediately prior to the introduction of the neural network. By mathematically transforming a redundant, highly correlated 10-dimensional engineered feature space into two orthogonal principal components, the network’s structural input dimension was reduced by 80%. This compression significantly simplified the learning task and prevented the network from overfitting to high-frequency experimental noise. Furthermore, the internal volume of the network was kept intentionally compact with a single hidden layer, restricting the model’s total trainable weight parameters to match the compressed input space. The validation stability of this optimized network configuration was carefully tested using a 5-fold stratified cross-validation protocol. The proposed PCA-ANN model achieved a mean cross-validation accuracy of 98.3%, demonstrating exceptional predictive consistency. Critically, the cross-validation process yielded a remarkably narrow standard deviation of just ±0.9%. This tight standard deviation serves as definitive statistical proof of model stability, demonstrating that the network’s generalization performance is entirely invariant to specific data partitions and is free from overfitting artifacts, representing the lowest performance variance among all evaluated machine learning architectures.
4.4. Handling Dataset Class Imbalance
The partial discharge dataset exhibits a pronounced class imbalance, with the High PD Activity class dominating approximately 78.2% of the total observations, while Normal PD (3.1%) and Medium PD (3.7%) classes are sparsely represented. In standard machine learning frameworks, such severe imbalance typically causes the classifier to optimize for the majority class, leading to a drastic deterioration in minority class predictive accuracy. In this study, this limitation was resolved entirely at the data-representation level through physics-informed feature engineering rather than empirical data resampling. By mapping the raw measurements (|Q|, V) into physics-grounded nonlinear transformations, the underlying multi-class boundaries were significantly expanded. This feature enhancement achieved complete geometric isolation and distinct structural separation of the minority classes within the higher-dimensional space prior to classification. The empirical validation of this feature-driven separability is demonstrated by the model’s exceptional per-class metrics. Despite their minor representation, the minority Medium PD class achieved an absolute 100.0% Precision, 100.0% Recall, and 100.0% F1-score on the independent test set, while the Normal PD class reached 100.0% Recall. The network learned the definitive physical signatures of these transient states without being overwhelmed by the dominant class. Consequently, standard synthetic data generation techniques (such as SMOTE) or undersampling strategies were intentionally avoided. Unconstrained data augmentation risks creating unphysical feature correlations that violate real-world high-voltage transformer physics governed by IEC 60270 standards, as well as introducing data leakage into validation loops. Locking the exact class ratios via stratified splits ensured that all validation metrics reflect authentic structural classification performance.
4.5. Per-Class Performance Metrics
Table 4 and
Figure 12 summarize the per-class precision, recall, and F1-score for the optimal PCA-ANN model. The results confirm the strong performance observed in the confusion matrix analysis. Medium PD Activity and High PD Activity both achieved precision, recall, and F1-score values of 100.0%, indicating complete separation of the most critical PD severity classes in the test set.
Low PD Activity achieved a precision of 100.0%, a recall of 88.9%, and an F1-score of 94.1%, reflecting the single Low PD sample that was assigned to the Normal class. Normal Operation achieved a recall of 100.0% but a lower precision of 67.0%, again due to the same misclassification at the Normal-Low PD boundary. The weighted-average precision, recall, and F1-score were 99.3%, 98.3%, and 98.6%, respectively, demonstrating strong overall classification performance.
These results suggest that the proposed PCA-ANN framework is highly reliable for identifying the more severe PD conditions, while the remaining classification difficulty is concentrated in the lower-severity region, where discharge signatures are inherently weaker and more similar to background or early-stage activity. This behaviour is consistent with the raw-data and engineered-feature distributions discussed earlier.
4.6. Comparison with Alternative Feature Configurations
Table 5 compares the performance of the ANN classifier under three input-feature configurations: (i) the raw measurement pair
, (ii) the full set of 10 engineered features without PCA, and (iii) the proposed PCA-ANN configuration using the optimal reduced representation with
. The purpose of this comparison is to evaluate the effect of feature engineering and dimensionality reduction on classification performance. The results show that the raw-feature ANN produced the lowest performance, with a training accuracy of 96.6% and a test accuracy of 91.5%. When the ANN was trained on the full 10-dimensional engineered feature space, performance improved substantially to 100.0% training accuracy and 98.3% test accuracy. The proposed PCA-ANN model achieved the same test accuracy of 98.3% while using only two input dimensions. This demonstrates that PCA preserved the dominant discriminatory information contained in the engineered feature space while reducing the dimensionality of the classifier input.
Figure 13 illustrates these results graphically. The comparison confirms that feature engineering plays a major role in improving PD severity classification relative to the raw measurements alone.
At the same time, the PCA stage provides a more compact representation of the engineered feature set without any loss in predictive performance. Therefore, the proposed PCA-ANN configuration offers the most efficient trade-off between model compactness and classification accuracy among the three alternatives considered.
4.7. Multi-Classifier Benchmarking
Table 6 compares the proposed PCA-ANN model with five alternative classifiers, namely Gaussian Naïve Bayes (GNB), K-Nearest Neighbours (KNN) with
and
, Support Vector Machine (SVM) with radial basis function kernel, and an ANN trained without PCA. The benchmark models were evaluated using the full 10-dimensional engineered feature space, whereas the proposed model used only the two retained principal components. In addition to training and test accuracy, 5-fold stratified cross-validation was used to assess robustness and variability in model performance. The proposed PCA-ANN model achieved a training accuracy of 100.0% and a test accuracy of 98.3%, matching the best held-out test performance among all evaluated models. It also produced the highest mean cross-validation accuracy, 98.3%, together with the lowest standard deviation, ±0.9%, indicating the most stable performance across folds. The ANN without PCA achieved the same training and test accuracy, but its mean cross-validation accuracy was slightly lower, at 97.9%, with a larger standard deviation of ±1.3%. Among the non-ANN benchmark methods, GNB achieved the highest mean cross-validation accuracy at 97.0%, followed by KNN at 96.6% and SVM at 96.2%.
Figure 14 provides a visual comparison of training accuracy, test accuracy, and cross-validation performance for all six classifiers. Overall, the results indicate that the proposed PCA-ANN framework achieves competitive or superior classification performance while using a substantially reduced input representation.
This is an important outcome, as it shows that the compressed PCA feature space is sufficient not only for accurate classification, but also for consistent generalization across different data partitions.
4.8. Discussion
The results of this study confirm that partial discharge severity classification benefits substantially from combining physics-informed feature engineering with dimensionality reduction prior to neural-network classification. From a diagnostic perspective, this outcome is consistent with the established understanding that PD behaviour is governed by the interaction between discharge magnitude, electric-field stress, and insulation condition, rather than by any single raw measured variable alone [
8,
9,
10,
11]. In the present study, the use of derived features based on apparent charge magnitude and applied voltage improved class separability relative to the raw measurement pair, which explains why the ANN trained on engineered features outperformed the ANN trained directly on
. This result is in line with prior work showing that physically meaningful representations can improve the diagnostic value of transformer condition-monitoring data by exposing relationships that are not sufficiently visible in the original measurement space [
2,
3,
4,
12].
The strong inter-feature correlation observed among the engineered variables also supports the use of PCA as a rational preprocessing stage. Since several of the constructed descriptors were different transformations of the same underlying physical quantities, redundancy in the feature space was expected. PCA is specifically intended to address this problem by projecting correlated variables onto a smaller set of orthogonal components that retain the dominant variance structure of the data [
5]. The finding that the first two principal components captured more than 90% of the cumulative variance, while still supporting 98.3% test accuracy, therefore agrees with the theoretical role of PCA as an information-preserving compression method. In practical terms, the result suggests that the dominant discriminatory structure of the PD dataset is largely governed by two underlying factors associated with discharge magnitude and applied-voltage behaviour, which were successfully preserved in the reduced feature space.
The near-equivalence in performance between the 10-feature ANN and the proposed PCA-ANN model is particularly important. Rather than merely improving accuracy, PCA reduced the dimensionality of the classifier input from 10 variables to 2 retained components without loss of predictive performance. This indicates that the benefit of the engineered feature space lies not only in its richness, but also in the latent structure that it reveals after compression. Similar findings have been reported in transformer fault diagnosis studies where PCA was used to simplify multidimensional diagnostic data while maintaining or improving ANN performance [
2,
3,
4]. In that sense, the present results extend the value of PCA-ANN integration from dissolved-gas-analysis-oriented diagnosis to IEC 60270-aligned PD severity classification.
The class-wise results are also technically meaningful. The model achieved perfect precision and recall for the Medium PD and High PD classes on the test set, indicating that the proposed feature representation was particularly effective in separating the most operationally significant insulation states. This is consistent with the broader PD literature, which shows that more severe discharge conditions generally produce stronger and more distinguishable electrical signatures due to higher apparent charge levels, more pronounced field effects, and greater insulation disturbance [
8,
9,
11]. By contrast, the only residual ambiguity in the present study occurred at the boundary between Normal Operation and Low PD Activity. This behaviour is expected because low-level PD signatures are often weak, overlap with background activity, and are more difficult to distinguish reliably using magnitude-based measurements alone [
11,
12]. The single Low PD sample misclassified as Normal Operation is therefore not anomalous, but rather reflects a recognized challenge in practical PD diagnosis.
The benchmark comparison against GNB, KNN, SVM, and ANN without PCA further reinforces the validity of the proposed framework. The PCA-ANN model matched the best held-out test accuracy and achieved the highest mean 5-fold cross-validation accuracy with the lowest standard deviation. This combination of high accuracy and low fold-to-fold variability suggests that the reduced PCA space was not only discriminative, but also more stable under repeated partitioning of the data. That observation is consistent with the machine-learning literature on overfitting, where redundant or highly correlated inputs may inflate model complexity without improving generalization [
15]. By compressing the feature space into orthogonal components, PCA likely reduced this sensitivity and enabled the ANN to learn a more robust decision function. The lower cross-validation variance of the PCA-ANN model relative to the ANN without PCA supports exactly this interpretation.
From a practical transformer-monitoring perspective, these findings are significant because they suggest that accurate PD severity classification does not require a large raw-input space, but rather an informative and physically grounded one. IEC 60270 remains the standard framework for PD electrical measurement [
11], and the present study shows that even within such a conventional measurement setting, enhanced diagnostic performance can be obtained when the raw measurements are transformed into physically meaningful descriptors and then rationalized through PCA. This aligns with the broader direction of transformer condition assessment research, which increasingly combines established measurement standards with intelligent diagnostic algorithms to improve fault recognition and maintenance decision support [
2,
3,
4,
8,
12].
At the same time, the discussion must acknowledge the limitations that frame these results. The dataset was strongly imbalanced, with High PD samples dominating the observations, while the Normal and Medium PD classes had very limited support. Although stratified splitting and cross-validation help to mitigate this issue, the small number of minority-class samples means that caution is required when interpreting the apparent perfection of some per-class metrics. This is particularly relevant for Medium PD, where the test support was only two samples. In addition, the current feature set was derived primarily from discharge magnitude and applied voltage and did not incorporate richer descriptors such as phase-resolved PD patterns, pulse repetition characteristics, or waveform-shape parameters, all of which are known to provide additional discriminatory information in PD diagnostics [
8,
9,
10,
11]. Accordingly, the present framework should be understood as a strong proof of concept rather than a final universal model.
5. Conclusions
This study developed a partial discharge severity classification framework for high-voltage transformers by combining physics-informed feature engineering, principal component analysis, and an artificial neural network. Starting from the measured partial discharge magnitude and applied voltage, the feature space was expanded to include derived descriptors associated with discharge energy, electrical stress, scaling, and relative intensity. PCA was then used to reduce redundancy in the engineered feature space and retain the most informative variation before classification. The results showed that the proposed PCA-ANN framework achieved 98.3% test accuracy using only two retained principal components, while maintaining the same performance as the ANN trained on all 10 engineered features and improving substantially on classification based only on the raw measurements. The model also achieved perfect precision and recall for the medium- and high-severity classes on the test set, indicating strong capability in identifying the more critical insulation conditions. Beyond accuracy alone, the findings show that the combination of domain-based feature design and dimensionality reduction can produce a compact and effective representation of PD behaviour. This is important because transformer condition-monitoring systems require classification methods that are not only accurate, but also computationally efficient and interpretable in relation to the physical meaning of the measured quantities. In this regard, the proposed approach demonstrates that a reduced feature representation can preserve diagnostic information while simplifying the learning task of the classifier. However, the results should be interpreted in light of the limitations of the study. The dataset comprised 294 samples and was strongly imbalanced, with the high-PD class dominating the observations, while the normal and medium classes had very limited representation. This means that although the classification results are promising, the reported performance may not fully capture the variability encountered in broader field conditions. In particular, the small support for some classes limits the strength of conclusions regarding generalization. In addition, the present study relied mainly on magnitude- and voltage-derived descriptors and did not incorporate richer phase-resolved discharge pattern information, which may further improve discrimination between neighbouring classes such as normal and low PD.
From a practical perspective, the study recommends that transformer PD severity assessment should not rely solely on raw apparent charge and voltage measurements when machine-learning models are used. Instead, feature engineering grounded in discharge physics should be incorporated to improve class separability before model training. The study further recommends the use of dimensionality-reduction techniques such as PCA when strong inter-feature correlation is present, since this can reduce computational burden without degrading diagnostic performance. For utilities and maintenance practitioners, the proposed framework may therefore serve as a useful basis for decision-support tools aimed at prioritizing transformers with elevated PD severity for further inspection or intervention.
Future work should focus on validating the proposed framework on larger and more diverse datasets obtained from multiple transformers, insulation conditions, and operating environments. Class imbalance should be addressed using resampling or data augmentation strategies, including methods such as SMOTE, to improve learning stability for underrepresented classes. Further studies should also integrate phase-resolved partial discharge features and compare deep-learning and hybrid diagnostic models against the present PCA-ANN approach. Finally, external validation under realistic online monitoring conditions will be necessary to assess the robustness, transferability, and deployment readiness of the method in practical transformer health-monitoring applications.