论文标题

通过有条件生成的对抗网络改善了基于传感器的动物行为分类性能

Improved Sensor-Based Animal Behavior Classification Performance through Conditional Generative Adversarial Network

论文作者

Zhao, Zhuqing, Ha, Dong, Damle, Abhishek, Dos, Barbara Roqueto, White, Robin, Ha, Sook

论文摘要

许多活动分类将数据分为固定的窗口大小,以进行特征提取和分类。但是,动物行为的各种持续时间与预定的窗口大小不符。密集的标记和密集的预测方法通过预测每个点的标签来解决此限制。因此,通过追踪起点和终点,我们可以知道所有发生的活动的时间位置和持续时间。尽管如此,稠密的预测可能会出现不一致的问题。我们修改了U-NET和有条件生成的对抗网络(CGAN),并具有自定义的损失功能,以减少分裂和其他未对准的训练策略。在CGAN中,歧视者和发电机像对抗性竞争一样互相训练。发电机产生密集的预测。在我们的情况下,鉴别器作为高级一致性检查,促使发电机以合理的持续时间预测活动。接受CGAN训练的模型在牛,Pig和UCI Hapt数据集中表现出更好或可比的性能。与以前的密集预测工作相比,pgan训练的U-NET从92.17%提高到94.66%,对猪数据的数据集提高到90.85%,从90.85%提高到93.18%。

Many activity classifications segments data into fixed window size for feature extraction and classification. However, animal behaviors have various durations that do not match the predetermined window size. The dense labeling and dense prediction methods address this limitation by predicting labels for every point. Thus, by tracing the starting and ending points, we could know the time location and duration of all occurring activities. Still, the dense prediction could be noisy with misalignments problems. We modified the U-Net and Conditional Generative Adversarial Network (cGAN) with customized loss functions as a training strategy to reduce fragmentation and other misalignments. In cGAN, the discriminator and generator trained against each other like an adversarial competition. The generator produces dense predictions. The discriminator works as a high-level consistency check, in our case, pushing the generator to predict activities with reasonable duration. The model trained with cGAN shows better or comparable performance in the cow, pig, and UCI HAPT dataset. The cGAN-trained modified U-Net improved from 92.17% to 94.66% for the UCI HAPT dataset and from 90.85% to 93.18% for pig data compared to previous dense prediction work.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源