Multi-dimensional Search with Strip Convolution and R-Squared Loss for Lane Detection

Zijun Lin,Peng Ye,Tao Chen,Shengji Tang

DOI: https://doi.org/10.1109/ijcnn60899.2024.10650216

2024-01-01

Abstract:Existing deep learning methods for lane detection mainly rely on commonly used backbones such as ResNet, equipped with various hand-craft modules for feature extraction and modeling. Such manually designed backbones and modules heavily rely on human expert experience, and it is difficult to guarantee their extracted features always fit well with various lanes. Besides backbones, the imbalanced number of lanes with different curvature in datasets and the ways modeling lane lines incur a strong curvature bias. Therefore, inspired by the recent popularity of AutoML techniques such as Neural Architecture Search (NAS), we propose an automatic lane detection architecture design framework, namely StripLaneNet-NAS, to solve the above problem. To enable the searched model structure well capture various kinds of lane features, we propose a multi-dimensional search space equipped with specially designed lane-specific Strip Convolution Modules (SCM), and correspondingly propose an adaptive solution space regularization loss, to accelerate and optimize the multi-dimensional search process. A novel R-squared loss is further proposed in the optimization objective to alleviate the curvature bias problem as mentioned above. Experiments are conducted on two benchmarks, i.e., TuSimple and CULane, and results show that our method outperforms state-of-the-art lane detection baselines, achieving the fastest inference speed with a maximum of 74.0% parameter reduction over the baselines.