arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.08366cs.LGcs.AIcs.CV

面向端到端自动驾驶系统的缩放感知数据选择

Scaling-Aware Data Selection for End-to-End Autonomous Driving Systems

New York University(纽约大学);NVIDIA(英伟达);University of Ottawa(渥太华大学)

Tolga Dimlioglu, Nadine Chang, Maying Shen, Rafid Mahmood, Jose M. Alvarez

AI总结:

本文提出MOSAIC框架,通过分域、拟合神经缩放规律和迭代优化数据混合,提升自动驾驶模型在EPDMS指标上的性能,比基线模型减少80%的数据使用。

摘要:

大规模深度学习模型用于物理AI应用依赖于多样化的训练数据收集。这些模型和相应训练数据必须满足不同评估标准,以确保在现实环境中部署。数据选择策略可指导训练集开发,但现有框架未考虑数据点对不同指标的影响模糊性。本文提出Mixture Optimization via Scaling-Aware Iterative Collection (MOSAIC),通过将数据集分为域、从每个数据域拟合神经缩放规律,并通过迭代添加最大化指标变化的域数据来优化数据混合。我们将其应用于自动驾驶(AD),其中End-to-End(E2E)规划模型在Extended Predictive Driver Model Score(EPDMS)上进行评估,EPDMS是驾驶规则合规性指标的汇总。在此,MOSAIC在EPDMS上优于多种基线模型,数据使用量减少高达80%。

英文摘要:

Large-scale deep learning models for physical AI applications depend on diverse training data collection efforts. These models and correspondingly, the training data, must address different evaluation criteria necessary for the models to be deployable in real-world environments. Data selection policies can guide the development of the training set, but current frameworks do not account for the ambiguity in how data points affect different metrics. In this work, we propose Mixture Optimization via Scaling-Aware Iterative Collection (MOSAIC), a general data selection framework that operates by: (i) partitioning the dataset into domains; (ii) fitting neural scaling laws from each data domain to the evaluation metrics; and (iii) optimizing a data mixture by iteratively adding data from domains that maximize the change in metrics. We apply MOSAIC to autonomous driving (AD), where an End-to-End (E2E) planner model is evaluated on the Extended Predictive Driver Model Score (EPDMS), an aggregate of driving rule compliance metrics. Here, MOSAIC outperforms a diverse set of baselines on EPDMS with up to 80\% less data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑