这篇论文告诉你,不用几千GPU天也能搜出好架构,普通3060显卡3小时就能找到比ResNet-20更小的CIFAR-10模型,还能直接优化欺诈检测的F1分数。
该论文提出了一种结合自回归Transformer控制器(强化学习训练)和人工蜂群算法(ABC)的混合NAS框架,可在消费级GPU(NVIDIA RTX 3060)上运行。在CIFAR-10数据集中,该方法用3小时搜索到参数量约17.4万的架构,达到84.85%准确率,规模和搜索成本均低于ResNet-20。在信用卡欺诈检测任务上,优化F1-Score达到0.71,模型仅需约4600参数。该框架解决了元启发式方法的冷启动问题,并通过深度惩罚抑制模型臃肿。
Transformer-Guided Swarm Intelligence for Frugal Neural Architecture Search
Neural Architecture Search (NAS) has automated the design of deep learning models but traditionally requires massive computational resources, often measured in thousands of GPU-days. In this paper, we propose a frugal and memetic NAS framework designed to democratize architecture design on consumer-grade hardware. Our approach combines the global macro-search capabilities of an autoregressive Transformer controller, trained via Reinforcement Learning (RL), with the local micro-exploitation of an Artificial Bee Colony (ABC) algorithm. To prevent premature convergence during the RL phase, we introduce a dynamic entropy mechanism that forces topological exploration upon detection of performance stagnation. Evaluated on a standard GPU (NVIDIA RTX 3060), our hybrid method effectively resolves the "cold-start" problem inherent in metaheuristics. By algorithmically penalizing network depth, our framework actively mitigates model bloat: on the CIFAR-10 dataset, it discovers an efficient architecture reaching 84.85% accuracy with only $\sim$174,000 parameters (significantly smaller than standard baselines like ResNet-20) in 3 hours of search time. Furthermore, we demonstrate the framework's flexibility by applying it to credit card fraud detection, directly optimizing the F1-Score on highly imbalanced tabular data to reach a F1-Score of 0.71 with a compact network of $\sim$4,600 parameters. These results suggest that our approach can yield tailored, accessible, and highly parameter-efficient deep learning models suitable for edge deployment.
- vLLM05:55原文