CarlaFLCAV

Federated Deep Learning Meets
Autonomous Vehicle Perception:
Design and Verification

IEEE Network · 2023

Shuai Wang1*, Chengyang Li2*, Derrick Wing Kwan Ng3, Yonina C. Eldar4, H. Vincent Poor5, Qi Hao6, and Chengzhong Xu7

1Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences 2HKUST (Guangzhou) 3University of New South Wales 4Weizmann Institute of Science 5Princeton University 6Southern University of Science and Technology; Shenzhen Research Institute of Trustworthy Autonomous Systems 7University of Macau

* Equal contribution

Vehicle, road sensor, edge server, and cloud connections in FLCAV
Figure 1. Vehicle–road–edge–cloud networking for collaborative perception.

Overview

Corner cases and visual occlusions make autonomous vehicle perception difficult in open driving scenarios. Federated learning assisted connected autonomous vehicles (FLCAV) learn from data distributed across vehicles and road sensors by exchanging model parameters and perception outputs.

CarlaFLCAV provides a CARLA-based platform for studying this system: synchronized multi-modal data generation, perception models, federated training, and resource allocation. The paper connects network design and sensor placement to the accuracy of downstream driving tasks.

Weather classification, traffic sign recognition, object detection, and trajectory prediction with their sensors and DNNs
Figure 2. Four perception tasks, their sensing modalities, and example neural networks.

Architecture

Three stages of federated training

Cloud pretraining initializes the perception model. Edge federated learning combines vehicle updates with road-sensor knowledge. Cloud federated learning then aggregates the regional models, with personalization for individual regions.

Cloud pretraining, edge federated learning, and cloud federated learning with vehicle and road sensor exchanges
Figure 3. Training across vehicles, edge servers, and the cloud.

Network resources and road sensor placement

Multi-layer graph resource allocation (MLGRA) distributes communication resources across training stages, tasks, and sensing modalities to reduce perception error. Vehicle-road contrastive sensor placement (VRCSP) prioritizes locations where vehicle and road-sensor detections disagree, helping reveal occluded objects.

Traffic-volume-based and vehicle-road contrastive sensor placement compared in CARLA
Figure 4. Road sensor placement based on traffic volume versus vehicle–road perception disagreement.

CARLA Simulation

A platform for system-level experiments

CarlaFLCAV combines dataset generation, perception tasks, federated learning frameworks, and optimization modules. Data collection includes synchronized sensor observations, calibration, and annotations.

The Data Recorder is based on our carla_dataset_tools repository.

Browse code and installation instructions ↗

Evaluation across three towns

The single-task experiment trains SECOND for 3D object detection. Training data come from Town02, Town05, and Town03; evaluation uses 2,432 Town05 point clouds. Wireless and wireline uplinks each have a 4 GB communication budget.

StageMapPoint-cloud samples
Cloud pretrainingTown025,000
Edge FLTown054 CAVs × 500
Cloud FLTown033 CAVs × 500
EvaluationTown052,432

Experimental Results

Single-task 3D object detection

MLGRA with contrastive road sensor placement achieves 84.58% mAP; adding device selection reaches 86.76% mAP. The figure compares the training configurations and their communication resource allocations.

Six single-task training configurations compared by object detection mAP and network resource allocation
Figure 5. Single-task object detection and resource allocation.

Multi-task perception

The proposed allocation achieves 99.0% weather classification accuracy, 88.8% traffic sign mAP, and 78.16% object detection mAP in the multi-task experiment.

Multi-task perception results with task-specific resource allocations and example predictions
Figure 6. Weather classification, traffic sign recognition, and object detection.

From CARLA to real-world KITTI data

To examine transfer beyond simulation, the paper trains SECOND on 3,000 CARLA point clouds for 60 epochs and evaluates it on KITTI data, reporting 58% mAP at an IoU threshold of 0.5.

Demos

Federated SECOND3D point-cloud object detection
Federated YOLOv52D image object detection
Federated LSTMBEV trajectory prediction
Vehicle–road cooperationRoad sensors for federated distillation

Citation

If CarlaFLCAV supports your research, please cite our paper.

@article{CarlaFLCAV,
  title={Federated deep learning meets autonomous vehicle perception: Design and verification},
  author={Shuai Wang and Chengyang Li and Derrick Wing Kwan Ng and Yonina C. Eldar and H. Vincent Poor and Qi Hao and Chengzhong Xu},
  journal={IEEE Network},
  year={2023},
  volume={37},
  number={3},
  pages={16--25}
}
Related paper: over-the-air federated learning

Edge Federated Learning via Unit-Modulus Over-the-Air Computation · IEEE Transactions on Communications, 2022

@article{CarlaFLOTA,
  title={Edge federated learning via unit-modulus over-the-air computation},
  author={Shuai Wang and Yuncong Hong and Rui Wang and Qi Hao and Yik-Chung Wu and Derrick Wing Kwan Ng},
  journal={IEEE Transactions on Communications},
  year={2022},
  volume={70},
  number={5},
  pages={3141--3156}
}

Contributors

Shuai Wang: bearswang

Chengyang Li: KevinLADLee