FedTrack: A Collaborative Target Tracking Framework Based on Adaptive Federated Learning

• Published in: IEEE transactions on vehicular technology Vol. 73; no. 9; pp. 13868 - 13882
• Main Authors: Pan, Yongqi, Zhu, Cheng, Luo, Lailong, Liu, Yi, Cheng, Ziwen
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.09.2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Adaptive algorithms; Adaptive node selection; Algorithms; Collaboration; collaborative target tracking; Computational modeling; Data collection; Data models; Data transmission; Federated learning; Machine learning; reputation value; Servers; Target recognition; Target tracking; Tracking; Tracking devices; Training
• ISSN: 0018-9545; 1939-9359
• DOI: 10.1109/TVT.2024.3395292

Tracking mobile targets is a basic task for many applications, e.g., wildlife protection, area surveillance, battlefield reconnaissance, disaster rescue, etc. In these cases, multiple edge devices are usually deployed to collect images or videos from the target area. Two mainstream tracking methodologies have been proposed to recognize and track targets from such data sources cooperatively. The centralized strategies collect data from the edge devices and thereafter run mining and learning algorithms upon such data. On the contrary, the distributed strategies implement such algorithms in a distributed manner. However, these methods incur either high transmission costs or slow convergence speeds. To this end, this paper presents a novel cooperative tracking framework (i.e., FedTrack) based on adaptive federated learning. A dual reputation mechanism has been formulated, and subsequently, an adaptive node selection algorithm has been suggested to ascertain the nodes suitable for involvement in the training process. Furthermore, a strategy for selecting the aggregation node based on capability has been developed to enhance the efficiency of aggregation. As far as we know, FedTrack is the first federated learning framework for collaborative tracking. Experimental results demonstrate that FedTrack achieves comparable or even better accuracy than state-of-the-art methods, yet needs much fewer data transmission costs and much less time consumption.