Lightweight Multi-Task CNN for Simultaneous IGBT Switch Aging Diagnosis Using CWT-Based RGB Images
Abstract
1. Introduction
- 1.
- Problem extension. A multi-task CNN framework for simultaneous per-switch IGBT aging diagnosis from CWT images, predicting six four-class labels from a single forward pass and extending the prior system-level framework [6] to per-device resolution; to the authors’ knowledge this combination has not been reported (Section 2.5).
- 2.
- CWT-MTNet architecture. A 172 K-parameter DS-Conv network that stays within 0.84 pp of full MobileNet-v2 (3.7 M) in mean accuracy across five independent partitions while using one twenty-first of its parameters and one third of its FLOPs—a 0.7 MB footprint (≈0.17 MB under 8-bit quantization), small enough to reside in the on-chip memory of an embedded inverter controller alongside existing control firmware.
- 3.
- Domain gap analysis. Experimental demonstration that ImageNet-pretrained networks are inferior to scratch-trained networks for CWT multi-task diagnosis, with three identified causes: domain mismatch, multi-task gradient conflict, and parameter excess.
- 4.
- Gradient conflict quantification. Ablation study showing that multi-task learning reduces ResNet-18 Mild recall by 42.1 pp on the BH switch, while DS-Conv architectures change by at most 3.3 pp in either direction.
- 5.
- Interpretability. Grad-CAM and confusion matrix analyses explaining the performance advantage of DS-Conv over standard convolution and identifying Mild-to-Healthy misclassification as the dominant safety risk for maintenance scheduling.
- 6.
- Representation and front-end characterization. A controlled ablation that holds the deviation signals fixed and varies only the encoding quantifies what the representation contributes: discarding phase costs 4.0 pp, of which the CWT decomposition returns 2.4 while making per-switch accuracy twelve times more uniform. The same comparison shows that the encoding is not a necessary condition on this benchmark. A companion noise study establishes that independent sensor noise at 0.5% of the phase RMS exceeds the aging signature sixfold and defeats every encoding examined, fixing the suppression that the acquisition front end must supply.
2. Related Work
2.1. Conventional Condition Monitoring: From Hardware Sensing to Scalar Feature Extraction
2.2. CWT-Based CNN for Inverter Fault Diagnosis
2.3. Multi-Task Learning and Lightweight Architecture Design
2.4. Grad-CAM for CNN Interpretability
2.5. Positioning of This Work
3. System Description and Dataset
3.1. System Overview
3.2. Inverter Model, Aging States, and CWT Image Generation
3.3. Multi-Task Dataset
4. CWT-MTNet Architecture
4.1. Design Principles
- 1.
- DS-Conv superiority in the CWT domain. Depthwise Separable Convolution (DS-Conv) factorizes a standard convolution into a channel-wise depthwise step and a pointwise step. For CWT RGB images, this factorization is physically well suited: the R, G, B channels encode three physically distinct symmetrical components (, , ), so processing each channel independently in the depthwise step preserves their physical identities before cross-channel fusion in the pointwise step. In contrast, a standard convolution mixes spatial and channel information simultaneously, blurring the component-wise aging signatures from the first layer onward. Preliminary experiments confirm that MobileNet-v2 (DS-Conv) consistently outperforms standard-convolution networks on per-switch Mild recall (Section 6.2), and Grad-CAM analysis (Section 6.10) shows that DS-Conv activates distributed regions across the entire time–frequency plane, whereas standard convolution concentrates activation in narrow horizontal bands.
- 2.
- Parameter excess in full MobileNet-v2. MobileNet-v2’s 3.7 M parameters were designed for 1000-class ImageNet recognition from natural photographic images. Its later inverted-residual blocks progressively widen the channel dimension to 320 and then 1280, capturing fine-grained texture and object detail that does not exist in CWT scalograms; on a 15,625-sample dataset with four aging classes per switch, these layers add capacity without adding discriminative content while enlarging the parameter surface over which the six heads compete. CWT-MTNet therefore retains the DS-Conv factorization and discards the late-stage width. The ablation of Section 6.6 is consistent with this choice: under the six-head operation, CWT-MTNet changes by to pp of Mild recall relative to its single-head counterpart—a range comparable to full MobileNet-v2 ( to pp) and far below the standard-convolution ResNet-18 ( to pp). The reduction does carry an accuracy cost (Section 6.4), but it is obtained at a twenty-one-fold smaller parameter budget and a ten-fold shorter CPU inference time (Section 6.5).
- 3.
- 7 × 7 feature map preserves CWT structure. Each stride-2 operation in DS-1 through DS-5 halves the spatial resolution of the input, yielding a final feature map of . Each cell of this map covers a region of the scalogram. Since the -sample, s record is resized to 224 columns, one cell spans roughly 14 ms of record time, slightly under one fundamental period at 60 Hz, together with a proportional band in the scale axis. This resolution localizes the time–frequency energy concentrations that distinguish Healthy from Mild without over-compressing spatial detail: followed by Global Average Pooling, the map accumulates a global energy summary across 49 time–frequency zones before the shared fully connected representation layer. Grad-CAM analysis (Section 6.10) confirms that this resolution enables activation to span all 49 zones, in contrast to standard convolutions that concentrate energy in narrow horizontal frequency bands. This is a design rationale rather than an optimum: the parameter study of Section 6.8 subsequently measured a 112-pixel input, which yields a final map under the same network, and found it to be the better operating point on every metric. The configuration is retained throughout for continuity with the prior studies on this benchmark, and the discrepancy is reported rather than resolved in favor of the inherited choice.
4.2. Architecture
4.3. Parameter Comparison
5. Training Strategy
5.1. Multi-Task Loss Function
5.2. Training Configuration
5.2.1. Proposed Method: CWT-MTNet (Scratch Training)
5.2.2. Comparison: Baseline CNN (Scratch Training)
5.2.3. Comparison: ImageNet-Pretrained Networks (Two-Stage Fine-Tuning)
5.3. Reproducibility
6. Experimental Results
6.1. Pretrained vs. Scratch Training
- 1.
- Domain gap. ImageNet convolutional filters are optimized for natural image textures (edges, colors, object parts) that are structurally different from CWT time–frequency patterns. The pretrained weights bias the shared backbone toward irrelevant feature directions that are difficult to overcome with fine-tuning on only 10,937 training images.
- 2.
- Multi-task gradient conflict. Six prediction heads generate competing gradient signals that update the shared backbone simultaneously. For large pretrained networks (11.8M–138.5M parameters), this gradient interference is amplified by the high dimensionality of the parameter space, causing the backbone to oscillate rather than converge toward a stable shared representation. The ablation study in Section 6.6 quantifies this effect directly.
- 3.
- Parameter excess. VGG-16 (138.5M) and ResNet-50 (25.7M) contain far more parameters than can be reliably optimized on 10,937 samples. The over-parameterized network searches an excessively large weight space, and fine-tuning from pretrained initialization cannot adequately redirect it toward the CWT-specific manifold.
6.2. Scratch Training Results
6.3. Confusion Matrix Analysis
6.4. Multi-Seed Statistical Validation
6.5. Deployment Cost
6.6. Ablation Study: Gradient Conflict Quantification
- 1.
- Baseline CNN: mild SH advantage. The SH model outperforms the MT model by +0.82 pp (BH) and +6.92 pp (CH). The compact standard-convolution backbone (425K parameters) is more resistant to gradient conflict than ResNet-18 because its limited parameter space reduces the degrees of freedom for conflicting gradients to interfere.
- 2.
- ResNet-18: severe gradient conflict. When trained as a dedicated single-switch network (SH), ResNet-18 achieves 96.73% Mild recall on BH, demonstrating sufficient capacity for the per-switch diagnosis task. In the multi-task setting (MT), however, this drops to 54.60%: SH outperforms MT by +42.13 pp, confirming that gradient conflict—not insufficient network capacity—is the limiting factor. The Mild-to-Healthy rate on BH rises from 1.43% (SH) to 39.67% (MT)—a 28-fold increase—further underscoring the safety risk of gradient-conflict-induced collapse in this backbone.
- 3.
- MobileNet-v2: robust to gradient conflict. The SH advantage is only +3.27 pp (BH) and +0.42 pp (CH). The factored DS-Conv structure induces implicit gradient regularization: because depthwise and pointwise filters occupy different parameter subspaces, task-specific gradients are partially decoupled in the parameter update, reducing direct interference in the shared representation.
- 4.
- CWT-MTNet: no systematic multi-task penalty. CWT-MTNet is the only network whose changes sign between the two switches: pp on BH, where the six-head model is the better of the two, and pp on CH. Neither magnitude approaches the ResNet-18 collapse, and the inconsistent direction indicates that no systematic gradient conflict is present at this scale. A plausible reading is that with 172 K parameters the six correlated switch objectives act as a regularizer on the shared representation rather than as competing demands on it, although the present two-switch experiment cannot separate this explanation from ordinary run-to-run variation.
6.7. What the Representation Contributes
- 1.
- Phase preserved, no decomposition. The deviations enter a 1-D network directly as five real channels (, , , , ).
- 2.
- Phase discarded, no decomposition. The same signals are reduced to magnitude— as its analytic envelope, together with and —matching the magnitude-only construction of the image channels.
- 3.
- Phase discarded, CWT decomposition. The proposed pipeline: normalized by (4), quantized to 8 bits and resized to .
6.8. Sensitivity to the CWT Parameters
6.9. Measurement-Noise Robustness
6.10. Grad-CAM Analysis
7. Discussion
7.1. Why DS-Conv Outperforms Standard Convolution in CWT Domain
7.2. CWT-MTNet vs. MobileNet-v2: Efficiency and Stability
7.3. Why Scratch Training Outperforms ImageNet Pretraining on CWT Images
7.4. Mild Misclassification and Safety Implications
7.5. Where the Binding Constraints Lie
7.6. Limitations and Future Work
8. Conclusions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Yang, S.; Xiang, D.; Bryant, A.; Mawby, P.; Ran, L.; Tavner, P. Condition monitoring for device reliability in power electronic converters: A review. IEEE Trans. Power Electron. 2010, 25, 2734–2752. [Google Scholar] [CrossRef] [Scilit]
- Yang, S.; Bryant, A.; Mawby, P.; Xiang, D.; Ran, L.; Tavner, P. An industry-based survey of reliability in power electronic converters. IEEE Trans. Ind. Appl. 2011, 47, 1441–1451. [Google Scholar] [CrossRef] [Scilit]
- Choi, U.M.; Lee, K.B.; Blaabjerg, F. Diagnosis and tolerant strategy of an open-switch fault for T-type three-level inverter systems. IEEE Trans. Ind. Appl. 2014, 50, 495–508. [Google Scholar] [CrossRef] [Scilit]
- Dimech, E.; Dawson, J.F. Electrical parameters characterization of aged IGBTs by thermo-electrical overstress. In IECON 2018—44th Annual Conference of the IEEE Industrial Electronics Society; IEEE: New York, NY, USA, 2018; pp. 5924–5929. [Google Scholar] [CrossRef] [Scilit]
- Park, H.M.; Lee, J.H.; Jun, H.S.; Hwang, K.B.; Park, S.J.; Park, J.H. Reliability diagnosis and fault prediction technique for three-phase inverters using artificial neural networks. IEEE Access 2026, 14, 4576–4590. [Google Scholar] [CrossRef] [Scilit]
- Park, H.M.; Park, J.H. CWT-based RGB image representation for IGBT aging diagnosis and combined aging index regression in three-phase inverters using convolutional neural networks. IEEE Access, 2026; manuscript under review.
- Yu, T.; Kumar, S.; Gupta, A.; Levine, S.; Hausman, K.; Finn, C. Gradient surgery for multi-task learning. Proc. Proc. Adv. Neural Inf. Process. Syst. 2020, 33, 5824–5836. [Google Scholar]
- Oh, H.; Han, B.; McCluskey, P.; Han, C.; Youn, B.D. Physics-of-failure, condition monitoring, and prognostics of insulated gate bipolar transistor modules: A review. IEEE Trans. Power Electron. 2015, 30, 2413–2426. [Google Scholar] [CrossRef] [Scilit]
- Abuelnaga, A.; Narimani, M.; Bahman, A.S. A review on IGBT module failure modes and lifetime testing. IEEE Access 2021, 9, 9643–9663. [Google Scholar] [CrossRef] [Scilit]
- Choi, U.M.; Blaabjerg, F.; Lee, K.B. Study and handling methods of power IGBT module failures in power electronic converter systems. IEEE Trans. Power Electron. 2015, 30, 2517–2533. [Google Scholar] [CrossRef] [Scilit]
- Choi, U.M.; Blaabjerg, F.; Jørgensen, S.; Munk-Nielsen, S.; Rannestad, B. Reliability improvement of power converters by means of condition monitoring of IGBT modules. IEEE Trans. Power Electron. 2017, 32, 7990–7997. [Google Scholar] [CrossRef] [Scilit]
- Sun, P.; Gong, C.; Du, X.; Peng, Y.; Wang, B.; Zhou, L. Condition monitoring IGBT module bond wires fatigue using short-circuit current identification. IEEE Trans. Power Electron. 2017, 32, 3777–3786. [Google Scholar] [CrossRef] [Scilit]
- Wang, C.; He, Y.; Wang, C.; Li, L.; Wu, X. Multi-chip IGBT module failure monitoring based on module transconductance with temperature calibration. Electronics 2020, 9, 1559. [Google Scholar] [CrossRef] [Scilit]
- Liu, W.; Zhou, D.; Iannuzzo, F.; Hartmann, M.; Blaabjerg, F. Separation and validation of bond-wire and solder layer failure modes in IGBT modules. IEEE Trans. Ind. Appl. 2022, 58, 2324–2331. [Google Scholar] [CrossRef] [Scilit]
- Dai, Z.; Ge, X.; Lin, C.; Wang, H.; Xu, Z.; Liang, G. A bond wire aging monitoring method for IGBT modules based on bond wire degradation voltage. IEEE J. Emerg. Sel. Top. Power Electron. 2024, 12, 5534–5543. [Google Scholar] [CrossRef] [Scilit]
- Liu, H.; Wang, F.; Zhang, X.; Xia, W.; Ren, L. An online monitoring method for bond wire fatigue in IGBT module. IEEE J. Electron Devices Soc. 2024, 12, 440–449. [Google Scholar] [CrossRef] [Scilit]
- Liu, H.; Zhang, S.; Tang, H. Hybrid method for remaining useful life prediction of power IGBT modules in high-speed trains. IEEE Trans. Power Electron. 2024, 39, 15101–15117. [Google Scholar] [CrossRef] [Scilit]
- Park, H.M.; Park, J.H. Improved IGBT Aging Diagnosis for Three-Phase Inverters via Phase-Angle Feature Redesign and Kernel SHAP Analysis. IEEE Access 2026, 14, 97179–97192. [Google Scholar] [CrossRef] [Scilit]
- Sun, Q.; Yu, X.; Li, H.; Peng, F.; Sun, G. Fault detection for power electronic converters based on continuous wavelet transform and convolution neural network. J. Intell. Fuzzy Syst. 2022, 42, 3537–3549. [Google Scholar] [CrossRef] [Scilit]
- Kou, L.; Liu, C.; Cai, G.; Zhang, Z. Fault diagnosis for power electronics converters based on deep feedforward network and wavelet compression. Electr. Power Syst. Res. 2020, 185, 106370. [Google Scholar] [CrossRef] [Scilit]
- Fu, Y.; Ji, Y.; Meng, G.; Chen, W.; Bai, X. Three-phase inverter fault diagnosis based on an improved deep residual network. Electronics 2023, 12, 3460. [Google Scholar] [CrossRef] [Scilit]
- El-Naeem, K.S.A.; Nayel, M.A.; Abdelrahem, M.; Alkabbany, I. Detecting open-circuit faults in power electronic converters using continuous wavelet transform and convolutional neural networks for simultaneous charging systems. Arab. J. Sci. Eng. 2025, 51, 10823–10845. [Google Scholar] [CrossRef] [Scilit]
- Lu, F.; Guo, Q.; Dou, Z.; Chen, Y.; Wang, Q.; An, X.; Dou, H. A novel simultaneous diagnosis method for IGBT open-circuit faults and current sensor faults of three-phase SPWM inverter. IEEE Trans. Power Electron. 2025, 40, 11369–11379. [Google Scholar] [CrossRef] [Scilit]
- Arif, M.N.; Ud Din, Z.; Ul Haq, A.; Cheema, K.M.; Milyani, A.H.; Naeem-ul-Islam; Ashfaq, I. Open switch fault diagnosis of cascaded H-bridge 5-level inverter using deep learning. Front. Energy Res. 2024, 12, 1388273. [Google Scholar] [CrossRef] [Scilit]
- Yao, C.; Xu, S.; Ren, G.; Wu, S.; Li, G.; Sun, Z.; Ma, G. Online open-circuit fault diagnosis for ANPC inverters using edge-based lightweight two-dimensional CNN. IEEE Trans. Power Electron. 2024, 39, 3979–3984. [Google Scholar] [CrossRef] [Scilit]
- Ma, G.; Yao, C.; Xu, S.; Ren, G.; Sun, Z.; Wu, S. Real-time diagnosis of multiple open-circuit faults in ANPC inverters based on lightweight deployment of edge 2D-CNN. IEEE Trans. Ind. Electron. 2025, 72, 11885–11896. [Google Scholar] [CrossRef] [Scilit]
- Liu, Q.; Chen, C.; Ouyang, H.; Xiao, M.; Lei, W. IHBA-optimized DR-SE-NPCNet for robust open-circuit fault diagnosis in three-level NPC inverters under mixed and noisy conditions. Sci. Rep. 2025, 16, 3826. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Sivapriya, A.; Kalaiarasi, N.; Verma, R.; Chokkalingam, B.; Munda, J.L. Fault diagnosis of cascaded multilevel inverter using multiscale kernel convolutional neural network. IEEE Access 2023, 11, 79513–79530. [Google Scholar] [CrossRef] [Scilit]
- Yan, Y.; Wu, J.; Cao, Y.; Liu, B.; Li, C.; Shi, T. An open-circuit fault diagnosis method for three-level neutral point clamped inverters based on multi-scale shuffled convolutional neural network. Sensors 2024, 24, 1745. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Chai, Q.; Li, H.; Wang, W.; Yan, Q. Transfer learning based open-circuit fault diagnosis method for three-phase inverters. J. Power Electron. 2025, 25, 1030–1040. [Google Scholar] [CrossRef] [Scilit]
- Xia, Y.; Xu, Y. A transferrable data-driven method for IGBT open-circuit fault diagnosis in three-phase inverters. IEEE Trans. Power Electron. 2021, 36, 13478–13488. [Google Scholar] [CrossRef] [Scilit]
- Lei, X.; Wu, F.; Liu, Y. An online convolutional neural network based method for open-circuit fault diagnosis in three-phase inverters under extremely unbalanced loading condition. IEEE Trans. Power Electron. 2026, 41, 16084–16098. [Google Scholar] [CrossRef] [Scilit]
- Luo, W.; Xie, Z.; Li, Y.; Chen, M.; He, R.; Peng, Y.; Zhang, X. Enhanced 1-D convolutional neural network-based open-circuit fault diagnosis and hybrid fault-tolerant control for three-level NPC converters. IEEE Trans. Instrum. Meas. 2025, 74, 1–14. [Google Scholar] [CrossRef] [Scilit]
- Łuczak, D. Machine fault diagnosis through vibration analysis: Continuous wavelet transform with complex Morlet wavelet and time–frequency RGB image recognition via convolutional neural network. Electronics 2024, 13, 452. [Google Scholar] [CrossRef] [Scilit]
- He, K.; Zhang, X.; Ren, S.; Sun, J. Deep residual learning for image recognition. In 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2016; pp. 770–778. [Google Scholar] [CrossRef] [Scilit]
- Simonyan, K.; Zisserman, A. Very deep convolutional networks for large-scale image recognition. arXiv 2015, arXiv:1409.1556. [Google Scholar] [CrossRef] [Scilit]
- Sandler, M.; Howard, A.; Zhu, M.; Zhmoginov, A.; Chen, L.C. MobileNetV2: Inverted residuals and linear bottlenecks. In 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition; IEEE: New York, NY, USA, 2018; pp. 4510–4520. [Google Scholar] [CrossRef] [Scilit]
- Szegedy, C.; Liu, W.; Jia, Y.; Sermanet, P.; Reed, S.; Anguelov, D.; Erhan, D.; Vanhoucke, V.; Rabinovich, A. Going deeper with convolutions. In 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2015; pp. 1–9. [Google Scholar] [CrossRef] [Scilit]
- Zhao, Z.; Li, T.; Wu, J.; Sun, C.; Wang, S.; Yan, R.; Chen, X. Deep learning algorithms for rotating machinery intelligent diagnosis: An open source benchmark study. ISA Trans. 2020, 107, 224–255. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Wang, B.; Zhao, S. MTAGCN: Multi-task graph-guided convolutional network with attention mechanism for intelligent fault diagnosis of rotating machinery. Machines 2025, 13, 347. [Google Scholar] [CrossRef] [Scilit]
- Bhardwaj, D.; Londhe, N.D.; Raj, R. Fault-MTL: A multi-task deep learning approach for simultaneous fault classification and localization in power systems. J. Control Autom. Electr. Syst. 2024, 35, 884–898. [Google Scholar] [CrossRef] [Scilit]
- Chen, S.; Zheng, X.; Wu, H. A multi-rate sensor fusion and multi-task learning network for concurrent fault diagnosis of hydraulic systems. Digit. Signal Process. 2025, 156, 104796. [Google Scholar] [CrossRef] [Scilit]
- Zhang, Z.; Chen, X. A knowledge-driven method for IGBT remaining useful life prediction using bidirectional learning and physics-enhanced pathformer networks. J. Comput. Des. Eng. 2025, 12, 327–344. [Google Scholar] [CrossRef] [Scilit]
- Brito, L.C.; Susto, G.A.; Brito, J.N.; Duarte, M.A.V. An explainable artificial intelligence approach for unsupervised fault detection and diagnosis in rotating machinery. Mech. Syst. Signal Process. 2022, 163, 108105. [Google Scholar] [CrossRef] [Scilit]
- Kendall, A.; Gal, Y.; Cipolla, R. Multi-task learning using uncertainty to weigh losses for scene geometry and semantics. arXiv 2018, arXiv:1705.07115. [Google Scholar] [CrossRef] [Scilit]
- Iandola, F.N.; Han, S.; Moskewicz, M.W.; Ashraf, K.; Dally, W.J.; Keutzer, K. SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size. arXiv 2016, arXiv:1602.07360. [Google Scholar] [CrossRef] [Scilit]
- Sahu, R.; Panigrahi, P.K.; Lal, D.K.; Pradhan, R.; Mahanty, C. Robust deep learning for multiclass power system fault diagnosis using edge deployment. Algorithms 2026, 19, 299. [Google Scholar] [CrossRef] [Scilit]
- Liang, Y.P.; Chen, H.; Chung, C.C. A one-dimensional depthwise separable convolutional neural network for bearing fault diagnosis implemented on FPGA. Sensors 2024, 24, 7831. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Li, J.; Wang, N.; Wang, D.; Gao, G.; Pan, W. Lightweight neural networks with anti-colored noise for bearing fault diagnosis using deep separable convolution and transfer learning. Sci. Rep. 2025, 15, 44691. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Selvaraju, R.R.; Cogswell, M.; Das, A.; Vedantam, R.; Parikh, D.; Batra, D. Grad-CAM: Visual explanations from deep networks via gradient-based localization. In 2017 IEEE International Conference on Computer Vision (ICCV); IEEE: New York, NY, USA, 2017; pp. 618–626. [Google Scholar] [CrossRef] [Scilit]
- Mallat, S. A Wavelet Tour of Signal Processing, 2nd ed.; Academic Press: San Diego, CA, USA, 1999. [Google Scholar]
- Torrence, C.; Compo, G.P. A practical guide to wavelet analysis. Bull. Amer. Meteor. Soc. 1998, 79, 61–78. [Google Scholar] [CrossRef] [Scilit]
- Gealy, C.B.; George, A.D. Characterizing parameter scaling with quantization for deployment of CNNs on real-time systems. ACM Trans. Embed. Comput. Syst. 2024, 23, 1–35. [Google Scholar] [CrossRef] [Scilit]
- Russakovsky, O.; Deng, J.; Su, H.; Krause, J.; Satheesh, S.; Ma, S.; Huang, Z.; Karpathy, A.; Khosla, A.; Bernstein, M.; et al. ImageNet large scale visual recognition challenge. Int. J. Comput. Vis. 2015, 115, 211–252. [Google Scholar] [CrossRef] [Scilit]
- Tzeng, E.; Hoffman, J.; Saenko, K.; Darrell, T. Adversarial Discriminative Domain Adaptation. In 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR); IEEE: New York, NY, USA, 2017; pp. 7167–7176. [Google Scholar] [CrossRef] [Scilit]
- Ben-David, S.; Blitzer, J.; Crammer, K.; Kulesza, A.; Pereira, F.; Vaughan, J.W. A theory of learning from different domains. Mach. Learn. 2010, 79, 151–175. [Google Scholar] [CrossRef] [Scilit]
- Kumar, A.; Zhou, Y.; Mucchi, E.; Wang, D. Explainable artificial intelligence based simulation-to-real domain adaptation for robust rotor condition monitoring. Adv. Eng. Inform. 2026, 74, 104652. [Google Scholar] [CrossRef] [Scilit]




| Study | Year | Input and Representation | Degradation Type | Diagnostic Output |
|---|---|---|---|---|
| Xia and Xu [31] | 2021 | Phase currents, transfer-learned features | OC fault | Faulty switch identity |
| Sivapriya et al. [28] | 2023 | Multiscale kernel CNN | OC fault | Faulty switch identity |
| Arif et al. [24] | 2024 | Measured signals, deep network | Open-switch | Faulty switches, up to two at once |
| Yao et al. [25] | 2024 | 2-D image, edge-deployed CNN | OC fault | Faulty switch identity |
| Ma et al. [26] | 2025 | 2-D image, edge-deployed CNN | Multiple OC faults | Faulty switch identities |
| Lu et al. [23] | 2025 | Model-based residuals | OC fault and sensor fault | Faulty switch and faulty sensor |
| Abd El-Naeem et al. [22] | 2025 | CWT scalogram, CNN | OC fault | Faulty switch identity |
| Zhang and Chen [43] | 2025 | Physics-coupled sequence model | Progressive aging | Remaining useful life, one device |
| Park et al. [5] | 2026 | Phase-voltage rates of change and symmetrical-component magnitudes; six scalars, ANN | Graded aging | System risk level and aging index |
| Park and Park [18] | 2026 | Symmetrical-component magnitudes and phase angles; six scalars, ANN with SHAP | Graded aging | System risk level and aging index |
| Park and Park [6] | — | Symmetrical-component deviations as a CWT-RGB image, CNN | Graded aging | System risk level and aging index |
| This Work | — | Symmetrical-component deviations as a CWT-RGB image, DS-Conv multi-task network | Graded aging | Four-class aging state for each of six switches, one forward pass |
| State | () | Class Label |
|---|---|---|
| Master | 0.007 | — (reference only) |
| Healthy | 0.00665 | 1 |
| Mild | 0.00623 | 2 |
| Moderate | 0.00574 | 3 |
| Severe | 0.005 | 4 |
| Class | Count (Approx.) | Ratio (%) |
|---|---|---|
| Healthy (incl. Master) | 3125 | 20.0 |
| Mild | 3125 | 20.0 |
| Moderate | 3125 | 20.0 |
| Severe | 6250 | 40.0 |
| Total | 15,625 | 100.0 |
| Block | Operation | Output Size | Params |
|---|---|---|---|
| Stem | Conv(, 32, s2) + BN + ReLU | 896 | |
| DS-1 | DWConv(, s2) + BN + ReLU6 → PWConv(, 64) + BN + ReLU6 | 2.6 K | |
| DS-2 | DWConv(, s2) + BN + ReLU6 → PWConv(, 128) + BN + ReLU6 | 9.3 K | |
| DS-3 | DWConv() + BN + ReLU6 → PWConv(, 128) + BN + ReLU6 | 17.7 K | |
| DS-4 | DWConv(, s2) + BN + ReLU6 → PWConv(, 256) + BN + ReLU6 | 34.0 K | |
| DS-5 | DWConv(, s2) + BN + ReLU6 → PWConv(, 256) + BN + ReLU6 | 68.0 K | |
| Shared | GAP + Dropout(0.4) + FC(128) + ReLU | 128 | 32.9 K |
| Heads | 6 × FC(4) + Softmax | 6×4 | 3.1 K |
| Total parameters | 171,672 | ||
| Network | Pretrained (Two-Stage) | Scratch (100 ep) |
|---|---|---|
| SqueezeNet | 40.14 | 46.48 |
| MobileNet-v2 | 49.72 | 98.88 |
| GoogLeNet | 40.49 | 40.49 |
| ResNet-18 | 73.72 | 88.50 |
| ResNet-50 | 57.46 | 74.85 |
| VGG-16 | 40.14 | 40.49 |
| Network | Params | Mean Acc (%) | Mean F1 (%) | Mean Mild Recall (%) | Mean Mild → Healthy (%) |
|---|---|---|---|---|---|
| CWT-MTNet (Proposed) | 172 K | 98.39 | 98.31 | 96.73 | 2.12 |
| Baseline CNN | 425 K | 96.34 | 95.98 | 90.00 | 6.05 |
| SqueezeNet | 1.4 M | 46.48 | 24.90 | 0.00 | — |
| ShuffleNet | 1.5 M | 96.02 | 95.82 | 82.18 | 17.02 |
| MobileNet-v2 | 3.7 M | 98.88 | 98.87 | 94.88 | 4.87 |
| EfficientNet-B0 | 5.4 M | 96.28 | 96.13 | 83.12 | 15.78 |
| NASNet-Mobile | 5.5 M | 94.12 | 93.75 | 74.34 | 24.21 |
| GoogLeNet | 7.1 M | 40.49 | 14.41 | 0.00 | — |
| ResNet-18 | 11.8 M | 88.50 | 87.15 | 59.90 | 34.87 |
| ResNet-50 | 25.7 M | 74.85 | 68.13 | 26.45 | 63.93 |
| VGG-16 | 138.5 M | 40.49 | 14.41 | 0.00 | — |
| Accuracy (%) | Mild Recall (%) | |||||||
|---|---|---|---|---|---|---|---|---|
| Switch | Baseline | MobileNet | ResNet | CWT- | Baseline | MobileNet | ResNet | CWT- |
| CNN | -v2 | -18 | MTNet | CNN | -v2 | -18 | MTNet | |
| AH (IGBT1) | 96.55 | 98.64 | 87.93 | 98.08 | 88.18 | 92.95 | 48.64 | 97.05 |
| AL (IGBT2) | 96.67 | 98.64 | 88.91 | 98.38 | 91.07 | 93.97 | 63.84 | 96.65 |
| BH (IGBT3) | 95.99 | 98.98 | 85.29 | 98.55 | 89.57 | 96.32 | 54.60 | 97.96 |
| BL (IGBT4) | 96.93 | 98.51 | 86.78 | 98.34 | 93.20 | 92.76 | 55.70 | 96.71 |
| CH (IGBT5) | 95.69 | 99.79 | 90.96 | 98.21 | 86.37 | 99.37 | 67.92 | 94.97 |
| CL (IGBT6) | 96.20 | 98.72 | 91.13 | 98.76 | 91.60 | 93.91 | 68.70 | 97.06 |
| Mean | 96.34 | 98.88 | 88.50 | 98.39 | 90.00 | 94.88 | 59.90 | 96.73 |
| Switch | Baseline CNN | ResNet-18 | ResNet-50 | CWT-MTNet |
|---|---|---|---|---|
| AH (IGBT1) | 7.3 | 44.1 | 61.6 | 3.0 |
| AL (IGBT2) | 6.5 | 29.7 | 67.4 | 1.1 |
| BH (IGBT3) | 7.2 | 39.7 | 88.1 | 1.2 |
| BL (IGBT4) | 2.4 | 36.0 | 82.5 | 1.8 |
| CH (IGBT5) | 6.1 | 29.6 | 32.7 | 4.0 |
| CL (IGBT6) | 6.9 | 30.3 | 51.3 | 1.7 |
| Mean | 6.05 | 34.87 | 63.93 | 2.12 |
| Metric | CWT-MTNet | MobileNet-v2 | Baseline CNN | ResNet-18 |
|---|---|---|---|---|
| (0.17 M) | (3.7 M) | (0.43 M) | (11.8 M) | |
| Accuracy (%) | ||||
| Macro-F1 (%) | ||||
| Mild recall (%) | ||||
| Mild→Healthy (%) |
| Network | Params | FLOPs | Size | CPU |
|---|---|---|---|---|
| (M) | (G) | (MB) | (ms/img) | |
| CWT-MTNet (Ours) | 0.172 | 0.098 | 0.7 | 3.1 ± 0.4 |
| Baseline CNN | 0.425 | 0.103 | 1.7 | 3.9 ± 0.6 |
| SqueezeNet | 1.367 | 0.380 | 5.5 | 5.0 ± 0.6 |
| MobileNet-v2 | 3.653 | 0.300 | 14.6 | 32.4 ± 1.5 |
| GoogLeNet | 7.130 | 1.500 | 28.5 | 139.4 ± 4.7 |
| ResNet-18 | 11.826 | 1.814 | 47.3 | 100.4 ± 2.6 |
| ResNet-50 | 25.715 | 4.089 | 102.9 | 142.9 ± 3.5 |
| VGG-16 | 138.49 | 15.470 | 554.0 | 40.7 ± 0.7 |
| FLOPs: literature values; CWT-MTNet/CNN: analytical. Latency: median ± std over 45 runs after 5 warm-up. | ||||
| Network | BH (IGBT3) | CH (IGBT5) | ||||
|---|---|---|---|---|---|---|
| SH (%) | MT (%) | (pp) | SH (%) | MT (%) | (pp) | |
| Baseline CNN | 90.39 | 89.57 | 93.29 | 86.37 | ||
| ResNet-18 | 96.73 | 54.60 | +42.13 | 93.50 | 67.92 | +25.58 |
| MobileNet-v2 | 99.59 | 96.32 | 99.79 | 99.37 | ||
| CWT-MTNet | 95.71 | 97.96 | 97.48 | 94.97 | ||
| Encoding | Params | Acc. | Macro-F1 | Mild | Mild → H | SD | Range |
|---|---|---|---|---|---|---|---|
| (%) | (%) | (%) | (%) | (pp) | (pp) | ||
| Phase preserved, no CWT | 168,280 | 100.00 | 100.00 | 100.00 | 0.00 | 0.00 | 0.00 |
| Phase discarded, no CWT | 167,832 | 96.03 | 95.66 | 95.78 | 1.53 | 3.46 | 8.40 |
| Phase discarded, CWT | 171,672 | 98.39 | 98.31 | 96.73 | 2.12 | 0.24 | 0.68 |
| Wavelet | Resolution | Accuracy | Macro-F1 | Mild | Mild → H |
|---|---|---|---|---|---|
| (%) | (%) | (%) | (%) | ||
| Morlet (deployed) | 224 | 98.39 | 98.31 | 96.73 | 2.12 |
| Morlet (control) | 224 | 98.49 | 98.43 | 95.53 | 3.18 |
| Morlet | 99.59 | 99.57 | 98.72 | 0.91 | |
| Morse | 224 | 97.25 | 97.06 | 93.79 | 3.61 |
| Bump | 224 | 93.85 | 93.36 | 87.37 | 7.98 |
| p (%) | Accuracy (%) | Macro-F1 (%) | Mild Recall (%) | Mild → Healthy (%) |
|---|---|---|---|---|
| 0 | 98.39 | 98.31 | 96.73 | 2.12 |
| 0.5 | 41.76 | 29.13 | 14.10 | 56.98 |
| 1 | 43.20 | 33.42 | 13.23 | 55.46 |
| 2 | 34.71 | 26.23 | 6.51 | 45.42 |
| 3 | 32.01 | 23.76 | 5.33 | 43.23 |
| 5 | 29.56 | 21.12 | 3.87 | 41.09 |
| 10 | 28.33 | 19.40 | 5.09 | 36.69 |
| Front End | Suppression | ||||
|---|---|---|---|---|---|
| Acc. (%) | Mild (%) | Acc. (%) | Mild (%) | ||
| None | 1.0× | 98.39 | 96.73 | 41.76 | 14.10 |
| Low-pass 20 kHz | 5.0× | 98.29 | 96.37 | 48.48 | 17.10 |
| Low-pass 5 kHz | 10.0× | 32.57 | 33.20 | 29.95 | 16.63 |
| Added Noise | Effective SNR | Phase Preserved | Phase Discarded | Phase Discarded |
|---|---|---|---|---|
| p (%) | (dB) | No CWT | No CWT | CWT (Proposed) |
| 0.0 | 42.9 | 100.00 | 96.03 | 98.39 |
| 0.5 | 41.1 | 52.43 | 30.14 | 41.76 |
| 1.0 | 38.2 | 41.98 | 26.33 | 43.20 |
| 2.0 | 33.5 | 33.03 | 26.60 | 34.71 |
| 5.0 | 25.9 | 30.85 | 27.96 | 29.56 |
| 10.0 | 20.0 | 30.19 | 27.48 | 28.33 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the author. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Park, J.-H. Lightweight Multi-Task CNN for Simultaneous IGBT Switch Aging Diagnosis Using CWT-Based RGB Images. Electronics 2026, 15, 4298. https://doi.org/10.3390/electronics15184298
Park J-H. Lightweight Multi-Task CNN for Simultaneous IGBT Switch Aging Diagnosis Using CWT-Based RGB Images. Electronics. 2026; 15(18):4298. https://doi.org/10.3390/electronics15184298
Chicago/Turabian StylePark, Jin-Hyun. 2026. "Lightweight Multi-Task CNN for Simultaneous IGBT Switch Aging Diagnosis Using CWT-Based RGB Images" Electronics 15, no. 18: 4298. https://doi.org/10.3390/electronics15184298
APA StylePark, J.-H. (2026). Lightweight Multi-Task CNN for Simultaneous IGBT Switch Aging Diagnosis Using CWT-Based RGB Images. Electronics, 15(18), 4298. https://doi.org/10.3390/electronics15184298

