Research

Triangle Method for Dense ReLU Layers [SWR-25-72]

DOE

Source: doe_osti

Questions this answers

Sources

Source record

The underlying research record — supporting evidence for the object above.

Key Details

Posted Date
Source
doe_osti

Description

This software is an implementation of the methods for initializing and training neural networks to be more efficient per parameter, described more fully below and in the related publication: In theory, depth should make a ReLU network EXPONENTIALLY more efficient by enabling it to produce an exponential number of piecewise linear sections in its output. This reasoning is largely based on the work of mathematicians that have hand-constructed networks that make good use of depth. In practice however, even very deep ReLU networks that have been randomly initialized will behave identically to their shallow counterparts - missing an entire exponential dimension of efficiency. The triangle method is a first attempt at realizing the exponential potential of deep networks. Instead of randomly setting weights, we force pairs of neurons in each layer learn to build triangles (i.e. functions from [0,1] -> [0,1] that look like triangles). This is a very efficient pattern for generating lots of linear pieces because composing two triangular functions doubles the number of pieces with each composition. The triangle method is more than just a different initialization, it is a new paradigm of training. Instead of making direct updates to the matrix weights, we do an extra step of backpropagation to collect the derivatives of the loss function with respect to the shapes of the triangles, training them to tilt left or right. This process essentially holds the networks hand throughout the loss landscape and forces it to always use depth effectively by producing triangular shapes internally. This can produce several orders of magnitude of improvement on convex one-dimensional regression problems. Much more theoretical work is needed to realize its full potential beyond this context, but the implementation in this repository will still work in arbitrary numbers of dimensions. The file Triangle_Method.py is a generalized form of the method that will build each neuron its own custom 1. Authors: Milkert, Max [National Renewable Energy Laboratory (NREL), Golden, CO (United States)]. DOE Contract: AC36-08GO28308

Frequently Asked Questions

What is this record?+
It is a publication associated with DOE — published findings, a technical standard, or a notice. It reports results or specifications. There is nothing here to apply for.

Where this record sits

What else do you want to know?

100M+ government records — search across all categories