Dataset Augmentation in Feature Space

By Terrance Devries, Graham W. Taylor. 2017

460 citations recorded by the source corpus.

Dataset augmentation, the practice of applying a wide array of domain-specific transformations to synthetically expand a training set, is a standard tool in supervised learning. While effective in tasks such as visual recognition, the set of transformations must be carefully designed, implemented, and tested for every new domain, limiting its re-use and generality. In this paper, we adopt a simpler, domain-agnostic approach to dataset augmentation. We start with existing data points and apply simple transformations such as adding noise, interpolating, or extrapolating between them. Our main insight is to perform the transformation not in input space, but in a learned feature space. A re-kindling of interest in unsupervised representation learning makes this technique timely and more effective. It is a simple proposal, but to-date one that has not been tested empirically. Working in the space of context vectors generated by sequence-to-sequence models, we demonstrate a technique that is effective for both static and sequential data.

Subfields: Artificial Intelligence, Cancer Research, Representation & Self-Supervised Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Quoc V. Le
  2. Mohit Bansal
  3. Yoshua Bengio
  4. Sergey Levine
  5. Trevor Darrell
  6. Percy Liang
  7. Dacheng Tao
  8. Jianfeng Gao

In Cancer Research

  1. Yang Zhang
  2. Christina Heinze-Deml
  3. David Lopez-Paz
  4. Jinwoo Shin
  5. Hung-Yu Tseng
  6. Sergey Zakharov
  7. Alex ChiChung Kot
  8. Ming-Hsuan Yang

In Representation & Self-Supervised Learning

  1. Quoc V. Le
  2. Ekin Dogus Cubuk
  3. Yoshua Bengio
  4. Honglak Lee
  5. Barret Zoph
  6. Jonathon Shlens
  7. Kihyuk Sohn
  8. Pieter Abbeel

Citations in the corpus