English

ProFi-Net: Prototype-based Feature Attention with Curriculum Augmentation for WiFi-based Gesture Recognition

Machine Learning 2025-04-30 v1

Abstract

This paper presents ProFi-Net, a novel few-shot learning framework for WiFi-based gesture recognition that overcomes the challenges of limited training data and sparse feature representations. ProFi-Net employs a prototype-based metric learning architecture enhanced with a feature-level attention mechanism, which dynamically refines the Euclidean distance by emphasizing the most discriminative feature dimensions. Additionally, our approach introduces a curriculum-inspired data augmentation strategy exclusively on the query set. By progressively incorporating Gaussian noise of increasing magnitude, the model is exposed to a broader range of challenging variations, thereby improving its generalization and robustness to overfitting. Extensive experiments conducted across diverse real-world environments demonstrate that ProFi-Net significantly outperforms conventional prototype networks and other state-of-the-art few-shot learning methods in terms of classification accuracy and training efficiency.

Keywords

Cite

@article{arxiv.2504.20193,
  title  = {ProFi-Net: Prototype-based Feature Attention with Curriculum Augmentation for WiFi-based Gesture Recognition},
  author = {Zhe Cui and Shuxian Zhang and Kangzhi Lou and Le-Nam Tran},
  journal= {arXiv preprint arXiv:2504.20193},
  year   = {2025}
}

Comments

This paper was accepted at The 9th APWeb-WAIM joint International Conference on Web and Big Data

R2 v1 2026-06-28T23:14:25.063Z