arXiv · 2609.18955
KDTwin: Task-Aware Knowledge Distillation for Lightweight Multi-Task Driving Scene Segmentation
Abstract
Efficient perception models are essential for real-time autonomous driving, where accuracy and computational cost must be carefully balanced. However, applying knowledge distillation to multi-task driving scene segmentation is challenging because drivable-area and lane segmentation exhibit different spatial characteristics and class imbalance. We propose KDTwin, a task-aware distillation framework for lightweight multi-task segmentation networks. The proposed method performs distillation at both the shared encoder and task-specific decoders. Encoder-level pairwise distillation transfers spatial relational knowledge to enhance the student's shared representation. For the decoders, we use a weighted loss for drivable-area segmentation and a boundary-aware loss for lane segmentation, enabling task-adaptive knowledge transfer without increasing inference complexity. Experiments on BDD100K show consistent improvements across the evaluated CNN-based and Transformer-based student models without increasing inference-time parameters or FLOPs. The results show that designing distillation objectives according to task-specific characteristics can effectively enhance multi-task segmentation performance for autonomous driving. The source code is available at https://github.com/chequanghuy/KDTwin.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Huy Che, Minh-Khoi Do, Dinh-Duy Phan, Duc-Khai Lam. 2026-07-28. KDTwin: Task-Aware Knowledge Distillation for Lightweight Multi-Task Driving Scene Segmentation. https://arxiv.org/abs/2609.18955
Cite the original work for its findings. Save a collection to share your selection of sources.