如何用超大规模真实驾驶视频数据集做端到端自动驾驶

 姓名:丁英琦

 学号:17101223408

 转载自:https://yq.aliyun.com/articles/228725 有改动

【嵌牛导读】本文讲的是如何用超大规模真实驾驶视频数据集做端到端自动驾驶,自动驾驶是人工智能落地的重要应用之一,许多业界公司和学校实验室都在以自己的方法研究自动驾驶系统。本次雷锋网 AI 科技评论请到了许华哲博士介绍他们团队研究的端到端自动驾驶系统。

【嵌牛鼻子】人工智能 自动驾驶

【嵌牛提问】如何用超大规模真实驾驶视频数据集做端到端自动驾驶?

【嵌牛正文】

许华哲是UC Berkeley 博士,师从 Prof. Trevor Darrell。对计算机视觉、自动驾驶、强化学习感兴趣。本科毕业于清华大学电子工程系。

这次分享主要基于徐华哲和他的同学们发表的CVPR2017的收录文章「End-to-end Learning of Driving Models from Large-scale Video Datasets」,这是对端到端自动驾驶的一个尝试。除了文章本身,许华哲也会介绍学术角度上自动驾驶领域中有意思的任务和问题。

分享的内容包括:

自动驾驶的前世今生

许华哲他们发表的超大规模数据集

为了解决自动驾驶问题,如何用数学公式定义它

论文中构建的FCN-LSTM模型

论文中用到的新技巧“Learning with privileged information”

实验结果

自动驾驶主要分为这样几类:基于规则的自动驾驶系统,端到端的自动驾驶系统(输出直接是对车的操作),综合性方法(Intermediate,决策系统的输入是前处理系统的输出)。最后还有一个研究方向是对未来会发生的事情做预测以便规避风险。

图中是基于规则的自动驾驶系统的经典工作,检测出路上的车和车道,然后进行决策。

基于规则的系统,好处是在工业界得到了广泛使用,比如GoogleX的自动驾驶就是以基于规则的系统为主;以及系统的可解释性非常强。不好之处是难以应对不确定的情况,以及在复杂的情况下无法对所有可能的操作进行建模。

完全端到端的方法中早期的典型工作是ALVINN,浅层神经网络的结果已经表明端到端学习有形成自动驾驶系统的潜力。

Yann LeCun提出的DAVE已经可以回避路上的障碍物。

在2017年10月最新的文章中,已经可以通过高阶的运动意图对低阶的实际驾驶操作进行条件(conditioned)控制,在许华哲看来已经是水平最领先的文章。

不过话说回来,这个领域影响最大的还是NVIDIA的这篇文章,第一次借助三个摄像头用端到端的方法开起了真车。

端到端驾驶的好处是,模型可以直接对图像输入作出回应,不需要人类干预;坏处是,系统对于不同的车辆系统需要逐一校准,以及解释性差

这项综合自动驾驶系统的典型方法中,用深度学习方法学到一系列行驶中的路况信息,然后把这些路况信息输入决策模型;决策模型中则可以再设定一些规则。

这种方法的解释性介于端到端和基于规则的系统之间,但还是需要人工定义许多规则,仍然有“无法完全定义系统规则”的问题

未来视频预测的目的是帮助训练自动驾驶系统,效果还有很多提升的空间,但是许华哲认为与现阶段各大自动驾驶平台的自动驾驶模拟器相比,这方面研究的意义还有待讨论。

接下来,值得聊的话题是数据集。大家都知道ImageNet对深度学习领域有着深远的影响,因为它提供了检验算法的绝佳条件。

从近年的错误率显著下降来看,大家都认为图像分类问题是一个近乎于解决了的问题,ImageNet就在其中起到了非常大的贡献。

那么,许华哲认为自动驾驶领域也需要一个起到同样作用的数据集,它应当由众包得到,应当是真实的、足够难的、有一个明确的任务目标

对于任务目标,他们把它公式化为了一个预测车辆未来位置的问题。

这样做的好处是,泛化性强,可以运用于对任意的车辆,也可以用未经过校准的数据;他们的模型是可以端到端训练的;同时模型可以一次预测多个方向

模型中要预测的egomotion用方向角α和位移s描述,这样也就不需要直接对车辆内的方向盘和踏板进行控制,简化了系统的前期研究。

模型的数学定义如上图,评估给定的状态s和动作a的得分,其中s包括视觉信息和车辆的当前状态;a可以是分离的几个状态,也可以是连续的数值输出;最精确的是输出6个自由度上的运动

下一个问题就是如何评估这个模型。借助数据集的话,借助语言模型里perplexity预测的思路,把过去的状态作为序列,预测序列中的下一个元素

在他们的数据集之前,大家比较熟悉几个数据集包括KITTI、Cityscape、英特尔的GTA。

KITTI的特点是数据量比较小,但图像质量高,数据种类丰富,横向很宽

Torcs是在游戏里采的数据

Cityscape是一个高质量的图像语意分割数据集,但是并没有放出对应的视频

英特尔的GTA数据集也是以语意分割为主,虽然基于虚拟的游戏环境但是很逼真,可以看作最接近真实世界的游戏

几个数据集做对比的话,场景的丰富程度上KITTI力压所有数据集,其余更多方面则都有缺失

时长方面,Oxford的最长,围绕城市采了200小时数据。

这几个数据集都不算令人满意。为了大幅度提升时长和丰富程度,许华哲他们想到了众包的方法,与Uber合作,采集、使用了司机视角的视频。这个数据集的特点是基于真实世界的、第一人称视角,从右侧样例里可以看到内容也非常丰富。

最重要的是,这个数据集也是足够长的,比现存的任何数据集都要多出2个数量级。

这样对比下来,他们的数据集BDD-V就全面领先于其它的数据集

BDD-V数据集更多的指标如上图,30fps/720p,包括了惯性传感器IMU、陀螺仪Gyroscope、GPS和指南针的数据,而且是全面开放的。许华哲还偷偷透露,除了这个视频数据集,他们还会同步发布其它用于驾驶任务的高质量、大规模数据集。

这里正式介绍文章中的内容,一张图像输入到模型中后,模型要判断车如何行驶,最后输出中绿色部分表示可能的动作的概率分布(离散或者连续)。

论文中使用的模型结构是Dilated Fully Convolutional Networks+LSTM+Segmentation Loss。FCN用来保留空间信息,LSTM对之前时间中的信息加以利用,Segmentation语义分割则是为了把车辆从背景中分离出来(这部分属于Privileged Learning,等下还会讲到)。

上图显示的是预测离散的车辆动作(也就是前后左右)的结果,G代表ground truth,P代表预测的结果。

对于红绿灯也作出了合理的预测结果(黄灯时前进和停止一半一半,红灯时预测结果都是停止)

对于预测连续的车辆动作,沿着路的时候能判断出“不能右转”,到了路口就有了更多方向的选择

在连续预测中,有个重要问题是能否预测多个模态。可以看到,到了路口时预测结果明显分为了两个部分。

这里,为了更好地展示模型学到的特征,徐华哲从模型中抽出了conv-4层学到的特征。对于右侧的街道图,左侧是模型注意力的heat-map;两图并不是完全对齐的。可以看到,行车线、红绿灯、骑车的人和汽车(表现为两个车轮)都获得了很高的注意力。

Privileged Learning是指,模型一边预测未来的运动,一边训练一个语义分割网络,然后用语义分割网络更好地帮助模型做运动预测。P1、P2是Baseline,P3是带有Privileged Learning之后的结果,对图中这样“半辆车”的情况有更好的识别能力。

总结整篇论文

首先介绍了一个不依赖执行机构的自动驾驶通用模型

收集、公开了最大、最丰富的基于真实世界的数据集

提出了新的网络架构,能够输出多模态行为

确认了Privileged Learning方法的有效性,进一步提升了预测效果

第一个长度可以和人类一生开车时间相比的训练数据集

数据、模型和代码可以在https://github.com/gy20073/BDD_Driving_Model看到。

未来他们也还会继续采集、扩充数据集。目前总长度(最下方)大概是一个人一生开车时间的一半,未来希望扩充到10万个小时。

同时他们也在改进模型,并在测试场地上把真车开起来。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,377评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,390评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,967评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,344评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,441评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,492评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,497评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,274评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,732评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,008评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,184评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,837评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,520评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,156评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,407评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,056评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,074评论 2 352

推荐阅读更多精彩内容