HyDRA: A Hybrid Dual-Mode Network for Closed- and Open-Set RFFI with Optimized VMD
Organizations: School of Electronics and Communication Engineering, Sun Yat-sen University, Shenzhen, 518107, China
Abstract
Device recognition is vital for security in wireless communication systems, particularly for applications like access control. Radio Frequency Fingerprint Identification (RFFI) offers a non-cryptographic solution by exploiting hardware-induced signal distortions. This paper proposes HyDRA, a Hybrid Dual-mode RF Architecture that integrates an optimized Variational Mode Decomposition (VMD) with a novel architecture based on the fusion of Convolutional Neural Networks (CNNs), Transformers, and Mamba components, designed to support both closed-set and open-set classification tasks. The optimized VMD enhances preprocessing efficiency and classification accuracy by fixing center frequencies and using closed-form solutions. HyDRA employs the Transformer Dynamic Sequence Encoder (TDSE) for global dependency modeling and the Mamba Linear Flow Encoder (MLFE) for linear-complexity processing, adapting to varying conditions. Evaluation on public datasets demonstrates state-of-the-art (SOTA) accuracy in closed-set scenarios and robust performance in our proposed open-set classification method, effectively identifying unauthorized devices. Deployed on NVIDIA Jetson Xavier NX, HyDRA achieves millisecond-level inference speed with low power consumption, providing a practical solution for real-time wireless authentication in real-world environments. The source code is published on https://github.com/Crazy-Bull/HyDRA.
Figures & tables
| Dataset | Preprocessing | HyDRA (TDSE) | HyDRA (MLFE) | ||
|---|---|---|---|---|---|
| ACC (%) | F1 (%) | ACC (%) | F1 (%) | ||
| SingleDay | None | ||||
| VMD (ADMM), k=3 | |||||
| lossless VMD, k=3 | |||||
| ManyTx | None | ||||
| VMD (ADMM), k=3 | |||||
| Dataset | Tx No. | Length | Format | Entries | Entries/Tx |
|---|---|---|---|---|---|
| SingleDay | 28 | 256 | IQ | 224000 | 8000 |
| ManyTx | 150 | 256 | IQ | 511515 | 1556–3600 |
| Dataset | Method | ACC (%) | Model Size | Parameter Count |
|---|---|---|---|---|
| SingleDay | CNN [ 44 ] | 85 | - | 60,272 |
| CNN-TFLite [ 45 ] | 99 | 462.02 KB | 116,808 | |
| Transformer-ResNet [ 46 ] | 93 | - | - | |
| Transformer-TFLite [ 45 ] | 98 | 210.20 KB | 47,964 | |
| HyDRA(TDSE) | 99.96 | 600 KB | 146,876 | |
| HyDRA(MLFE) | 99.93 | 394 KB | 96,060 |
| Mode Number | Central DFT index |
|---|---|
| 2 | |
| 3 | |
| 4 | |
| 5 | |
| 6 | |
| 7 |
| Parameter | Meaning | Value |
|---|---|---|
| optimizer | Adam | |
| initial learning rate | ||
| scheduler | ReduceLROnPlateau | |
| factor of lr reduction | 0.1 | |
| patience | patience to trigger lr reduction | 10 |
| minimal learning rate |
| Parameter | Meaning | Value |
|---|---|---|
| Number of TDSE layers | 2 | |
| Number of MLFE layers | 1 | |
| Temporal convolutional kernel size | 3 | |
| Fixed convolutional kernel size | 15 | |
| Feature dimension | 64 | |
| Number of attention heads (TDSE) | 4 |
| Property | GeForce RTX 3070Ti | Jetson Xavier NX |
|---|---|---|
| GPU | NVIDIA GeForce RTX 3070Ti | NVIDIA Volta |
| CPU | Core i7-8700K | Carmel Arm v8.2 |
| RAM | 64 GB | 8 GB |
| Power usage | 240 W | 10 W / 15 W / 20 W |
| Purpose | Training and testing | Real-world deployment |
| Property | HyDRA (TDSE) | HyDRA (MLFE) |
|---|---|---|
| Accuracy | 99.96% | 99.93% |
| Training Time | 4929.3s | 856.0s |
| Model Size | 600 KB | 394 KB |
| FLOPs | 15,746,816 | 15,746,816 |
| Parameter Count | 146,876 | 96,060 |
| Inf. Time (GeForce RTX 3070Ti) | 0.4219ms | 0.3962ms |
| Metric Type | ACC (%) | Precision (%) | Recall (%) | F1 (%) |
|---|---|---|---|---|
| Macro-Averaged | 94.67 | 96.74 | 96.76 | 96.34 |
| Weighted-Averaged | 94.67 | 96.39 | 94.67 | 95.28 |