Learning to Flow from Generative Pretext Tasks for Neural Architecture Encoding
This addresses a practical bottleneck for researchers and practitioners in neural architecture search by making flow-based encoders faster and more efficient.
The paper tackles the slow processing speed of flow-based neural architecture encoders by introducing FGP, a pre-training method that trains encoders to capture information flow without specialized structures, achieving up to 106% improvement in Precision-1%.
The performance of a deep learning model on a specific task and dataset depends heavily on its neural architecture, motivating considerable efforts to rapidly and accurately identify architectures suited to the target task and dataset. To achieve this, researchers use machine learning models-typically neural architecture encoders-to predict the performance of a neural architecture. Many state-of-the-art encoders aim to capture information flow within a neural architecture, which reflects how information moves through the forward pass and backpropagation, via a specialized model structure. However, due to their complicated structures, these flow-based encoders are significantly slower to process neural architectures compared to simpler encoders, presenting a notable practical challenge. To address this, we propose FGP, a novel pre-training method for neural architecture encoding that trains an encoder to capture the information flow without requiring specialized model structures. FGP trains an encoder to reconstruct a flow surrogate, our proposed representation of the neural architecture's information flow. Our experiments show that FGP boosts encoder performance by up to 106% in Precision-1%, compared to the same encoder trained solely with supervised learning.