arXiv:2603.06601cs.LG2026-03

让神经网络按需激活单元,动态节省计算资源。

Switchable Activation Networks

  • 每个神经元自带输入决定的开关,动态控制是否参与计算。
  • 推理时计算量减少40%以上,精度几乎不变。
  • 适合部署在手机、传感器等算力受限设备上。

深度神经网络,尤其是大规模生成模型如大语言模型(LLMs)和大视觉-动作模型(LVAs),在多个领域表现卓越,但其高昂的计算成本阻碍了在资源受限环境中的部署。现有高效技术仅能部分缓解问题:丢弃法改善训练正则化,但对推理无影响;剪枝与低秩分解虽可压缩模型,但生成静态结构,适应性差。本文提出SWAN(可切换激活网络),为每个神经单元引入确定性、依赖输入的二值门控机制,使网络学习在何时激活或关闭单元。该动态控制策略实现计算自适应分配,减少冗余同时保持精度。不同于传统剪枝,SWAN并非训练后简单压缩;它学习有结构的、上下文相关的激活模式,支持高效动态推理,并可转化为紧凑的密集模型用于部署。通过将效率重构为学习激活控制的问题,SWAN统一了稀疏性、剪枝与自适应推理的优势于单一范式。除计算增益外,这一视角揭示了更普遍的神经计算原则:激活不应固定,而应随上下文变化,指向可持续人工智能、边缘智能及受生物大脑可塑性启发的未来架构。

原文摘要 · Abstract (English)

Deep neural networks, and more recently large-scale generative models such as large language models (LLMs) and large vision-action models (LVAs), achieve remarkable performance across diverse domains, yet their prohibitive computational cost hinders deployment in resource-constrained environments. Existing efficiency techniques offer only partial remedies: dropout improves regularization during training but leaves inference unchanged, while pruning and low-rank factorization compress models post hoc into static forms with limited adaptability. Here we introduce SWAN (Switchable Activation Networks), a framework that equips each neural unit with a deterministic, input-dependent binary gate, enabling the network to learn when a unit should be active or inactive. This dynamic control mechanism allocates computation adaptively, reducing redundancy while preserving accuracy. Unlike traditional pruning, SWAN does not simply shrink networks after training; instead, it learns structured, context-dependent activation patterns that support both efficient dynamic inference and conversion into compact dense models for deployment. By reframing efficiency as a problem of learned activation control, SWAN unifies the strengths of sparsity, pruning, and adaptive inference within a single paradigm. Beyond computational gains, this perspective suggests a more general principle of neural computation, where activation is not fixed but context-dependent, pointing toward sustainable AI, edge intelligence, and future architectures inspired by the adaptability of biological brains.

神经网络动态计算轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。