English

Adapt, But Don't Forget: Fine-Tuning and Contrastive Routing for Lane Detection under Distribution Shift

Computer Vision and Pattern Recognition 2025-07-28 v1 Machine Learning

Abstract

Lane detection models are often evaluated in a closed-world setting, where training and testing occur on the same dataset. We observe that, even within the same domain, cross-dataset distribution shifts can cause severe catastrophic forgetting during fine-tuning. To address this, we first train a base model on a source distribution and then adapt it to each new target distribution by creating separate branches, fine-tuning only selected components while keeping the original source branch fixed. Based on a component-wise analysis, we identify effective fine-tuning strategies for target distributions that enable parameter-efficient adaptation. At inference time, we propose using a supervised contrastive learning model to identify the input distribution and dynamically route it to the corresponding branch. Our framework achieves near-optimal F1-scores while using significantly fewer parameters than training separate models for each distribution.

Keywords

Cite

@article{arxiv.2507.18653,
  title  = {Adapt, But Don't Forget: Fine-Tuning and Contrastive Routing for Lane Detection under Distribution Shift},
  author = {Mohammed Abdul Hafeez Khan and Parth Ganeriwala and Sarah M. Lehman and Siddhartha Bhattacharyya and Amy Alvarez and Natasha Neogi},
  journal= {arXiv preprint arXiv:2507.18653},
  year   = {2025}
}

Comments

Accepted to ICCV 2025, 2COOOL Workshop. Total 14 pages, 5 tables, and 4 figures