Abstract:Although Transformer architectures have been successfully applied to graph data with the advent of Graph Transformer, current design of Graph Transformer still heavily relies on human labor and expertise knowledge to decide proper neural architectures and suitable graph encoding strategies at each Transformer layer. In literature, there have been some works on automated design of Transformers focusing on non-graph data such as texts and images without considering graph encoding strategies, which fail to handle the non-euclidean graph data. In this paper, we study the problem of automated graph Transformer, for the first time. However, solving these problems poses the following challenges: i) how can we design a unified search space for graph Transformer, and ii) how to deal with the coupling relations between Transformer architectures and the graph encodings of each Transformer layer. To address these challenges, we propose Automated Graph Transformer (AutoGT), a neural architecture search framework that can automatically discover the optimal graph Transformer architectures by joint optimization of Transformer architecture and graph encoding strategies. Specifically, we first propose a unified graph Transformer formulation that can represent most of state-of-the-art graph Transformer architectures. Based upon the unified formulation, we further design the graph Transformer search space that includes both candidate architectures and various graph encodings. To handle the coupling relations, we propose a novel encoding-aware performance estimation strategy by gradually training and splitting the supernets according to the correlations between graph encodings and architectures. The proposed strategy can provide a more consistent and fine-grained performance prediction when evaluating the jointly optimized graph encodings and architectures. Extensive experiments and ablation studies show that our proposed AutoGT gains sufficient improvement over state-of-the-art hand-crafted baselines on all datasets, demonstrating its effectiveness and wide applicability.

Genetic Algorithm-based Transformer Architecture Design for Neural Machine Translation.

Evolving transformer architecture for neural machine translation

Searching Better Architectures for Neural Machine Translation

The Evolved Transformer

Accelerating Transformer for Neural Machine Translation.

Deep Transformer Modeling Via Grouping Skip Connection for Neural Machine Translation

Design of a Modified Transformer Architecture Based on Relative Position Coding

Transformer: A General Framework from Machine Translation to Others

Twin-GAN for Neural Machine Translation

AutoGT: Automated Graph Transformer Architecture Search

Optimizing the Structures of Transformer Neural Networks Using Parallel Simulated Annealing

Dynamic Past and Future for Neural Machine Translation

Massive Exploration of Neural Machine Translation Architectures

Modeling Past and Future for Neural Machine Translation

Lattice-Based Transformer Encoder for Neural Machine Translation

Explicitly Modeling Word Translations in Neural Machine Translation

AutoTrans: Automating Transformer Design via Reinforced Architecture Search

Improving Neural Machine Translation with Soft Template Prediction

Learning Deep Transformer Models For Machine Translation

Optimization of English Machine Translation by Deep Neural Network under Artificial Intelligence

Improving Neural Machine Translation Model with Deep Encoding Information