about
AQuaSurF: Discover better activation functions for your ML task (arxiv.org)
2 points by binghamgarrett on Dec 7, 2023 | hide | past | pdf | 1 comment on HN

In plain words: Instead of training a network with each candidate activation function, this study predicts how well one will do from cheap pre-training measurements. Using that shortcut, it found a smooth S-shaped function that beat every tested activation, including the rectifier shapes deep learning relies on.

Abstract · Efficient Activation Function Optimization through Surrogate Modeling

Carefully designed activation functions can improve the performance of neural networks in many machine learning tasks. However, it is difficult for humans to construct optimal activation functions, and current activation function search algorithms are prohibitively expensive. This paper aims to improve the state of the art through three steps: First, the benchmark datasets Act-Bench-CNN, Act-Bench-ResNet, and Act-Bench-ViT were created by training convolutional, residual, and vision transformer architectures from scratch with 2,913 systematically generated activation functions. Second, a characterization of the benchmark space was developed, leading to a new surrogate-based method for optimization. More specifically, the spectrum of the Fisher information matrix associated with the model's predictive distribution at initialization and the activation function's output distribution were found to be highly predictive of performance. Third, the surrogate was used to discover improved activation functions in several real-world tasks, with a surprising finding: a sigmoidal design that outperformed all other activation functions was discovered, challenging the status quo of always using rectifier nonlinearities in deep learning. Each of these steps is a contribution in its own right; together they serve as a practical and theoretical foundation for further research on activation function optimization.

Garrett Bingham, Risto Miikkulainen
arXiv:2301.05785 · cs.LG, cs.NE · submitted Jan 13, 2023 · updated Nov 8, 2023
abstract · pdf · html · NeurIPS 2023. 28 pages, 16 figures, 6 tables

add comment on HN

AQuaSurF is a new algorithm that automatically discovers better activation functions for your machine learning task, helping to improve performance.

Here's how it works:

• Embrace data-driven insights: AQuaSurF analyzes thousands of activation functions across diverse architectures and datasets, identifying what truly matters for performance. No more intuition-based guesswork!

• Unleash the power of the AQuaSurF algorithm: This efficient algorithm leverages a unique surrogate performance measure to efficiently navigate the vast space of activation functions, uncovering hidden gems quickly.

• Customize and experiment: AQuaSurF's open-source code provides the flexibility to tailor the search to your specific needs and experiment with different search strategies.

• Benchmark and compare: Utilize the AQuaSurF benchmark datasets to compare the performance of various activation functions and evaluate the effectiveness of different search algorithms — in seconds.

• Challenge the status quo: Don't settle for the mediocrity of the mainstream. AQuaSurF empowers you to challenge the status quo and pioneer the future of deep learning. Discover the activation functions that will take your models to the next level and beyond.

Further reading: • Paper: https://arxiv.org/abs/2301.05785 • AQuaSurF — find a better activation function for your task: https://github.com/cognizant-ai-labs/aquasurf • Act-Bench — test out new search algorithms (similar to NAS-Bench-101, etc.): https://github.com/cognizant-ai-labs/act-bench/ • NeurIPS link: https://neurips.cc/virtual/2023/poster/71442