“一心多用”的方法——多任务学习

本文首发于公众号“我AI”,欢迎关注,共同进步。

本篇文章从自动驾驶中的多任务开始讲起,引出多任务学习问题的定义、挑战以及研究方向。

金庸先生的《射雕英雄传》中塑造了一个“老顽童”周伯通的角色,天性爱玩。在被黄药师困于桃花岛的十五年里,为了打发无聊的时间,萌生出“左手与右手”打架的想法,遂创出“左右互搏术”这一精妙且有趣的武学。

左右互搏的本质上是一心二用,左手与右手能够同时做不同的任务,例如一手画圆、一手画方。

如果让我形象解释多任务学习的话:左右互搏便是一个非常经典的多任务学习。

图源网络

虽然普通人不经过训练,很难做到左右手互搏,但是“一心二用”甚至“一心多用”这件事情却是在普通人的日常生活中再稀疏平常不过了:比如我现在就是在一边写这篇文章一边听着歌。

在开车的任务中,我们更是要“眼观六路、耳听八方”:同时处理跟前车保持车距、观察其他车辆、保持车道、观察红绿灯/道路标志牌等多个任务。

图源[1]

基于视觉的自动驾驶,需要的不仅仅是一个深度神经网络,而是多个深度神经⽹络各司其职的组合体。在驾驶场景中,需要以视觉系统完成不同物体的检测与分类,包括感知道路上的车辆、行人、交通灯(不分辨状态)和路标;对交通灯的状态进行分类(红色、黄色或绿色);识别路标类型(停车/限速/单行道等);检测识别车辆必须停车和等待的情况,例如交叉路口/大型停车场……

对于这些目标,它们都对应同等重要的识别任务,必须以最高效的方式得到并行感知和处理。

为了完成对复杂环境的感知,Tesla设计了一个多任务的深度学习模型HydraNet:包括了48个神经网络、1000个不同的输出,完整的训练一次需要70000个GPU小时(一台有8块GPU的训练服务器需要耗费一年时间完成一次训练)。[1]

图源网络,Tesla FSD beta版

多任务学习(Multi-task learning)是机器学习中的一种重要方法,它是指给定一批学习任务,通过对这批任务的联合学习,达到以下一个或多个目的

1. 降低模型大小和复杂度,提升运算速度,减小计算开销;

2. 利用任务之间的相关性,提升各个任务的性能;

3. 减少某些任务对样本数量的依赖,减轻因样本稀少造成的过拟合。

多任务学习与迁移学习有异曲同工之妙,都涉及到不同学习任务之间的相互帮助,但差别在于迁移学习的帮助是单向的(从A任务迁移知识到B任务上),而多任务学习往往是双向的(A与B互相促进)。

尽管多任务学习能够带来上面提到的3点优点,但在具体的算法实现上,面临的挑战极大:

    1. 任务间干扰:有些任务在训练过程中可能是正相关的,能够彼此促进,有些则可能彼此干扰。因此需要甄别任务的相关性,避免出现由于任务之间干扰导致训练过程中出现Negative transfer的现象(多任务训练结果差于任何一个单任务训练)。本想“一心二用”,结果成了“三心二意”;

图源网络

    2. 训练难度大:对同一个神经网络,不同任务的训练难易度也会不同。训练难易度可能体现在所需数据量上,往往会导致一个任务已经过拟合(Overfitting)需要停止训练而另一个任务还在欠拟合(Underfitting),需要继续训练;

    3. 模型的训练和推理可能具有时变性:以自动驾驶视觉系统为例,随着时间的推移,系统需要根据当前的车辆自动驾驶需求,灵活地、分批次地、甚至同时并发地激活多任务网络的子网络。

针对这些挑战,在多任务学习的领域中,可以从以下几个研究方向考虑能做一些工作:

    1. 多任务网络结构设计

    2. 多任务损失函数设计

    3. 辅助学习(任务的筛选方法)

    4. 多任务网络的训练方法

对这些研究方向的展开,以后会以论文解读的形式逐步展开,进一步和大家分享。

参考资料:

[1] Multi-task learning in the wildness. https://slideslive.com/38917690/multitask-learning-in-the- wilderness

- END -

新朋友们可以看看我过往的相关文章

【相关推荐阅读】

模式识别学科发展报告丨前言

走不得的捷径:shortcut learning捷径学习

Vision:从具象到抽象

深度学习在计算机视觉中应用的三大问题

AI随想(三):你玩得起AI吗

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

推荐阅读更多精彩内容