play and learn: using video games to train computer …schmidtm/documents/2016_nipsw_playlea… ·...

12
PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER VISION MODELS Alireza Shafaei, James J. Little, and Mark Schmidt DECEMBER 2016 – CIFAR LEARNING IN MACHINES AND BRAINS

Upload: others

Post on 06-Jul-2020

0 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER VISION MODELSAlireza Shafaei, James J. Little, and Mark SchmidtD E C E M B E R 2 0 1 6 – C I F A R L E A R N I N G I N M A C H I N E S A N D B R A I N S

Page 2: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Single-Label vs. Dense-Labels• CNNs are now the dominant models in computer vision.• Key ingredient: large labeled datasets (ImageNet: >13 million images).

• What about dense labels as in image segmentation?• Much harder to get large labeled datasets (Cityscapes: 5k images).

Page 3: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Simulations vs. Reality• Classic solution: simulate more labeled examples.• This work: can video games augment real data?

Page 4: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Why Video Games?• Free and scaleable groundtruth annotation for many tasks:• Segmentation, depth, surface normal, shadows, optical flow, and so on.

• Active control of environment:• Viewpoints, time of day, weather, dangerous situations, and so on.

Page 5: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew
Page 6: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew
Page 7: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Fine-tuning performance (Segmentation)

Page 8: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Fine-tuning performance (Segmentation)

Page 9: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Performance without fine-tuning.

Page 10: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Fine-Tuning Performance (Depth)

Page 11: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

Take-Home Message• Training on video game synthetic data achieved:• Similar test accuracy to training on another real dataset.• Better test accuracy when using real data to fine-tune.

• Video games may offer a way to compile large labeled datasets.

Page 12: PLAY AND LEARN: USING VIDEO GAMES TO TRAIN COMPUTER …schmidtm/Documents/2016_NIPSw_PlayLea… · Self-Driving Cars Will Use GTA V To Learn How To Drive Run For The Hills Andrew

UBCCSis(also)hiringinallranks/areas.