车辆协同检测交通标志,不传原始数据也能高效训练模型。
Argos: A Decentralized Federated System for Detection of Traffic Signs in CAVs
- 用联邦学习让车辆本地训练,只共享模型参数不传数据。
- 20轮训练后准确率超80%,8-10个本地轮次最高效。
- 适合关注隐私保护与智能交通系统的研发人员。
联网自动驾驶汽车每天产生海量传感器数据,传统集中式机器学习面临隐私与通信挑战。本文提出一种去中心化的联邦学习框架,专用于车载网络中的交通标志检测,实现无需共享原始数据的协作模型训练。该框架将交通标志类别分派至不同车辆,采用轻量级目标检测器进行本地训练,通过FedProx、FedAdam和FedAVG等算法在Flower仿真环境中聚合模型参数,并评估了不同服务器轮次、本地训练轮次、客户端参与比例及数据分布的影响。实验表明:服务器轮次从2增至20,准确率由不足0.1提升至超过0.8;中等本地轮次(8-10)效率最优,准确率约0.67;更高客户端参与比例可提升泛化性能至0.83;FedProx在处理异质性方面优于其他聚合器;非独立同分布(non-IID)数据导致性能下降;训练时长主要随轮次数增加而增长,而非聚合策略。结论表明,该联邦方法为实际车载部署提供了可扩展、隐私保护的解决方案,未来可结合鲁棒聚合与通信优化进一步推动智能交通系统发展。
原文摘要 · Abstract (English)
Connected and automated vehicles generate vast amounts of sensor data daily, raising significant privacy and communication challenges for centralized machine learning approaches in perception tasks. This study presents a decentralized, federated learning framework tailored for traffic sign detection in vehicular networks to enable collaborative model training without sharing raw data. The framework partitioned traffic sign classes across vehicles for specialized local training using lightweight object detectors, aggregated model parameters via algorithms like FedProx, FedAdam and FedAVG in a simulated environment with the Flower framework, and evaluated multiple configurations including varying server rounds, local epochs, client participation fractions, and data distributions. Experiments demonstrated that increasing server rounds from 2 to 20 boosted accuracy from below 0.1 to over 0.8, moderate local epochs (8-10) provided optimal efficiency with accuracies around 0.67, higher client participation fractions enhanced generalization up to 0.83, FedProx outperformed other aggregators in handling heterogeneity, non-IID data distributions reduced performance compared to IID, and training duration primarily scaled with the number of rounds rather than aggregation strategy. We conclude that this federated approach may offer a scalable, privacy-preserving solution for real-world vehicular deployments, potentially guiding future integrations of robust aggregation and communication optimizations to advance intelligent transportation systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。