End-to-end vision-based imitation learning in autonomous driving has shown promise, but traditional approaches lack confidence estimation and precision.
A dual-head neural network architecture is introduced that combines regression and classification heads to improve decision reliability.
The regression head predicts continuous driving actions, while the classification head estimates confidence for corrections in low-confidence scenarios.
Experimental results demonstrate improved driving stability, reduced lane deviation, and enhanced trajectory accuracy compared to regression-only models.