Younger: The First Dataset for Artificial Intelligence-Generated Neural Network Architecture

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Zhengxin, Gao, Wanling, Peng, Luzhou, Huang, Yunyou, Tang, Fei, Zhan, Jianfeng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929394258280448
author Yang, Zhengxin
Gao, Wanling
Peng, Luzhou
Huang, Yunyou
Tang, Fei
Zhan, Jianfeng
author_facet Yang, Zhengxin
Gao, Wanling
Peng, Luzhou
Huang, Yunyou
Tang, Fei
Zhan, Jianfeng
contents Designing and optimizing neural network architectures typically requires extensive expertise, starting with handcrafted designs and then manual or automated refinement. This dependency presents a significant barrier to rapid innovation. Recognizing the complexity of automatically generating neural network architecture from scratch, we introduce Younger, a pioneering dataset to advance this ambitious goal. Derived from over 174K real-world models across more than 30 tasks from various public model hubs, Younger includes 7,629 unique architectures, and each is represented as a directed acyclic graph with detailed operator-level information. The dataset facilitates two primary design paradigms: global, for creating complete architectures from scratch, and local, for detailed architecture component refinement. By establishing these capabilities, Younger contributes to a new frontier, Artificial Intelligence-Generated Neural Network Architecture (AIGNNA). Our experiments explore the potential and effectiveness of Younger for automated architecture generation and, as a secondary benefit, demonstrate that Younger can serve as a benchmark dataset, advancing the development of graph neural networks. We release the dataset and code publicly to lower the entry barriers and encourage further research in this challenging area.
format Preprint
id arxiv_https___arxiv_org_abs_2406_15132
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Younger: The First Dataset for Artificial Intelligence-Generated Neural Network Architecture
Yang, Zhengxin
Gao, Wanling
Peng, Luzhou
Huang, Yunyou
Tang, Fei
Zhan, Jianfeng
Machine Learning
Artificial Intelligence
Designing and optimizing neural network architectures typically requires extensive expertise, starting with handcrafted designs and then manual or automated refinement. This dependency presents a significant barrier to rapid innovation. Recognizing the complexity of automatically generating neural network architecture from scratch, we introduce Younger, a pioneering dataset to advance this ambitious goal. Derived from over 174K real-world models across more than 30 tasks from various public model hubs, Younger includes 7,629 unique architectures, and each is represented as a directed acyclic graph with detailed operator-level information. The dataset facilitates two primary design paradigms: global, for creating complete architectures from scratch, and local, for detailed architecture component refinement. By establishing these capabilities, Younger contributes to a new frontier, Artificial Intelligence-Generated Neural Network Architecture (AIGNNA). Our experiments explore the potential and effectiveness of Younger for automated architecture generation and, as a secondary benefit, demonstrate that Younger can serve as a benchmark dataset, advancing the development of graph neural networks. We release the dataset and code publicly to lower the entry barriers and encourage further research in this challenging area.
title Younger: The First Dataset for Artificial Intelligence-Generated Neural Network Architecture
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2406.15132