ai.hackcv
论文精选 61arXiv

Transformer-Guided Swarm Intelligence for Frugal Neural Architecture Search· 节俭神经架构搜索的Transformer引导群智能

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.

AI 解读论文

结合Transformer和蜂群算法,实现高效且节俭的神经架构搜索。

核心方法
提出了一种结合自回归Transformer控制器和人工蜂群算法的混合框架,使用动态熵机制防止早熟收敛,并通过算法惩罚网络深度来减少模型膨胀。
适合谁读
研究者 / 工程师 / 产品
要解决的问题
神经架构搜索通常需要大量计算资源,难以在普通硬件上进行。
关键实验
在CIFAR-10数据集上达到了84.85%的准确率,参数量仅为约174,000;在信用卡欺诈检测中,F1-Score达到0.71,参数量约为4,600。
主要贡献
在标准GPU上解决了元启发式算法的“冷启动”问题,发现了比基线模型更小且更高效的神经网络架构。
意义与局限
该方法使得神经架构搜索更加民主化和高效,适用范围广,尤其适合边缘部署。但可能在更复杂的任务上效果有限。
领域:cs.LG作者:Romain Amigon
相关推荐

本站内容由 LLM 精选聚合,原文版权归 arXiv 所有 · 摘录仅供参考