Deep Symbolic Superoptimization Without Human Knowledge - MIT-IBM Watson AI Lab

image

Published on

09/25/2019

Deep symbolic superoptimization refers to the task of applying deep learning methods to simplify symbolic expressions. Existing approaches either perform supervised training on human-constructed datasets that defines equivalent expression pairs, or apply reinforcement learning with human-defined equivalent trans-formation actions. In short, almost all existing methods rely on human knowledge to define equivalence, which suffers from large labeling cost and learning bias, because it is almost impossible to define and comprehensive equivalent set. We thus propose HISS, a reinforcement learning framework for symbolic super-optimization that keeps human outside the loop. HISS introduces a tree-LSTM encoder-decoder network with attention to ensure tractable learning. Our experiments show that HISS can discover more simplification rules than existing human-dependent methods, and can learn meaningful embeddings for symbolic expressions, which are indicative of equivalence.

Please cite our work using the BibTeX below.

@inproceedings{
Shi2020Deep,
title={Deep Symbolic Superoptimization Without Human Knowledge},
author={Hui Shi and Yang Zhang and Xinyun Chen and Yuandong Tian and Jishen Zhao},
booktitle={International Conference on Learning Representations},
year={2020},
url={https://openreview.net/forum?id=r1egIyBFPS}
}