Federated Multi-Task Attention for Cross-Individual Human Activity Recognition

Federated Multi-Task Attention for Cross-Individual Human Activity Recognition

Qiang Shen, Haotian Feng, Rui Song, Stefano Teso, Fausto Giunchiglia, Hao Xu

Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence
Main Track. Pages 3423-3429. https://doi.org/10.24963/ijcai.2022/475

Federated Learning (FL) is an emerging privacy-aware machine learning technique that applies successfully to the collaborative learning of global models for Human Activity Recognition (HAR). As of now, the applications of FL for HAR assume that the data associated with diverse individuals follow the same distribution. However, this assumption is impractical in real-world scenarios where the same activity is frequently performed differently by different individuals. To tackle this issue, we propose FedMAT, a Federated Multi-task ATtention framework for HAR, which extracts and fuses shared as well as individual-specific multi-modal sensor data features. Specifically, we treat the HAR problem associated with each individual as a different task and train a federated multi-task model, composed of a shared feature representation network in a central server plus multiple individual-specific networks with attention modules stored in decentralized nodes. In this architecture, the attention module operates as a mask that allows to learn individual-specific features from the global model, whilst simultaneously allowing for features to be shared among different individuals. We conduct extensive experiments based on publicly available HAR datasets, which are collected in both controlled environments and real-world scenarios. Numeric results verify that our proposed FedMAT significantly outperforms baselines not only in generalizing to existing individuals but also in adapting to new individuals.
Keywords:
Machine Learning: Multi-task and Transfer Learning
Humans and AI: Computational Sustainability and Human Well-Being
Knowledge Representation and Reasoning: Reasong about actions
Machine Learning: Meta-Learning
Machine Learning: Multi-modal learning