35 research outputs found
Automated Data Augmentations for Graph Classification
Data augmentations are effective in improving the invariance of learning
machines. We argue that the corechallenge of data augmentations lies in
designing data transformations that preserve labels. This is
relativelystraightforward for images, but much more challenging for graphs. In
this work, we propose GraphAug, a novelautomated data augmentation method
aiming at computing label-invariant augmentations for graph
classification.Instead of using uniform transformations as in existing studies,
GraphAug uses an automated augmentationmodel to avoid compromising critical
label-related information of the graph, thereby producing
label-invariantaugmentations at most times. To ensure label-invariance, we
develop a training method based on reinforcementlearning to maximize an
estimated label-invariance probability. Comprehensive experiments show that
GraphAugoutperforms previous graph augmentation methods on various graph
classification tasks