RIE-SenseNet: Riemannian Manifold Embedding of Multi-Source Industrial Sensor Signals for Robust Pattern Recognition
Xu Wang, Puyu Han, Jiaju Kang, Weichao Pan, Luqi Gong
TL;DR
This work addresses the dual challenges of nonlinear structure and distribution shifts in industrial sensor signals by introducing RIE-SenseNet, a geometry-aware Transformer that operates on a hyperbolic Riemannian manifold with a learnable curvature parameter $c$. It combines Möbius-based hyperbolic mapping, a curvature-aware Transformer backbone, and a manifold-based data augmentation pipeline to preserve data geometry and generate faithful synthetic samples. The approach yields state-of-the-art performance, achieving over $>90 ext{%}$ F1-score and substantial robustness improvements over Euclidean CNNs and standard Transformers. The findings demonstrate the practical value of non-Euclidean feature representations and geometry-consistent augmentation for reliable pattern recognition in industrial sensing, with potential for broader IIoT applications. Key contributions include the geometry-aware embedding, hyperbolic attention mechanism, and a principled augmentation strategy that respects manifold structure.
Abstract
Industrial sensor networks produce complex signals with nonlinear structure and shifting distributions. We propose RIE-SenseNet, a novel geometry-aware Transformer model that embeds sensor data in a Riemannian manifold to tackle these challenges. By leveraging hyperbolic geometry for sequence modeling and introducing a manifold-based augmentation technique, RIE-SenseNet preserves sensor signal structure and generates realistic synthetic samples. Experiments show RIE-SenseNet achieves >90% F1-score, far surpassing CNN and Transformer baselines. These results illustrate the benefit of combining non-Euclidean feature representations with geometry-consistent data augmentation for robust pattern recognition in industrial sensing.
