论文标题
跨域联合对象检测
Cross-domain Federated Object Detection
论文作者
论文摘要
一方(包括服务器)培训的检测模型在分发给其他用户(客户)时可能会面临严重的性能退化。联合学习可以使多方协作学习无需泄漏客户数据。在本文中,我们专注于特殊的跨域场景,其中服务器具有大规模标记的数据,而多个客户只有少量的标记数据。同时,客户之间的数据分布存在差异。在这种情况下,传统的联邦学习方法无法帮助客户学习所有参与者的全球知识及其独特的知识。为了弥补这一限制,我们提出了一个跨域联合对象检测框架,名为FedOd。拟议的框架首先执行联合培训,以通过多教师蒸馏获得公共全球汇总的模型,并将汇总模型发送回每个客户端以微调其个性化的本地模型。经过几轮通信后,在每个客户端,我们可以对公共全球模型和个性化本地模型进行加权合奏推理。我们建立了一个联合对象检测数据集,该数据集具有基于多个公共自主驾驶数据集的重大背景差异和实例差异,然后在数据集上进行大量实验。实验结果验证了所提出的方法的有效性。
Detection models trained by one party (including server) may face severe performance degradation when distributed to other users (clients). Federated learning can enable multi-party collaborative learning without leaking client data. In this paper, we focus on a special cross-domain scenario in which the server has large-scale labeled data and multiple clients only have a small amount of labeled data; meanwhile, there exist differences in data distributions among the clients. In this case, traditional federated learning methods can't help a client learn both the global knowledge of all participants and its own unique knowledge. To make up for this limitation, we propose a cross-domain federated object detection framework, named FedOD. The proposed framework first performs the federated training to obtain a public global aggregated model through multi-teacher distillation, and sends the aggregated model back to each client for fine-tuning its personalized local model. After a few rounds of communication, on each client we can perform weighted ensemble inference on the public global model and the personalized local model. We establish a federated object detection dataset which has significant background differences and instance differences based on multiple public autonomous driving datasets, and then conduct extensive experiments on the dataset. The experimental results validate the effectiveness of the proposed method.