用近邻统计增强分布对齐,提升无线感知跨环境泛化能力
KNN-MMD: Cross Domain Wireless Sensing via Local Distribution Alignment
- 基于目标域近邻构建帮助集,实现类别内局部分布对齐
- 在3个真实场景数据集上达到89.7%~94.2%准确率,优于现有方法
- 无需目标域标签即可自动确定最佳训练停止点,适合少样本场景
无线感知在家庭、办公及公共空间广泛应用,通过分析信道状态信息(CSI)可实现人员识别、手势识别和跌倒检测。然而,CSI对环境变化极为敏感,微小变动即导致模式失真,使源域训练模型在新环境中性能下降甚至失效。现有领域对齐方法虽能对齐全局特征分布,却忽略类别间关系,造成类别错位。为此,本文提出KNN-MMD:先用目标域K近邻构造帮助集,再以最大均值差异(MMD)实现各类别内的局部分布对齐。同时解决训练中性能震荡问题,并通过将目标域支持集作为验证集,无需标签即可自动确定最佳停止点。在三个真实场景数据集上,该方法在少样本设置下实现89.7%~94.2%的分类准确率,显著优于基线方法。代码与数据已开源。
原文摘要 · Abstract (English)
Wireless sensing has recently found widespread applications in diverse environments, including homes, offices, and public spaces. By analyzing patterns in channel state information (CSI), it is possible to infer human actions for tasks such as person identification, gesture recognition, and fall detection. However, CSI is highly sensitive to environmental changes, where even minor alterations can significantly distort the CSI patterns. This sensitivity often leads to performance degradation or outright failure when applying wireless sensing models trained in one environment to another. To address this challenge, Domain Alignment (DAL) has been widely adopted for cross-domain classification tasks, as it focuses on aligning the global distributions of the source and target domains in feature space. Despite its popularity, DAL often neglects inter-category relationships, which can lead to misalignment between categories across domains, even when global alignment is achieved. To overcome these limitations, we propose K-Nearest Neighbors Maximum Mean Discrepancy (KNN-MMD), a novel few-shot method for cross-domain wireless sensing. Our approach begins by constructing a help set using KNN from the target domain, enabling local alignment between the source and target domains within each category using MMD. Additionally, we address a key instability issue commonly observed in cross-domain methods, where model performance fluctuates sharply between epochs. Further, most existing methods struggle to determine an optimal stopping point during training due to the absence of labeled data from the target domain. Our method resolves this by excluding the support set from the target domain during training and employing it as a validation set to determine the stopping criterion. The dataset and code are publicly available at https://github.com/RS2002/KNN-MMD .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。