A bi-contrast self-supervised learning framework for enhancing multi-label classification in Industrial Internet of ThingsShow others and affiliations
2025 (English)In: Journal of Industrial Information Integration, ISSN 2467-964X, E-ISSN 2452-414X, Vol. 44, article id 100777Article in journal (Refereed) Published
Abstract [en]
In the Industrial Internet of Things (IIoT), multi-label classification is challenging due to limited labeled data, class imbalance, and the necessity to consider temporal and spatial dependencies. We propose BiConED, a bi-contrast encoder–decoder self-supervised model integrating two contrasting methods: RAC employs an encoder–decoder with augmented data to capture temporal dependencies and boost information entropy, enhancing generalization under label scarcity. QuadC captures spatial dependencies across channels through convolutions on hidden vectors. Evaluated on the real-world industrial benchmark SKAB, BiConED improves feature extraction for underrepresented classes, achieving a 26% increase in F1 score, a 67.72% reduction in False Alarm Rate (FAR), and a 57.25% decrease in Missed Alarm Rate (MAR) compared to models without the proposed contrasts. Even with limited labeled data, BiConED maintains a FAR below 1% and recovers up to 85% of the F1 score without resampling, demonstrating its robustness in imbalanced IIoT environments.
Place, publisher, year, edition, pages
Elsevier BV , 2025. Vol. 44, article id 100777
Keywords [en]
Contrasting learning, Industrial Internet of Things (IIoT), Label scarcity, Multi-label imbalanced classification, Self-supervised learning
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-359285DOI: 10.1016/j.jii.2025.100777ISI: 001408961900001Scopus ID: 2-s2.0-85215581229OAI: oai:DiVA.org:kth-359285DiVA, id: diva2:1932612
Note
QC 20250226
2025-01-292025-01-292025-02-26Bibliographically approved