Rethinking Self-driving: Multi-task Knowledge for Better Generalization and Accident Explanation Ability
Integrates multi-task perception (segmentation, depth) into end-to-end driving, boosting generalization by 20% in unseen environments and enabling accident explanation.
Key Findings
Methodology
• 输入:RGB图像;• 感知模块:采用残差网络结构(ResNet)提取特征,解码生成分割图(‘什么&哪里’)和深度图(‘多远’);• 训练策略:先用多任务损失(交叉熵和二值交叉熵)训练感知模块,学习像素级理解;• 固定感知模块参数,输入其输出特征到驾驶决策模块;• 驱动模块:基于残差结构,输出四个高层指导(直行、左转、右转、直行+转弯),每个方向对应一组控制(转向、油门/刹车);• 训练:在CARLA模拟器中,结合数据增强和数据平衡,优化模型性能。
AI Executive Summary
Autonomous driving has seen rapid advances through end-to-end deep learning models, yet challenges in generalization and interpretability persist. Traditional models often perform well in training environments but falter in unseen scenarios, limiting safety and deployment. Moreover, their black-box nature hampers understanding of failure causes, impeding trust and regulatory approval. This paper introduces a novel framework that incorporates multi-task perception—specifically segmentation and depth estimation—as intermediate representations within an end-to-end driving system. The perception module, based on residual networks with hybrid dilated convolutions, learns pixel-level scene understanding, providing 'what & where' and 'how far' knowledge. These outputs serve as interpretable cues, enabling visualization of the model’s perception and diagnosis of failures. The driving module, built with residual blocks, utilizes these cues to generate control commands for various high-level guidance signals. The training employs a stepwise strategy: first, the perception module learns multi-task knowledge; then, its parameters are frozen while the driving module is trained. Extensive experiments in CARLA simulator demonstrate that this approach achieves a 20% success rate increase in untrained environments, outperforming existing methods. The visualization of segmentation and depth maps allows practitioners to trace back the cause of errors, significantly enhancing safety and reliability. This work not only advances the state-of-the-art in autonomous driving robustness but also sets a new standard for model interpretability. Future work aims to incorporate multi-modal sensors, optimize real-time performance, and validate on real-world roads, moving closer to safe, explainable autonomous vehicles.
Deep Dive
Plain Language Accessible to non-experts
想象你在厨房做饭,厨房里有很多不同的工具和食材。传统的自动驾驶模型就像只用一把刀切菜,虽然简单,但遇到不同的菜和厨房布局就容易出错。而这篇论文提出的方法,像是先学会识别所有食材(分割图)和距离(深度图),然后再用这些基础知识去做菜。这样,无论厨房怎么变,厨房里的食材和距离都能帮你做出更好的菜,也更容易找到出错的原因。它就像是学会看清楚每个食材和它们的位置,做饭时就更有信心,也能告诉你为什么菜会失败,比如盐放多了或者火太大。这个方法让自动驾驶变得更聪明、更可靠,也更容易理解它为什么会出错,就像厨师知道自己哪里做得不对一样。
ELI14 Explained like you're 14
想象你在玩一款超级复杂的游戏,你需要控制角色跑、跳、避障,但如果只用一个按钮,遇到新关卡就可能搞不定。这篇论文就像是教你先学会识别每个障碍物和距离(用特别的眼镜看场景),然后再用这些信息来决定怎么操作。这样,不管关卡怎么变,你都能更快适应,也能知道自己为什么会失败,比如没看到前面的坑或者误判了障碍。它让游戏变得更聪明,也让你更容易理解自己哪里出错了。就像学会用不同的工具和信息帮你应对新挑战,不再只靠记忆,而是用理解去解决问题。
Abstract
Current end-to-end deep learning driving models have two problems: (1) Poor generalization ability of unobserved driving environment when diversity of training driving dataset is limited (2) Lack of accident explanation ability when driving models don't work as expected. To tackle these two problems, rooted on the believe that knowledge of associated easy task is benificial for addressing difficult task, we proposed a new driving model which is composed of perception module for \textit{see and think} and driving module for \textit{behave}, and trained it with multi-task perception-related basic knowledge and driving knowledge stepwisely. Specifically segmentation map and depth map (pixel level understanding of images) were considered as \textit{what \& where} and \textit{how far} knowledge for tackling easier driving-related perception problems before generating final control commands for difficult driving task. The results of experiments demonstrated the effectiveness of multi-task perception knowledge for better generalization and accident explanation ability. With our method the average sucess rate of finishing most difficult navigation tasks in untrained city of CoRL test surpassed current benchmark method for 15 percent in trained weather and 20 percent in untrained weathers. Demonstration video link is: https://www.youtube.com/watch?v=N7ePnnZZwdE