An Exploratory Study of Frequency-Aware Task Weighting for YOLOv8-Based Unified Driving Perception
Organizations: NYU Shanghai
Abstract
Unified perception enables autonomous driving systems to perform object detection, drivable-area segmentation, and lane segmentation within a single network, improving efficiency and reducing deployment complexity. Jointly optimizing multiple perception tasks remains challenging because tasks exhibit different convergence rates, loss scales, and optimization stability. Existing task-weighting methods use loss magnitude, learned uncertainty, short-term loss changes, or gradient statistics; here, we explore the frequency structure of a recent loss-history window as a complementary signal. We implement and examine Frequency-aware Task Weighting (FTW), a dynamic task-balancing rule that estimates a loss-trajectory stability proxy from the low-frequency energy ratio of recent loss histories. FTW assigns larger weights to tasks whose mean-centered loss trajectories contain a larger proportion of low-frequency power. We document FTW and two baselines under full-network static training and progressive freezing using a unified YOLOv8-based perception framework with three task-specific heads. Experiments on Mapillary Vistas compare FTW with fixed and uncertainty-based weighting under both configurations. Final holdout metrics are reported for the checkpoint with the lowest per-epoch validation loss in each run. Across six single-run configurations, static FTW has the largest derived overall score and lane mIoU, progressive FTW has the largest detection mAP, and static uncertainty weighting has the largest drivable-area mIoU. Without repeated-seed estimates, single-task baselines, or FTW ablations, these rankings are descriptive. The evidence supports the feasibility of loss-frequency-based weighting in this pipeline, but does not establish improvement over the baselines or generalization beyond the reported runs.
Figures & tables
| Weighting strategy | Training schedule | Selected epoch | Overall | Det. mAP 50:95 | Det. mAP 50 | Drive mIoU | Lane mIoU |
|---|---|---|---|---|---|---|---|
| Fixed | Static | 7 | 0.3763 | 0.0570 | 0.1032 | 0.8533 | 0.4589 |
| Fixed | Progressive | 6 | 0.3891 | 0.0555 | 0.1007 | 0.8526 | 0.5276 |
| Uncertainty | Static | 7 | 0.3842 | 0.0435 | 0.0809 | 0.8688 | 0.5091 |
| Uncertainty | Progressive | 9 | 0.3887 | 0.0441 | 0.0814 | 0.8671 | 0.5324 |
| FTW | Static | 11 | 0.4135 | 0.0575 | 0.1043 | 0.8555 | 0.6403 |
| FTW | Progressive | 11 | 0.3911 | 0.0593 | 0.1062 | 0.8539 | 0.5265 |
Adaptive Loss Weighting
Adaptive loss weighting adjusts the influence of training examples, tasks, or objectives as learning proceeds.
See all 92 papers